Deep Learning Aided Low Complex Sphere Decoding for MIMO Detection

Jieyu Liao,Junhui Zhao,Feifei Gao,Geoffrey Ye Li

DOI: https://doi.org/10.1109/tcomm.2022.3218630

IF: 6.166

2022-01-01

IEEE Transactions on Communications

Abstract:In this paper, we propose a deep learning based sphere decoding (SD) scheme to reduce the detection complexity for the multiple-input multiple-output (MIMO) communication systems. Specifically, we first design the sparsely connected deep neural network (SC-DNN) to find a moderate radius for the SD algorithm. Then, we develop the SC-SD algorithm to reduce the computational complexity by deciding the detection order from the output of the SC-DNN, the zero-forcing (ZF) detector, and the transmit power. We further reduce the complexity of the SC-SD by defining partial layers without searching. For multi-stream MIMO, where a large number of parameters in neural networks should be trained, we propose a partitioned training procedure to achieve a reasonable computational complexity. Simulation results demonstrate that the SC-SD almost achieves the performance of the maximum likelihood (ML) in MIMO system but is much faster than the classic SD algorithm.