Learned Conjugate Gradient Descent Network for Massive MIMO Detection

Abstract

In this work, we consider the use of model-driven deep learning (DL) techniques for signal detection in massive multiple-input multiple-output (MIMO) system. Massive MIMO promises improved spectral efficiency, coverage and reliability, compared to conventional MIMO systems. Unfortunately, these benefits usually come at the cost of significantly increased computational complexity. To address this difficulty, a learned conjugate gradient descent network, referred to as LcgNet, is presented by unfolding the iterative conjugate gradient descent (CG) detector. In the proposed network, instead of calculating the exact values of the scalar step-sizes for every problem instance, we explicitly learn their universal values. We show that the performance of the proposed network can be greatly improved by augmenting the dimensions of these step-sizes. Furthermore, due to the limited learnable parameters to be optimized, the proposed networks are easy and fast to train. Numerical results demonstrate that this approach can achieve superior performance over some state-of-the-art MIMO detectors such as the CG detector, the linear minimum mean squared error (LMMSE) detector etc., with much lower computational complexity.