Spatial-Temporal Cellular Traffic Prediction for 5G and Beyond: A Graph Neural Networks-Based Approach

Abstract

During the past decade, Industry 4.0 has greatly promoted the improvement of industrial productivity by introducing advanced communication and network technologies in the manufacturing process. With the continuous emergence of new communication technologies and networking facilities, especially the rapid evolution of cellular networks for 5 G and beyond, the requirements for smarter, more reliable, and more efficient cellular network services have been raised from the Industry 5.0 blueprint. To meet these increasingly challenging requirements, proactive and effective allocation of cellular network resources becomes essential. As an integral part of the cellular network resource management system, cellular traffic prediction faces severe challenges with stringent requirements for accuracy and reliability. One of the most critical problems is how to improve the prediction performance by jointly exploring the spatial and temporal information within the cellular traffic data. A promising solution to this problem is provided by Graph Neural Networks (GNNs), which can jointly leverage the cellular traffic in the temporal domain and the physical or logical topology of cellular networks in the spatial domain to make accurate predictions. In this paper, we present the spatial-temporal analysis of a real-world cellular network traffic dataset and review the state-of-the-art researches in this field. Based on this, we further propose a time-series similarity-based graph attention network, TSGAN, for the spatial-temporal cellular traffic prediction. The simulation results show that our proposed TSGAN outperforms three classic prediction models based on GNNs or GRU on a real-world cellular network dataset in short-term, mid-term, and long-term prediction scenarios.