Abstract: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.

On the size generalizibility of graph neural networks for learning resource allocation

Size Generalization for Resource Allocation with Graph Neural Networks

Recursive GNNs for Learning Precoding Policies with Size-Generalizability

Scaling Graph Neural Networks for Large-Scale Power Systems Analysis: Empirical Laws for Emergent Abilities

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

Transferability Properties of Graph Neural Networks

Learning Resource Allocation Policy: Vertex-GNN or Edge-GNN?

Wasserstein Barycenter Matching for Graph Size Generalization of Message Passing Neural Networks

Enhancing Size Generalization in Graph Neural Networks through Disentangled Representation Learning

Scalable Graph Neural Networks for Heterogeneous Graphs

Towards Scale-Invariant Graph-related Problem Solving by Iterative Homogeneous Graph Neural Networks

On Size-Oriented Long-Tailed Graph Classification of Graph Neural Networks

Towards Bridging Generalization and Expressivity of Graph Neural Networks

Fast Learning of Graph Neural Networks with Guaranteed Generalizability: One-hidden-layer Case

Towards Better Generalization with Flexible Representation of Multi-Module Graph Neural Networks

Generalizing Aggregation Functions in GNNs:High-Capacity GNNs via Nonlinear Neighborhood Aggregators

Robust Generalization of Graph Neural Networks for Carrier Scheduling

Graph Neural Networks for Wireless Networks: Graph Representation, Architecture and Evaluation

Transferability of Graph Neural Networks using Graphon and Sampling Theories

Message gain aggregation architecture: a scalable graph neural network for combining large-scale neighborhoods

Generalization Bounds for Message Passing Networks on Mixture of Graphons