Deep Learning for Joint MIMO Detection and Channel Decoding

We propose a deep-learning approach for the joint MIMO detection and channel decoding problem. Conventional MIMO receivers adopt a model-based approach for MIMO detection and channel decoding in linear or iterative manners. However, due to the complex MIMO signal model, the optimal solution to the joint MIMO detection and channel decoding problem (i.e., the maximum likelihood decoding of the transmitted codewords from the received MIMO signals) is computationally infeasible. As a practical measure, the current model-based MIMO receivers all use suboptimal MIMO decoding methods with affordable computational complexities. This work applies the latest advances in deep learning for the design of MIMO receivers. In particular, we leverage deep neural networks (DNN) with supervised training to solve the joint MIMO detection and channel decoding problem. We show that DNN can be trained to give much better decoding performance than conventional MIMO receivers do. Our simulations show that a DNN implementation consisting of seven hidden layers can outperform conventional model-based linear or iterative receivers. This performance improvement points to a new direction for future MIMO receiver design.

[1]  Stephan ten Brink,et al.  Scaling Deep Learning-Based Decoding of Polar Codes via Partitioning , 2017, GLOBECOM 2017 - 2017 IEEE Global Communications Conference.

[2]  Juyul Lee,et al.  Decoding of Polar Code by Using Deep Feed-Forward Neural Networks , 2018, 2018 International Conference on Computing, Networking and Communications (ICNC).

[3]  John S. Thompson,et al.  Fixing the Complexity of the Sphere Decoder for MIMO Detection , 2008, IEEE Transactions on Wireless Communications.

[4]  Alex B. Gershman,et al.  Training-based MIMO channel estimation: a study of estimator tradeoffs and optimal training signals , 2006, IEEE Transactions on Signal Processing.

[5]  Rainer Leupers,et al.  A Scalable VLSI Architecture for Soft-Input Soft-Output Single Tree-Search Sphere Decoding , 2009, IEEE Transactions on Circuits and Systems II: Express Briefs.

[6]  Tao Jiang,et al.  Deep learning for wireless physical layer: Opportunities and challenges , 2017, China Communications.

[7]  Alexander Vardy,et al.  Closest point search in lattices , 2002, IEEE Trans. Inf. Theory.

[8]  Robert F. H. Fischer,et al.  Low-complexity near-maximum-likelihood detection and precoding for MIMO systems using lattice reduction , 2003, Proceedings 2003 IEEE Information Theory Workshop (Cat. No.03EX674).

[9]  Volker Jungnickel,et al.  Performance of MIMO systems with channel inversion , 2002, Vehicular Technology Conference. IEEE 55th Vehicular Technology Conference. VTC Spring 2002 (Cat. No.02CH37367).

[10]  David Burshtein,et al.  Deep Learning Methods for Improved Decoding of Linear Codes , 2017, IEEE Journal of Selected Topics in Signal Processing.

[11]  Stephan ten Brink,et al.  Achieving near-capacity on a multiple-antenna channel , 2003, IEEE Trans. Commun..

[12]  Helmut Bölcskei,et al.  Soft–Input Soft–Output Single Tree-Search Sphere Decoding , 2009, IEEE Transactions on Information Theory.

[13]  Andrea J. Goldsmith,et al.  Capacity limits of MIMO channels , 2003, IEEE J. Sel. Areas Commun..

[14]  Shrinivas Kudekar,et al.  Design of Low-Density Parity Check Codes for 5G New Radio , 2018, IEEE Communications Magazine.

[15]  Aria Nosratinia,et al.  Outage and Diversity of Linear Receivers in Flat-Fading MIMO Channels , 2007, IEEE Transactions on Signal Processing.

[16]  Geoffrey Ye Li,et al.  Initial Results on Deep Learning for Joint Channel Equalization and Decoding , 2017, 2017 IEEE 86th Vehicular Technology Conference (VTC-Fall).

[17]  Edward W. Knightly,et al.  IEEE 802.11ac: from channelization to multi-user MIMO , 2013, IEEE Communications Magazine.

[18]  Erik G. Larsson,et al.  Soft MIMO Detection at Fixed Complexity , 2007, IEEE GLOBECOM 2007 - IEEE Global Telecommunications Conference.

[19]  Amitabha Ghosh,et al.  Essentials of LTE and LTE-A by Amitabha Ghosh , 2011 .

[20]  Erik G. Larsson,et al.  Fixed-Complexity Soft MIMO Detection via Partial Marginalization , 2008, IEEE Transactions on Signal Processing.

[21]  Guigang Zhang,et al.  Deep Learning , 2016, Int. J. Semantic Comput..

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

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

[24]  Giuseppe Caire,et al.  Asymptotic Performance of Linear Receivers in MIMO Fading Channels , 2008, IEEE Transactions on Information Theory.

[25]  Emanuele Viterbo,et al.  A universal lattice code decoder for fading channels , 1999, IEEE Trans. Inf. Theory.

[26]  E.G. Larsson,et al.  MIMO Detection Methods: How They Work [Lecture Notes] , 2009, IEEE Signal Processing Magazine.

[27]  Erdal Arikan,et al.  Channel Polarization: A Method for Constructing Capacity-Achieving Codes for Symmetric Binary-Input Memoryless Channels , 2008, IEEE Transactions on Information Theory.

[28]  Stephan ten Brink,et al.  On deep learning-based channel decoding , 2017, 2017 51st Annual Conference on Information Sciences and Systems (CISS).

[29]  Rainer Leupers,et al.  A Scalable VLSI Architecture for Soft-Input Soft-Output Depth-First Sphere Decoding , 2009, ArXiv.

[30]  Zhi-Quan Luo,et al.  Soft quasi-maximum-likelihood detection for multiple-antenna wireless channels , 2003, IEEE Trans. Signal Process..