IEEE Virtual Conference on Communications (VCC)
GNN-based Proactive Entanglement Distribution in Space-Air-Ground Quantum Networks
Mohamed Shaban, Muhammad Ismail, Walid Saad, Don Towsley
Abstract
Quantum Internet is essential for unlocking a variety of applications. SPARQ is a space-air-ground quantum network with promising architecture for realizing the quantum Internet. In the early stages of quantum network deployment, limited resources, such as memory units, necessitate efficient management to prevent bottlenecks that can hinder on-demand entanglement distribution. Bottlenecks in quantum networks occur when intermediate repeaters lack the memory needed to accommodate additional entanglement requests. In SPARQ, satellite mobility and fluctuating communication demands lead to continuous changes in bottleneck locations. To address this challenge, in this paper, we propose a proactive entanglement distribution strategy to support on-demand entanglement by predicting and avoiding bottlenecks. Specifically, we propose a graph convolutional neural network (GCNN) framework to predict bottlenecks in SPARQ. The resulting predictions are then integrated into an entanglement distribution policy to proactively avoid bottlenecks, thereby improving the entanglement distribution rate. Simulation results demonstrate the effectiveness of the proposed GCNN model, achieving an accuracy of 97.5% in predicting bottlenecks. The proposed GCNN also improves the accuracy by 6.5% compared to a recurrent neural network (RNN) baseline. This leads to an average improvement of $41.3 \%$ in the entanglement rate when avoiding bottlenecks based on GCNN predictions compared with RNN predictions. The results also show that the proposed approach for anticipating bottlenecks leads to a $51.6 \%$ improvement in the entanglement distribution rate compared to scenarios where bottlenecks are not addressed.