Deep Reinforcement Learning meets Graph Neural Networks: An optical network routing use case

Recent advances in Deep Reinforcement Learning (DRL) have shown a significant improvement in decision-making problems. The networking community has started to investigate how DRL can provide a new breed of solutions to relevant optimization problems, such as routing. However, most of the state-of-the-art DRL-based networking techniques fail to generalize, this means that they can only operate over network topologies seen during training, but not over new topologies. The reason behind this important limitation is that existing DRL networking solutions use standard neural networks (e.g., fully connected), which are unable to learn graph-structured information. In this paper we propose to use Graph Neural Networks (GNN) in combination with DRL. GNN have been recently proposed to model graphs, and our novel DRL+GNN architecture is able to learn, operate and generalize over arbitrary network topologies. To showcase its generalization capabilities, we evaluate it on an Optical Transport Network (OTN) scenario, where the agent needs to allocate traffic demands efficiently. Our results show that our DRL+GNN agent is able to achieve outstanding performance in topologies unseen during training.

[1]  Roberto Proietti,et al.  Deep-RMSA: A Deep-Reinforcement-Learning Routing, Modulation and Spectrum Assignment Agent for Elastic Optical Networks , 2018, 2018 Optical Fiber Communications Conference and Exposition (OFC).

[2]  Léon Bottou,et al.  Large-Scale Machine Learning with Stochastic Gradient Descent , 2010, COMPSTAT.

[3]  David Lynch,et al.  Two use cases of machine learning for SDN-enabled ip/optical networks: traffic matrix prediction and optical path performance prediction [Invited] , 2018, IEEE/OSA Journal of Optical Communications and Networking.

[4]  Samuel S. Schoenholz,et al.  Neural Message Passing for Quantum Chemistry , 2017, ICML.

[5]  Albert Cabellos-Aparicio,et al.  Unveiling the potential of Graph Neural Networks for network modeling and optimization in SDN , 2019, SOSR.

[6]  Matthew Roughan,et al.  The Internet Topology Zoo , 2011, IEEE Journal on Selected Areas in Communications.

[7]  Ian F. Akyildiz,et al.  QoS-Aware Adaptive Routing in Multi-layer Hierarchical Software Defined Networks: A Reinforcement Learning Approach , 2016, 2016 IEEE International Conference on Services Computing (SCC).

[8]  Joao Santos,et al.  Performance evaluation of integrated OTN/DWDM networks with single-stage multiplexing of optical channel data units , 2011, 2011 13th International Conference on Transparent Optical Networks.

[9]  Axel van Lamsweerde,et al.  Learning machine learning , 1991 .

[10]  Yusheng Ji,et al.  Understanding the Modeling of Computer Network Delays using Neural Networks , 2018, Big-DAMA@SIGCOMM.

[11]  Albert Cabellos-Aparicio,et al.  Challenging the generalization capabilities of Graph Neural Networks for network modeling , 2019, SIGCOMM Posters and Demos.

[12]  Jean C. Walrand,et al.  Knowledge-Defined Networking: Modelització de la xarxa a través de l’aprenentatge automàtic i la inferència , 2016 .

[13]  Jun Zhang,et al.  Wavelength converter placement in least-load-routing-based optical networks using genetic algorithms , 2004 .

[14]  Shane Legg,et al.  Human-level control through deep reinforcement learning , 2015, Nature.

[15]  Miriam Di Ianni,et al.  Efficient Delay Routing , 1998, Theor. Comput. Sci..

[16]  Albert Cabellos-Aparicio,et al.  Routing in optical transport networks with deep reinforcement learning , 2019, IEEE/OSA Journal of Optical Communications and Networking.

[17]  Demis Hassabis,et al.  Mastering Chess and Shogi by Self-Play with a General Reinforcement Learning Algorithm , 2017, ArXiv.

[18]  Richard S. Zemel,et al.  Gated Graph Sequence Neural Networks , 2015, ICLR.

[19]  Michael L. Littman,et al.  Packet Routing in Dynamically Changing Networks: A Reinforcement Learning Approach , 1993, NIPS.

[20]  Pietro Liò,et al.  Graph Attention Networks , 2017, ICLR.

[21]  Chi Harold Liu,et al.  Experience-driven Networking: A Deep Reinforcement Learning based Approach , 2018, IEEE INFOCOM 2018 - IEEE Conference on Computer Communications.

[22]  Razvan Pascanu,et al.  Relational inductive biases, deep learning, and graph networks , 2018, ArXiv.

[23]  Xavier Bresson,et al.  Convolutional Neural Networks on Graphs with Fast Localized Spectral Filtering , 2016, NIPS.

[24]  Razvan Pascanu,et al.  Interaction Networks for Learning about Objects, Relations and Physics , 2016, NIPS.

[25]  Ah Chung Tsoi,et al.  The Graph Neural Network Model , 2009, IEEE Transactions on Neural Networks.

[26]  Alex Graves,et al.  Playing Atari with Deep Reinforcement Learning , 2013, ArXiv.

[27]  Fernando Barreto,et al.  Fast emergency paths schema to overcome transient link failures in ospf routing , 2012, ArXiv.

[28]  Geoffrey E. Hinton,et al.  On the importance of initialization and momentum in deep learning , 2013, ICML.

[29]  Albert Cabellos-Aparicio,et al.  Routing Based on Deep Reinforcement Learning in Optical Transport Networks , 2019, 2019 Optical Fiber Communications Conference and Exhibition (OFC).