Mar, 2024

微型图神经网络用于无线资源管理

TL;DR无线资源管理的需求增加了对高效无线资源管理的精简神经网络结构的开发的需求。本文介绍了一种针对无线资源管理量身定制的图神经网络(GNN)的新方法:低秩消息传递图神经网络(LR-MPGNN)。该方法基于低秩逼近技术,将传统线性层替换为其低秩对应层,大大减小了模型尺寸和参数个数。通过多个关键指标对提出的 LR-MPGNN 模型的性能进行评估,包括模型尺寸、参数个数、通信系统的加权和速率以及权重矩阵的特征值分布。广泛的评估结果显示,LR-MPGNN 模型的模型尺寸减小了 60 倍,模型参数个数可以减少高达 98%。在性能方面,相对于原始 MPGNN 模型,LR-MPGNN 模型在最佳情况下的规范加权和速率仅降低了 2%,表现出很好的鲁棒性。此外,LR-MPGNN 模型的权重矩阵的特征值分布更加均匀,范围更广,表明了权重的战略性重新分布。