Improving Massive MIMO Belief Propagation Detector with Deep Neural Network

In this paper, deep neural network (DNN) is utilized to improve the belief propagation (BP) detection for massive multiple-input multiple-output (MIMO) systems. A neural network architecture suitable for detection task is firstly introduced by unfolding BP algorithms. DNN MIMO detectors are then proposed based on two modified BP detectors, damped BP and max-sum BP. The correction factors in these algorithms are optimized through deep learning techniques, aiming at improved detection performance. Numerical results are presented to demonstrate the performance of the DNN detectors in comparison with various BP modifications. The neural network is trained once and can be used for multiple online detections. The results show that, compared to other state-of-the-art detectors, the DNN detectors can achieve lower bit error rate (BER) with improved robustness against various antenna configurations and channel conditions at the same level of complexity.

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

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

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

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

[5]  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.

[6]  Bin Liu,et al.  Signal detection of MIMO-OFDM system based on auto encoder and extreme learning machine , 2017, 2017 International Joint Conference on Neural Networks (IJCNN).

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

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

[9]  Chao-Kai Wen,et al.  Fixed-point Implementation of Approximate Message Passing (AMP) algorithm in massive MIMO systems , 2016 .

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

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

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

[13]  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).

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

[15]  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).

[16]  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).

[17]  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.

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

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

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

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

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

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

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

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

[26]  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).

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

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

[29]  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).