Abstract:Deep learning is widely used in wireless communications but struggles with fixed neural network sizes, which limit their adaptability in environments where the number of users and antennas varies. To overcome this, this paper introduced a generalization strategy for precoding and power allocation in scalable wireless networks. Initially, we employ an innovative approach to abstract the wireless network into a homogeneous graph. This primarily focuses on bypassing the heterogeneous features between transmitter (TX) and user entities to construct a virtual homogeneous graph serving optimization objectives, thereby enabling all nodes in the virtual graph to share the same neural network. This "TX entity" is known as a base station (BS) in cellular networks and an access point (AP) in cell-free networks. Subsequently, we design a universal graph neural network, termed the information carrying graph neural network (ICGNN), to capture and integrate information from this graph, maintaining permutation invariance. Lastly, using ICGNN as the core algorithm, we tailor the neural network's input and output for specific problem requirements and validate its performance in two scenarios: 1) in cellular networks, we develop a matrix-inverse-free multi-user multi-input multi-output (MU-MIMO) precoding scheme using the conjugate gradient (CG) method, adaptable to varying user and antenna numbers; 2) in a cell-free network, facing dynamic variations in the number of users served by APs, the number of APs serving each user, and the number of antennas per AP, we propose a universal power allocation scheme. Simulations demonstrate that the proposed approach not only significantly reduces computational complexity but also achieves, and potentially exceeds, the spectral efficiency (SE) of conventional algorithms.

Size Generalization for Resource Allocation with Graph Neural Networks

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

Recursive GNNs for Learning Precoding Policies with Size-Generalizability

Enhancing Size Generalization in Graph Neural Networks through Disentangled Representation Learning

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

Understanding Why Neural Networks Generalize Well Through GSNR of Parameters

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

Towards Bridging Generalization and Expressivity of Graph Neural Networks

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

Scalable Graph Neural Networks for Heterogeneous Graphs

Generalization Bounds for Message Passing Networks on Mixture of Graphons

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

A Homogeneous Graph Neural Network for Precoding and Power Allocation in Scalable Wireless Networks

Multidimensional Graph Neural Networks for Wireless Communications

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

Joint Admission and Power Control for Massive Connections via Graph Neural Network

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

A new perspective for understanding generalization gap of deep neural networks trained with large batch sizes

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

Ada-GNN: Adapting to Local Patterns for Improving Graph Neural Networks

ByteGNN: Efficient Graph Neural Network Training at Large Scale