Mixture-of-Experts Diffusion Models for Adaptive Massive MIMO Channel Estimation via Variational Bayesian Inference

Abstract

Channel estimation is essential to massive multiple-input multiple-output (MIMO) systems. While recent generative model-based approaches using lightweight diffusion models (DMs) have achieved superior performance, they typically rely on a single data-driven prior, which limits their adaptability to varying channel distributions in real-world scenarios. To address this deficiency, we propose a mixture-of-experts (MoE) diffusion model (DM) framework combined with variational Bayesian inference. Specifically, our approach employs multiple pre-trained DMs, with each trained on a specific type of propagation channels. We then propose a probabilistic graphical model in which the channel is modeled as a latent variable drawn from one of these candidate generative priors with a certain probability. By integrating variational Bayesian inference with DM-based data priors, the underlying channel along with the expert indicator variable are jointly inferred, thus enabling automatic model adaptation for channel estimation. The effectiveness of our approach is evaluated on 3GPP CDL channels. Simulation results demonstrate that our proposed approach achieves a clear performance improvement over the standard DM-based method that employs a single prior trained on aggregated data from all channel types, particularly when the channel samples from different propagation environments are imbalanced.