Deep Expectation-Maximization for Joint MIMO Channel Estimation and Signal Detection

Abstract

To overcome the influence of channel estimation error on signal detection, this paper presents a model-driven deep learning method for joint channel estimation and signal detection in multiple-input multiple-output (MIMO) wireless communication systems. To improve the robustness of the method, we use the Student's <italic xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">t</i> -distribution to model the environment noise, and model the joint problem probabilistically by taking the channel state information (CSI) as a latent variable, then derive a generalized expectation maximization (GEM) algorithm to fit the model. In GEM, the expectation step is used for robust CSI estimation while the maximization step detects the transmitted signal with the estimated CSI. To reduce the computational complexity of GEM, we unfold it into a deep neural network. The network contains only a few trainable parameters, which is easy to train and has low space complexity. Based on our experimental results, we find that the proposed GEM algorithm outperforms a variety of signal detection algorithms which take the CSI estimation and signal detection as independent modules. Further, we find empirically that the proposed GEM network outperforms state-of-the-art deep learning based joint channel estimation and signal detection methods and has a good generalization ability against the number of antennas, the modulation mode, the level of signal-to-noise ratio and channel correlation.