一种基于图神经网络的多用户波束分配方法及系统

本发明公开了一种基于图神经网络的多用户波束分配方法及系统，该方法包括：对多用户分布式网络进行信道估计，获取信道矩阵并构建分布式多用户通信系统图结构；根据分布式多用户通信系统图结构构建图神经网络模型；构建虚拟仿真环境生成毫米波分布式多用户通信系统信道数据集并对图神经网络模型进行训练，得到训练后的图神经网络模型；基于训练后的图神经网络模型对信道矩阵进行多用户波束分配，得到最优多用户波束分配方案。本发明能够利用图神经网络实现最大化系统可达和速率，减少了波束分配问题的计算开销，同时具有较高的泛化性和鲁棒性。本发明作为一种基于图神经网络的多用户波束分配方法及系统，可广泛应用于毫米波通讯技术领域。