Documents
Presentation Slides
State-Augmented Information Routing in Communication Systems with Graph Neural Networks
- DOI:
- 10.60864/3m85-zb55
- Citation Author(s):
- Submitted by:
- Sourajit Das
- Last updated:
- 6 June 2024 - 10:23am
- Document Type:
- Presentation Slides
- Document Year:
- 2024
- Event:
- Presenters:
- Sourajit Das
- Paper Code:
- SPCOM-L4.4
- Categories:
- Log in to post comments
We consider the problem of routing network packets in a large-scale communication system where the nodes have access to only local information. We formulate this problem as a constrained learning problem, which can be solved using a distributed optimization algorithm. We approach this distributed optimization using a novel state-augmentation (SA) strategy to maximize the aggregate information packets at different source nodes, leveraging dual variables corresponding to flow constraint violations. The construction is based on graph neural networks (GNNs) that employ graph convolutions over the underlying communication network topology. We devise an unsupervised learning algorithm to transform the output of the GNN architecture into optimal routing decisions. The proposed method takes advantage of only the local information available at each node and efficiently routes the desired packets to the destination. We provide numerical results demonstrating the superiority of the proposed method over baseline routing algorithms.