MDL-AltMin: A Hybrid Precoding Scheme for mmWave Systems With Deep Learning and Alternate Optimization

The hybrid precoding structure composed of analog and digital precoders has received increasing attention in mmWave massive multiple-input multiple-output (MIMO) systems because it can balance the energy consumption and spectral efficiency (SE). However, it is challenging to obtain the optimal hybrid precoding scheme by joint optimization with lower computational complexity. This letter proposes a hybrid precoding scheme based on model-driven deep learning and alternate minimization (MDL-AltMin), which is implemented by alternately solving analog precoder and digital precoder. During the alternation, we design an analog precoding network (AP-Net) to solve the phase shift network in analog precoder with the goal of maximizing SE. The digital precoder is solved by the Lagrange multiplier method. In each alternate optimization process, the criteria for convergence is to minimize the error between the hybrid precoder and the optimal fully digital precoder. The simulation results show that the SE of our proposed scheme is very close to the fully digital precoding scheme based on singular value decomposition with lower computational complexity.