Learned Conjugate Gradient Descent Network for Massive MIMO Detection

Yi Wei, Ming Min Zhao, Mingyi Hong, Min Jian Zhao, Ming Lei

Research output: Chapter in Book/Report/Conference proceedingConference contribution

1 Scopus citations

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.

Original languageEnglish (US)
Title of host publication2020 IEEE International Conference on Communications, ICC 2020 - Proceedings
PublisherInstitute of Electrical and Electronics Engineers Inc.
ISBN (Electronic)9781728150895
DOIs
StatePublished - Jun 2020
Event2020 IEEE International Conference on Communications, ICC 2020 - Dublin, Ireland
Duration: Jun 7 2020Jun 11 2020

Publication series

NameIEEE International Conference on Communications
Volume2020-June
ISSN (Print)1550-3607

Conference

Conference2020 IEEE International Conference on Communications, ICC 2020
CountryIreland
CityDublin
Period6/7/206/11/20

Bibliographical note

Funding Information:
This work was supported in part by the National Natural Science Foundation of China under Grant 91938202, in part by the Zhejiang Provincial Natural Science Foundation of China under Grant LQ20F010010, in part by the Fundamental Research Funds for the Central Universities under Grant 2019QNA5011, and in part by the National Key Research and Development Project under grant 2018YFB1802303.

Keywords

  • Conjugate gradient descent
  • deep learning
  • massive MIMO detection
  • model-driven method

Fingerprint Dive into the research topics of 'Learned Conjugate Gradient Descent Network for Massive MIMO Detection'. Together they form a unique fingerprint.

Cite this