HPNet: A Compressed Neural Network for Robust Hybrid Precoding in Multi-User Massive MIMO Systems

Abstract

In multi-user millimeter wave (mmWave) communications, massive multiple-input multiple-output (MIMO) systems can achieve high gain and spectral efficiency significantly. To reduce the hardware complexity and energy consumption of massive MIMO systems, hybrid precoding as a crucial technique has attracted extensive attention. Most previous works for hybrid precoding developed algorithms based on optimization or exhaustive search approaches that either lead to sub-optimal performance or have high computational complexity. Motivated by the thought of cross-fertilization between Data-driven and Model-driven approaches, we consider applying deep learning approach and introduce the Hybrid Precoding Network(HPNet), which is a compressed deep neural network exploiting the feature extracting (thanks to convolutional kernels) and generalization ability of neural networks and the natural sparsity of mmWave channels. The HPNet takes imperfect channel state information (CSI) as the input and predicts the analog precoder and baseband precoder for multi-user massive MIMO systems. Moreover, in order to make the approach more practical in real scenarios, we further introduce a model compression algorithm, using network pruning, to greatly reduce the computational complexity and memory usage of the neural network while almost retaining the model performance and then assess the influence of pruned parameters in the network. Numerical experiments demonstrate that HPNet outperforms state-of-the-art hybrid precoding schemes with higher performance and stronger robustness. Finally, we analyze and compare the computational complexity of different schemes.