Fast Beamforming Design via Deep Learning

Abstract

Beamforming is considered as one of the most important techniques for designing advanced multiple-input and multiple-output (MIMO) systems. Among existing design criterions, sum rate maximization (SRM) under a total power constraint is a challenge due to its nonconvexity. Existing techniques for the SRM problem only obtain local optimal solutions but require huge amount of computation due to their complex matrix operations and iterations. Unlike these conventional methods, we propose a deep learning based fast beamforming design method without complex operations and iterations. Specifically, we first derive a heuristic solution structure of the downlink beamforming through the virtual equivalent uplink channel based on optimum MMSE receiver which separates the problem into power allocation and virtual uplink beamforming (VUB) design. Next, beamforming prediction network (BPNet) is designed to perform the joint optimization of power allocation and VUB design. Moreover, the BPNet is trained offline using two-step training strategy. Simulation results demonstrate that our proposed method is fast while obtains the comparable performance to the state-of-the-art method.