Zhixian Kong | Searcharxiv

arXiv2026

MIMO-OFDM AI Receiver Based on Incrementally Conditioned Diffusion with Soft Decision

Conventional iterative receiver usually begins with channel estimation using sparse pilot observations and follows with data detection based on the channel estimates, and then updates channel estimation using decision feedback, and so on. In Artificial Intelligence (AI)-based receiver design, it is also crucial to make use of such progressively enriched data observations to enhance the generative channel estimation. However, the statistical characteristics and reliability of the data decisions always evolve with the channel estimation processes, which brings great challenges to the design of the overall learning framework and algorithms. In this paper, we propose \textit{Diff-Rx}, an incrementally conditioned diffusion-based receiver with co-designed data detection and channel estimation, for multiple-input multiple-output orthogonal frequency division multiplexing (MIMO-OFDM) systems. Specifically, we develop a condition-adaptive post-training method which enables the generative channel estimator to adapt to the conditioning inputs that are progressively enriched by soft data decisions, and also to implicitly align the pilot- and data-induced channel feature spaces to mitigate potential estimation errors. The threshold-free soft decisions provide smooth condition updates without condition-specific reliability tuning. We further develop a conditional diffusion transformer that is capable of performing robust channel estimation under noisy observations and various pilot patterns while reducing the conventional multi-step diffusion generation to one single step. Simulations on both statistical and site-specific ray-tracing channels show that, Diff-Rx exhibits consistent gains across different noise levels, pilot densities and modulation schemes, and works well at a pilot density as low as 1/32 while achieving significant improvement in channel estimation and data detection performances.

eess.SP↗

arXiv2026

Diffusion Inpainting MIMO-OFDM Channels with Limited Noisy Observations

Acquiring the channel state information from limited and noisy observations at pilot positions is critical for wireless multiple-input multiple-output (MIMO)-orthogonal frequency division multiplexing (OFDM) systems. In this paper, we view this process as a conditional generative task in which the partial noisy channel estimates at the pilots are utilized as a ``prompt'' to guide the diffusion ``inpainting'' of the underlying channel. To this end, we resort to a general Conditional Diffusion Transformer (CDiT) framework with a well-designed network architecture and update rule. In particular, we design a dedicated embedding strategy to encode and adapt to different pilot patterns and noise levels, and utilize a special cross-attention mechanism to align the partial raw channel observations with the denoised channel at each time step of the generation process. This architecture effectively anchors the diffusion process, enabling the model to accurately recover full channel details from limited noisy observations. Comprehensive experimental results show that, the proposed approach achieves a performance gain of over 5 dB compared to the baselines under varying noise conditions, and provides robust channel acquisition even under a sparse pilot density of 1/32 without significant performance loss compared to the denser pilot cases. Moreover, it is capable of generating high-quality channel matrices within just 10 inference steps, effectively balancing estimation accuracy with computational efficiency and inference speed. Ablation studies demonstrate the rationality of the model design and the necessity of its modules.

eess.SP↗