RIGEL: Real-time Optical Anomaly Diagnosis with Stateful In-Network Inference based on Distributed On-switch GNNs

Zhen Wei, Yidong Wang, Yufan Zhu, Xuefeng Yan, Binjun Tang, Xiaoliang Chen, Zuqing Zhu

The recent booming of data-intensive applications has complicated optical network management, making real-time optical anomaly diagnosis a must-have feature. However, existing approaches are mostly based on centralized data analytics and thus can hardly avoid the latency and overhead due to message exchanges between data and control planes. In this work, we propose and prototype RIGEL, which, to the best of our knowledge, is the first real-time optical anomaly diagnosis system that realizes stateful distributed in-network inference through collaborative graph neural networks (GNNs) on Tofino switches. The system is designed to be fully in-network, and a software-hardware co-design is proposed to preprocess high-dimensional spectral data for being suitable for hardware-based in-network inference. Next, we first develop an effective model to combine an autoencoder with a GraphSAGE-based GNN, and then propose a generalizable method to adapt the model to Tofino switch. The effectiveness of RIGEL is showcased in a realistic packet-over-optical network testbed, verifying that it achieves highly accurate diagnosis to detect and locate optical anomalies timely and highlighting its benefits over the state-of-the-art methods.