Improving Massive MIMO Message Passing Detectors With Deep Neural Network

In this paper, deep neural network (DNN) is utilized to improve message passing detectors (MPDs) for massive multiple-input multiple-output (MIMO) systems. A general framework to construct DNN architecture for MIMO detection is first introduced by unfolding iterative MPDs. DNN MIMO detectors are then proposed based on modified MPDs including damped belief propagation (BP), max-sum (MS) BP, and simplified channel hardening-exploiting message passing (CHEMP). The correction factors are optimized via deep learning for better performance. Numerical results demonstrate that, compared with state-of-the-art (SOA) detectors including minimum mean-squared error (MMSE), BP, and CHEMP, the proposed DNN detectors can achieve better bit-error-rate (BER) and improve robustness against various antenna and channel conditions with similar complexity. The DNN is required to be trained only once and can be reused for multiple detections, which assures its high efficiency. The corresponding hardware architecture is also proposed. Implementation results with 65 nm CMOS technology approve the efficiency and flexibility of the proposed DNN framework.

[1]  Warren J. Gross,et al.  Neural offset min-sum decoding , 2017, 2017 IEEE International Symposium on Information Theory (ISIT).

[2]  Wei Tang,et al.  A 0.58mm2 2.76Gb/s 79.8pJ/b 256-QAM massive MIMO message-passing detector , 2016, 2016 IEEE Symposium on VLSI Circuits (VLSI-Circuits).

[3]  Yann LeCun,et al.  Learning Fast Approximations of Sparse Coding , 2010, ICML.

[4]  Yair Be'ery,et al.  Learning to decode linear codes using deep learning , 2016, 2016 54th Annual Allerton Conference on Communication, Control, and Computing (Allerton).

[5]  Xiaohu You,et al.  Improved polar decoder based on deep learning , 2017, 2017 IEEE International Workshop on Signal Processing Systems (SiPS).

[6]  Biing-Hwang Juang,et al.  Deep Learning in Physical Layer Communications , 2018, IEEE Wireless Communications.

[7]  Xiaohu You,et al.  Efficient architecture for soft-output massive MIMO detection with Gauss-Seidel method , 2016, 2016 IEEE International Symposium on Circuits and Systems (ISCAS).

[8]  Ali Mousavi,et al.  Learning to invert: Signal recovery via Deep Convolutional Networks , 2017, 2017 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP).

[9]  Martín Abadi,et al.  TensorFlow: Large-Scale Machine Learning on Heterogeneous Distributed Systems , 2016, ArXiv.

[10]  Xiaohu You,et al.  Belief propagation detection based on max-sum algorithm for massive MIMO systems , 2017, 2017 9th International Conference on Wireless Communications and Signal Processing (WCSP).

[11]  Geoffrey Ye Li,et al.  A Model-Driven Deep Learning Network for MIMO Detection , 2018, 2018 IEEE Global Conference on Signal and Information Processing (GlobalSIP).

[12]  Yahong Rosa Zheng,et al.  Brain-Inspired Wireless Communications: Where Reservoir Computing Meets MIMO-OFDM , 2018, IEEE Transactions on Neural Networks and Learning Systems.

[13]  Peng Li,et al.  Multiple output selection-LAS algorithm in large MIMO systems , 2010, IEEE Communications Letters.

[14]  Geoffrey Ye Li,et al.  Model-Driven Deep Learning for Physical Layer Communications , 2018, IEEE Wireless Communications.

[15]  B. Sundar Rajan,et al.  Low-complexity near-ML decoding of large non-orthogonal STBCs using reactive tabu search , 2009, 2009 IEEE International Symposium on Information Theory.

[16]  Claus-Peter Schnorr,et al.  Lattice basis reduction: Improved practical algorithms and solving subset sum problems , 1991, FCT.

[17]  Yun Zhang,et al.  Virtual MIMO blind detection clustered WSN system , 2015, 2015 Asia-Pacific Microwave Conference (APMC).

[18]  Geoffrey Ye Li,et al.  Power of Deep Learning for Channel Estimation and Signal Detection in OFDM Systems , 2017, IEEE Wireless Communications Letters.

[19]  Xiaohu You,et al.  Low-Complexity Belief Propagation Detection for Correlated Large-Scale MIMO Systems , 2017, Journal of Signal Processing Systems.

[20]  Zhu Han,et al.  Machine Learning Paradigms for Next-Generation Wireless Networks , 2017, IEEE Wireless Communications.

