MAGNNETO: A Graph Neural Network-based Multi-Agent system for Traffic Engineering
This addresses network congestion optimization for internet service providers, offering a faster, distributed ML-based solution that generalizes to unseen networks, though it builds incrementally on existing ML and optimization techniques.
The paper tackles traffic engineering in ISP networks by proposing MAGNNETO, a distributed framework using multi-agent reinforcement learning and graph neural networks to optimize link weights and minimize congestion, achieving comparable performance to state-of-the-art optimizers with significantly lower execution times across over 75 topologies.
Current trends in networking propose the use of Machine Learning (ML) for a wide variety of network optimization tasks. As such, many efforts have been made to produce ML-based solutions for Traffic Engineering (TE), which is a fundamental problem in ISP networks. Nowadays, state-of-the-art TE optimizers rely on traditional optimization techniques, such as Local search, Constraint Programming, or Linear programming. In this paper, we present MAGNNETO, a distributed ML-based framework that leverages Multi-Agent Reinforcement Learning and Graph Neural Networks for distributed TE optimization. MAGNNETO deploys a set of agents across the network that learn and communicate in a distributed fashion via message exchanges between neighboring agents. Particularly, we apply this framework to optimize link weights in OSPF, with the goal of minimizing network congestion. In our evaluation, we compare MAGNNETO against several state-of-the-art TE optimizers in more than 75 topologies (up to 153 nodes and 354 links), including realistic traffic loads. Our experimental results show that, thanks to its distributed nature, MAGNNETO achieves comparable performance to state-of-the-art TE optimizers with significantly lower execution times. Moreover, our ML-based solution demonstrates a strong generalization capability to successfully operate in new networks unseen during training.