论文标题

可扩展无线电资源管理的图形神经网络:体系结构设计和理论分析

Graph Neural Networks for Scalable Radio Resource Management: Architecture Design and Theoretical Analysis

论文作者

Shen, Yifei, Shi, Yuanming, Zhang, Jun, Letaief, Khaled B.

论文摘要

最近,深度学习已成为一种破坏性的技术,可以解决无线网络中挑战性的无线电资源管理问题。但是,现有作品采用的神经网络体系结构遭受了可扩展性,概括和缺乏解释性的差异。提高可扩展性和概括的长期方法是将目标任务的结构纳入神经网络体系结构中。在本文中,我们建议应用图形神经网络(GNN)来解决有效的神经网络架构设计和理论分析支持的大规模无线电资源管理问题。具体而言,我们首先证明无线电资源管理问题可以作为图形优化问题,这些问题享有通用置换率属性属性。然后,我们确定一类神经网络,称为\ emph {消息传递图神经网络}(mpgnns)。已经证明,他们不仅满足置换率属性的属性,而且可以在享受高计算效率的同时推广到大规模问题。为了解释性和理论保证,我们证明了MPGNN和一类分布式优化算法之间的等价性,然后将其用于分析基于MPGNN的方法的性能和概括。具有功率控制和光束形成为两个示例的广泛模拟将证明,该方法以无标记的样本,匹配,甚至超过经典优化的基于域的知识的算法,以无标记的样本,匹配或均不超过域的知识训练。值得注意的是,所提出的方法是高度可扩展的,可以在单个GPU上以$ 1000 $的收发器对解决干扰通道中的波束成分问题。

Deep learning has recently emerged as a disruptive technology to solve challenging radio resource management problems in wireless networks. However, the neural network architectures adopted by existing works suffer from poor scalability, generalization, and lack of interpretability. A long-standing approach to improve scalability and generalization is to incorporate the structures of the target task into the neural network architecture. In this paper, we propose to apply graph neural networks (GNNs) to solve large-scale radio resource management problems, supported by effective neural network architecture design and theoretical analysis. Specifically, we first demonstrate that radio resource management problems can be formulated as graph optimization problems that enjoy a universal permutation equivariance property. We then identify a class of neural networks, named \emph{message passing graph neural networks} (MPGNNs). It is demonstrated that they not only satisfy the permutation equivariance property, but also can generalize to large-scale problems while enjoying a high computational efficiency. For interpretablity and theoretical guarantees, we prove the equivalence between MPGNNs and a class of distributed optimization algorithms, which is then used to analyze the performance and generalization of MPGNN-based methods. Extensive simulations, with power control and beamforming as two examples, will demonstrate that the proposed method, trained in an unsupervised manner with unlabeled samples, matches or even outperforms classic optimization-based algorithms without domain-specific knowledge. Remarkably, the proposed method is highly scalable and can solve the beamforming problem in an interference channel with $1000$ transceiver pairs within $6$ milliseconds on a single GPU.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源