FusionNet: Enhanced Beam Prediction for mmWave Communications Using Sub-6 GHz Channel and a Few Pilots

In order to reduce the downlink training overhead of mmWave communications, we propose a novel downlink beamforming strategy using the uplink sub-6GHz channel and downlink mmWave pilots that are sent from a few active antennas. Specifically, we design a novel dual-input neural network architecture, called FusionNet, to merge the sub-6GHz channel and the channel of a few active mmWave antennas. The proposed fusion model could intelligently adjust the attention paid (by the neural network) for sub-6GHz channel and mmWave channel by an attention mechanism. The output of the FusionNet represents the probability for each beam being the optimal one. We also propose an antenna selection model that can choose better active antennas to send the downlink pilots, in which the gradient of antenna selection vector is approximated by that of an antenna probability vector. Simulation results demonstrate the superior performance of the proposed strategy compared to the existing one that purely relies on the sub-6GHz information or compared to the shallow model that directly adds uniform pilots.