# MIMO Channel Estimation using Score-Based Generative Models

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

## Key facts

- Cited by 50 papers in the map (data through 14 Oct 2026).
- Placed in Diffusion for compression and wireless, one of 14 directions in Diffusion models.
- Builds on 28 of its 70 references that are in the map.
- First posted on arXiv 14 Apr 2022.

## Placed in

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

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

- [Adam: A Method for Stochastic Optimization](https://mlsanity.com/paper/1412.6980/) (2014)
- [PyTorch: An Imperative Style, High-Performance Deep Learning Library](https://mlsanity.com/paper/1912.01703/) (2019)
- [U-Net: Convolutional Networks for Biomedical Image Segmentation](https://mlsanity.com/paper/1505.04597/) (2015)
- [Diffusion Models Beat GANs on Image Synthesis](https://mlsanity.com/paper/2105.05233/) (2021)
- [Generative Modeling by Estimating Gradients of the Data Distribution](https://mlsanity.com/paper/1907.05600/) (2019)
- [Improved Training of Wasserstein GANs](https://mlsanity.com/paper/1704.00028/) (2017)
- [WILDS: A Benchmark of in-the-Wild Distribution Shifts](https://mlsanity.com/paper/2012.07421/) (2020)
- [Beyond a Gaussian Denoiser: Residual Learning of Deep CNN for Image Denoising](https://mlsanity.com/paper/1608.03981/) (2016)
- [Improved Techniques for Training Score-Based Generative Models](https://mlsanity.com/paper/2006.09011/) (2020)
- [Maximum Likelihood Training of Score-Based Diffusion Models](https://mlsanity.com/paper/2101.09258/) (2021)
- [Solving Inverse Problems in Medical Imaging with Score-Based Generative Models](https://mlsanity.com/paper/2111.08005/) (2021)
- [RefineNet: Multi-Path Refinement Networks for High-Resolution Semantic Segmentation](https://mlsanity.com/paper/1611.06612/) (2016)
- [Compressed Sensing using Generative Models](https://mlsanity.com/paper/1703.03208/) (2017)
- [Come-Closer-Diffuse-Faster: Accelerating Conditional Diffusion Models for Inverse Problems through Stochastic Contraction](https://mlsanity.com/paper/2112.05146/) (2021)
- [Robust Compressed Sensing MRI with Deep Generative Priors](https://mlsanity.com/paper/2108.01368/) (2021)
- [Statistical Aspects of Wasserstein Distances](https://mlsanity.com/paper/1806.05500/) (2018)
- [The Roadmap to 6G -- AI Empowered Wireless Networks](https://mlsanity.com/paper/1904.11686/) (2019)
- [Deep Learning-Based Channel Estimation](https://mlsanity.com/paper/1810.05893/) (2018)
- [Intermediate Layer Optimization for Inverse Problems using Deep Generative Models](https://mlsanity.com/paper/2102.07364/) (2021)
- [Learned D-AMP: Principled Neural Network based Compressive Image Recovery](https://mlsanity.com/paper/1704.06625/) (2017)
- [Instance-Optimal Compressed Sensing via Posterior Sampling](https://mlsanity.com/paper/2106.11438/) (2021)
- [High Dimensional Channel Estimation Using Deep Generative Networks](https://mlsanity.com/paper/2006.13494/) (2020)
- [Approximating the Void: Learning Stochastic Channel Models from Observation with Variational Generative Adversarial Networks](https://mlsanity.com/paper/1805.06350/) (2018)
- [A Survey on Channel Estimation and Practical Passive Beamforming Design for Intelligent Reflecting Surface Aided Wireless Communications](https://mlsanity.com/paper/2110.01292/) (2021)
- [Dynamic Metasurface Antennas for 6G Extreme Massive MIMO Communications](https://mlsanity.com/paper/2006.07838/) (2020)
- [Intelligent Reflecting Surface-Enhanced OFDM: Channel Estimation and Reflection Optimization](https://mlsanity.com/paper/1909.03272/) (2019)
- [An Optimal Channel Estimation Scheme for Intelligent Reflecting Surfaces based on a Minimum Variance Unbiased Estimator](https://mlsanity.com/paper/1909.09440/) (2019)
- [Tracking mm-Wave Channel Dynamics: Fast Beam Training Strategies under Mobility](https://mlsanity.com/paper/1612.07957/) (2016)

## Built on by (most cited first, 30 of 50)

- [Enhancing Deep Reinforcement Learning: A Tutorial on Generative Diffusion Models in Network Optimization](https://mlsanity.com/paper/2308.05384/) (2023)
- [Solving Inverse Problems with Score-Based Generative Priors learned from Noisy Data](https://mlsanity.com/paper/2305.01166/) (2023)
- [Generative Diffusion Models for High Dimensional Channel Estimation](https://mlsanity.com/paper/2408.10501/) (2024)
- [Diffusion-based Generative Prior for Low-Complexity MIMO Channel Estimation](https://mlsanity.com/paper/2403.03545/) (2024)
- [Generative AI for Unmanned Vehicle Swarms: Challenges, Applications and Opportunities](https://mlsanity.com/paper/2402.18062/) (2024)
- [Generating High Dimensional User-Specific Wireless Channels using Diffusion Models](https://mlsanity.com/paper/2409.03924/) (2024)
- [Generative Diffusion Models for Wireless Networks: Fundamental, Architecture, and State-of-the-Art](https://mlsanity.com/paper/2507.16733/) (2025)
- [Conditional Denoising Diffusion Probabilistic Models for Data Reconstruction Enhancement in Wireless Communications](https://mlsanity.com/paper/2310.19460/) (2023)
- [On the Asymptotic Mean Square Error Optimality of Diffusion Models](https://mlsanity.com/paper/2403.02957/) (2024)
- [Generative Artificial Intelligence Assisted Wireless Sensing: Human Flow Detection in Practical Communication Environments](https://mlsanity.com/paper/2404.14140/) (2024)
- [Applications of Generative AI (GAI) for Mobile and Wireless Networking: A Survey](https://mlsanity.com/paper/2405.20024/) (2024)
- [Score-based Source Separation with Applications to Digital Communication Signals](https://mlsanity.com/paper/2306.14411/) (2023)
- [Blind Super-Resolution via Meta-learning and Markov Chain Monte Carlo Simulation](https://mlsanity.com/paper/2406.08896/) (2024)
- [Solving Linear Inverse Problems using Higher-Order Annealed Langevin Diffusion](https://mlsanity.com/paper/2305.05014/) (2023)
- [Leveraging Variational Autoencoders for Parameterized MMSE Estimation](https://mlsanity.com/paper/2307.05352/) (2023)
- [Diffusion Models for Wireless Communications](https://mlsanity.com/paper/2310.07312/) (2023)
- [GDM4MMIMO: Generative Diffusion Models for Massive MIMO Communications](https://mlsanity.com/paper/2412.18281/) (2024)
- [Diffusion Models for Future Networks and Communications: A Comprehensive Survey](https://mlsanity.com/paper/2508.01586/) (2025)
- [Diffusion Models for Accurate Channel Distribution Generation](https://mlsanity.com/paper/2309.10505/) (2023)
- [Probabilistic Constellation Shaping With Denoising Diffusion Probabilistic Models: A Novel Approach](https://mlsanity.com/paper/2309.08688/) (2023)
- [Generative Diffusion Receivers: Achieving Pilot-Efficient MIMO-OFDM Communications](https://mlsanity.com/paper/2506.18419/) (2025)
- [Environment-Aware Channel Inference via Cross-Modal Flow: From Multimodal Sensing to Wireless Channels](https://mlsanity.com/paper/2512.04966/) (2025)
- [Adversarial Score-Based Generative Models for MMSE-achieving AmBC Channel Estimation](https://mlsanity.com/paper/2309.05776/) (2023)
- [Accurate and Fast Channel Estimation for Fluid Antenna Systems with Diffusion Models](https://mlsanity.com/paper/2505.04930/) (2025)
- [Enhancements in Score-based Channel Estimation for Real-Time Wireless Systems](https://mlsanity.com/paper/2509.07839/) (2025)
- [Generalization Bounds for Transformer Channel Decoders](https://mlsanity.com/paper/2601.06969/) (2026)
- [Constrained Diffusion Models with Primal-Dual Inference](https://mlsanity.com/paper/2606.17192/) (2026)
- [Non-Identical Diffusion Models in MIMO-OFDM Channel Generation](https://mlsanity.com/paper/2509.01641/) (2025)
- [Robust MIMO Channel Estimation Using Energy-Based Generative Diffusion Models](https://mlsanity.com/paper/2510.22230/) (2025)
- [Diffusion Models for Wireless Transceivers: From Pilot-Efficient Channel Estimation to AI-Native 6G Receivers](https://mlsanity.com/paper/2510.24495/) (2025)

## Abstract (arXiv)

Channel estimation is a critical task in multiple-input multiple-output (MIMO) digital communications that substantially effects end-to-end system performance. In this work, we introduce a novel approach for channel estimation using deep score-based generative models. A model is trained to estimate the gradient of the logarithm of a distribution and is used to iteratively refine estimates given measurements of a signal. We introduce a framework for training score-based generative models for wireless MIMO channels and performing channel estimation based on posterior sampling at test time. We derive theoretical robustness guarantees for channel estimation with posterior sampling in single-input single-output scenarios, and experimentally verify performance in the MIMO setting. Our results in simulated channels show competitive in-distribution performance, and robust out-of-distribution performance, with gains of up to 5 dB in end-to-end coded communication performance compared to supervised deep learning methods. Simulations on the number of pilots show that high fidelity channel estimation with 25% pilot density is possible for MIMO channel sizes of up to 64 × 256. Complexity analysis reveals that model size can efficiently trade performance for estimation latency, and that the proposed approach is competitive with compressed sensing in terms of floating-point operation (FLOP) count.