[21]  Shouyi Yin,et al.  A 1.58 Gbps/W 0.40 Gbps/mm2 ASIC Implementation of MMSE Detection for $128\times 8~64$ -QAM Massive MIMO in 65 nm CMOS , 2018, IEEE Transactions on Circuits and Systems I: Regular Papers.

[22]  Thomas L. Marzetta,et al.  Noncooperative Cellular Wireless with Unlimited Numbers of Base Station Antennas , 2010, IEEE Transactions on Wireless Communications.

[23]  Philip Schniter,et al.  Onsager-corrected deep learning for sparse linear inverse problems , 2016, 2016 IEEE Global Conference on Signal and Information Processing (GlobalSIP).

[24]  Liang Liu,et al.  3.6 A 60pJ/b 300Mb/s 128×8 Massive MIMO precoder-detector in 28nm FD-SOI , 2017, 2017 IEEE International Solid-State Circuits Conference (ISSCC).

[25]  Toshihiko Nishimura,et al.  Low-Complexity Detection Based on Belief Propagation in a Massive MIMO System , 2013, 2013 IEEE 77th Vehicular Technology Conference (VTC Spring).

[26]  Xiaohu You,et al.  Improved symbol-based belief propagation detection for large-scale MIMO , 2015, 2015 IEEE Workshop on Signal Processing Systems (SiPS).

[27]  Xiaohu You,et al.  LOW-COMPLEXITY MESSAGE PASSING MIMO DETECTION ALGORITHM WITH DEEP NEURAL NETWORK , 2018, 2018 IEEE Global Conference on Signal and Information Processing (GlobalSIP).

[28]  Erik G. Larsson,et al.  Massive MIMO for next generation wireless systems , 2013, IEEE Communications Magazine.

[29]  Jian Sun,et al.  Deep Residual Learning for Image Recognition , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[30]  Chia-Hsiang Yang,et al.  An Integrated Message-Passing Detector and Decoder for Polar-Coded Massive MU-MIMO Systems , 2019, IEEE Transactions on Circuits and Systems I: Regular Papers.

[31]  Erik G. Larsson,et al.  Scaling Up MIMO: Opportunities and Challenges with Very Large Arrays , 2012, IEEE Signal Process. Mag..

[32]  Zhongfeng Wang,et al.  Reduced complexity message passing detection algorithm in large-scale MIMO systems , 2017, 2017 9th International Conference on Wireless Communications and Signal Processing (WCSP).

[33]  Demis Hassabis,et al.  Mastering the game of Go without human knowledge , 2017, Nature.

[34]  Mohsen Guizani,et al.  Deep neural network-aided Gaussian message passing detection for ultra-reliable low-latency communications , 2019, Future Gener. Comput. Syst..

[35]  Tara N. Sainath,et al.  Deep Neural Networks for Acoustic Modeling in Speech Recognition: The Shared Views of Four Research Groups , 2012, IEEE Signal Processing Magazine.

[36]  Kiran Karra,et al.  Learning to communicate: Channel auto-encoders, domain specific regularizers, and attention , 2016, 2016 IEEE International Symposium on Signal Processing and Information Technology (ISSPIT).

[37]  Ananthanarayanan Chockalingam,et al.  Channel Hardening-Exploiting Message Passing (CHEMP) Receiver in Large-Scale MIMO Systems , 2013, IEEE Journal of Selected Topics in Signal Processing.

[38]  Xiaojun Yuan,et al.  Achievable Rates of MIMO Systems With Linear Precoding and Iterative LMMSE Detection , 2011, IEEE Transactions on Information Theory.

[39]  Michael I. Jordan,et al.  Loopy Belief Propagation for Approximate Inference: An Empirical Study , 1999, UAI.

[40]  Arian Maleki,et al.  Optimality of large MIMO detection via approximate message passing , 2015, 2015 IEEE International Symposium on Information Theory (ISIT).

[41]  Yik-Chung Wu,et al.  On Convergence Conditions of Gaussian Belief Propagation , 2015, IEEE Transactions on Signal Processing.

[42]  Jimmy Ba,et al.  Adam: A Method for Stochastic Optimization , 2014, ICLR.

[43]  Ami Wiesel,et al.  Deep MIMO detection , 2017, 2017 IEEE 18th International Workshop on Signal Processing Advances in Wireless Communications (SPAWC).

[44]  Abhay Kumar Sah,et al.  An MMP-Based Approach for Detection in Large MIMO Systems Using Sphere Decoding , 2017, IEEE Wireless Communications Letters.