Distributed Learning-Based Beamforming Codebooks for Unevenly Distributed Users in mmWave Massive MIMO System

Millimeter wave (mmWave) massive multiple-input multiple-output (MIMO) technology represents a promising technology in wireless communication. This technology relies on beamforming codebooks for initial access and transmission. However, conventional codebooks comprise a multitude of single-lobe narrow beams, resulting in redundant beams that may never be utilized in beam training. While centralized machine learning methods can partially address the concern of redundancy, they tend to overlook the presence of minority users scattered across diverse regions. The equitable coverage of environmental adaptive codebooks depends on addressing this issue. Hence, we devise a distributed learning (DL) framework for codebook design, which is tailored for scenarios with uneven user distribution and fully exploits the decentralized and online learning features of DL. Our approach begins by segmenting the user channels into various subsets through a pre-classification process. Then, we introduce a novel DL architecture designed to process the subsets that are assigned to individual user equipments (UEs). Each UE then generates a phase shift matrix that contributes to the concatenation-based global aggregation in the base station. The simulation results confirm the effectiveness of DL in improving the performance of mmWave massive MIMO systems in scenarios with unevenly distributed users.