QT-Routenet: Improved GNN generalization to larger 5G networks by fine-tuning predictions from queueing theory
This improves generalization for 5G network modeling, though it is incremental as it builds on existing queueing theory and GNN methods.
The paper tackled the problem of poor generalization of Graph Neural Networks (GNNs) to larger 5G networks with longer paths and higher link capacities than seen in training, by fine-tuning queueing theory predictions with a modified GNN, reducing mean absolute percent error from 10.42 to 1.45 (1.27 with an ensemble).
In order to promote the use of machine learning in 5G, the International Telecommunication Union (ITU) proposed in 2021 the second edition of the ITU AI/ML in 5G challenge, with over 1600 participants from 82 countries. This work details the second place solution overall, which is also the winning solution of the Graph Neural Networking Challenge 2021. We tackle the problem of generalization when applying a model to a 5G network that may have longer paths and larger link capacities than the ones observed in training. To achieve this, we propose to first extract robust features related to Queueing Theory (QT), and then fine-tune the analytical baseline prediction using a modification of the Routenet Graph Neural Network (GNN) model. The proposed solution generalizes much better than simply using Routenet, and manages to reduce the analytical baseline's 10.42 mean absolute percent error to 1.45 (1.27 with an ensemble). This suggests that making small changes to an approximate model that is known to be robust can be an effective way to improve accuracy without compromising generalization.