Flow matching-based generative models for MIMO channel estimation

2025-11-14

AI Summary

To address the slow sampling speed and low efficiency of diffusion models in MIMO channel estimation, this paper introduces flow matching—its first application to this task—proposing a conditional velocity-field-based generative framework. By constructing a straight-line ODE trajectory from noise to the true channel and designing an analytical velocity field dependent solely on noise statistics, the method enables deterministic one-step sampling. It eliminates the multi-step iterative inference inherent in conventional diffusion models, drastically reducing sampling overhead. Furthermore, integrating a conditional generative network with noise-statistics-guided training enhances estimation accuracy across diverse channel scenarios, consistently outperforming state-of-the-art diffusion-based methods. Experimental results demonstrate that the proposed approach achieves high-fidelity CSI reconstruction while maintaining real-time capability and robustness under varying channel conditions.

Technology Category

• Computer Vision: Diffusion Models for Vision
• Search and Optimization: Sampling/Simulation-based Search
• Intelligent Robots: State Estimation

Application Category

• Systems and Infrastructure for Web, Mobile and WoT: Applied ML and AI for Web-based mobile applications
• User Modeling, Personalization and Recommendation: On-Device user modeling, personalization, and recommendation
• Graph Algorithms and Modeling for the Web: Foundation models and LLMs for Web-related graphs

Abstract

Diffusion model (DM)-based channel estimation, which generates channel samples via a posteriori sampling stepwise with denoising process, has shown potential in high-precision channel state information (CSI) acquisition. However, slow sampling speed is an essential challenge for recent developed DM-based schemes. To alleviate this problem, we propose a novel flow matching (FM)-based generative model for multiple-input multiple-output (MIMO) channel estimation. We first formulate the channel estimation problem within FM framework, where the conditional probability path is constructed from the noisy channel distribution to the true channel distribution. In this case, the path evolves along the straight-line trajectory at a constant speed. Then, guided by this, we derive the velocity field that depends solely on the noise statistics to guide generative models training. Furthermore, during the sampling phase, we utilize the trained velocity field as prior information for channel estimation, which allows for quick and reliable noise channel enhancement via ordinary differential equation (ODE) Euler solver. Finally, numerical results demonstrate that the proposed FM-based channel estimation scheme can significantly reduce the sampling overhead compared to other popular DM-based schemes, such as the score matching (SM)-based scheme. Meanwhile, it achieves superior channel estimation accuracy under different channel conditions.

Problem

• Slow sampling speed in diffusion-based MIMO channel estimation methods
• Need for high-precision channel state information acquisition
• Reducing sampling overhead while maintaining estimation accuracy

Innovation

• Flow matching framework for MIMO channel estimation
• Velocity field derived from noise statistics for training
• ODE Euler solver enables fast noise channel enhancement