# Sampling-Free Diffusion Transformers for Low-Complexity MIMO Channel Estimation

arXiv: https://arxiv.org/abs/2602.02202 · HTML: https://mlsanity.com/paper/2602.02202/

## Key facts

- No citing papers in the map yet (data through 14 Oct 2026).
- Placed in Diffusion for compression and wireless, one of 14 directions in Diffusion models.
- Builds on 10 of its 17 references that are in the map.
- First posted on arXiv 2 Feb 2026.

## Placed in

- Diffusion models / [Diffusion for compression and wireless](https://mlsanity.com/diffusion/diffusion-for-compression-and-wireless/)

## Builds on (most cited first, 10 of 10)

- [Score-Based Generative Modeling through Stochastic Differential Equations](https://mlsanity.com/paper/2011.13456/) (2020)
- [Scalable Diffusion Models with Transformers](https://mlsanity.com/paper/2212.09748/) (2022)
- [Flow Matching for Generative Modeling](https://mlsanity.com/paper/2210.02747/) (2022)
- [Elucidating the Design Space of Diffusion-Based Generative Models](https://mlsanity.com/paper/2206.00364/) (2022)
- [MIMO Channel Estimation using Score-Based Generative Models](https://mlsanity.com/paper/2204.07122/) (2022)
- [Diffusion-based Generative Prior for Low-Complexity MIMO Channel Estimation](https://mlsanity.com/paper/2403.03545/) (2024)
- [Wideband Channel Estimation with A Generative Adversarial Network](https://mlsanity.com/paper/2012.12063/) (2020)
- [Flow matching-based generative models for MIMO channel estimation](https://mlsanity.com/paper/2511.10941/) (2025)
- [Enhancements in Score-based Channel Estimation for Real-Time Wireless Systems](https://mlsanity.com/paper/2509.07839/) (2025)
- [Robust MIMO Channel Estimation Using Energy-Based Generative Diffusion Models](https://mlsanity.com/paper/2510.22230/) (2025)

## Abstract (arXiv)

Diffusion model-based channel estimators have shown impressive performance but suffer from high computational complexity because they rely on iterative reverse sampling. This paper proposes a sampling-free diffusion transformer (DiT) for low-complexity MIMO channel estimation, termed SF-DiT-CE. Exploiting angular-domain sparsity of MIMO channels, we train a lightweight DiT to directly predict the clean channels from their perturbed observations and noise levels. At inference, the least square (LS) estimate and estimation noise condition the DiT to recover the channel in a single forward pass, eliminating iterative sampling. Numerical results demonstrate that our method achieves superior estimation accuracy and robustness with significantly lower complexity than state-of-the-art baselines.
