Robust WMMSE Precoder with Deep Learning Design for Massive MIMO

In this paper, we investigate the downlink robust precoding with imperfect channel state information (CSI) for massive multiple-input-multiple-output (MIMO) communications. With the estimated channel and channel error statistics, the general design of the robust precoder is to maximize the ergodic sum rate subject to the total transmit power constraint. To make the problem more tractable, we find a lower bound of the ergodic sum rate and propose the robust weighted minimum mean-squared-error (WMMSE) precoder to maximize the bound. We characterize the structure of the precoding vectors by low-dimensional parameters, which are learned directly from the available CSI through a neural network. As such, the precoding vectors can be immediately computed without iterations. To extend the deep learning design to multi-antennas users, we present a flexible approach that allows the various antenna configurations at the user side to be handled. Simulation results show that the deep learning design can significantly reduce the computational complexity compared with the existing precoder designs while achieving near optimal performance.