Graph Neural Network and Spatial-Temporal Foundation Model Integration for Intelligent Mobile Network Traffic Prediction

Authors

  • Haolan Jiang Department of Computer Science and Engineering, University at Buffalo, Buffalo, NY, USA. Author

Keywords:

Graph neural networks; spatial-temporal foundation models; mobile network traffic prediction; model integration; AI governance; sustainable AI

Abstract

The rapid growth of mobile data traffic and the evolution toward ultra-dense and heterogeneous network architectures demand a fundamental shift in how network intelligence is realized. This paper presents a system-level investigation of the integration of graph neural networks (GNNs) with spatial-temporal foundation models for intelligent mobile network traffic prediction. Moving beyond algorithmic novelty, the analysis focuses on structural trade-offs, architectural design patterns, data infrastructure requirements, and governance mechanisms that shape the feasibility and trustworthiness of such integrated systems. We examine how GNNs can encode dynamic topological relationships among base stations and user mobility patterns, while large-scale spatial-temporal foundation models capture complex temporal dependencies across multiple scales and network regions. The fusion of these two paradigms opens new pathways for highly accurate and generalizable traffic forecasting, yet it introduces profound challenges related to computational sustainability, model interpretability, fairness across diverse geographic and demographic segments, and alignment with evolving regulatory frameworks. The paper discusses deployment strategies spanning centralized cloud, multi-access edge computing, and on-device inference, highlighting the tension between latency constraints and energy consumption. It further elaborates on data governance imperatives in the context of privacy-preserving, cross-operator learning and on the necessity of rigorous model auditing to ensure robustness against distributional shifts and adversarial perturbations. By synthesizing insights from network engineering, artificial intelligence, and technology policy, this work advances a holistic perspective that can guide the responsible integration of foundation model capabilities into critical communication infrastructures.

References

1. Yu, B., Yin, H., & Zhu, Z. (2018). Spatio-temporal graph convolutional networks: A deep learning framework for traffic forecasting. In Proceedings of the 27th International Joint Conference on Artificial Intelligence (pp. 3634–3640). AAAI Press.

2. Li, Y., Yu, R., Shahabi, C., & Liu, Y. (2018). Diffusion convolutional recurrent neural network: Data-driven traffic forecasting. In International Conference on Learning Representations.

3. Wu, Z., Pan, S., Long, G., Jiang, J., & Zhang, C. (2019). Graph WaveNet for deep spatial-temporal graph modeling. In Proceedings of the 28th International Joint Conference on Artificial Intelligence (pp. 1907–1913). AAAI Press.

4. Zhou, H., Zhang, S., Peng, J., Zhang, S., Li, J., Xiong, H., & Zhang, W. (2021). Informer: Beyond efficient transformer for long sequence time-series forecasting. In Proceedings of the AAAI Conference on Artificial Intelligence (Vol. 35, No. 12, pp. 11106–11115).

5. Garza, A., & Mergenthaler-Canseco, M. (2023). Lag-Llama: Towards foundation models for time series forecasting. arXiv preprint arXiv:2310.08278.

6. Das, A., Kong, S., Sen, R., & Zhou, Y. (2023). TSMixer: An all-MLP architecture for time series forecasting. Transactions on Machine Learning Research.

7. Zhang, J., Zheng, Y., & Qi, D. (2017). Deep spatio-temporal residual networks for citywide crowd flows prediction. In Proceedings of the AAAI Conference on Artificial Intelligence (Vol. 31, No. 1).

8. Wang, X., Chen, C., & Min, G. (2021). Spatio-temporal graph neural networks for mobile network traffic prediction: A survey. IEEE Communications Surveys & Tutorials, 23(3), 1808–1835.

9. Xu, J., Zhang, Y., & Wang, Q. (2022). Federated learning for mobile network traffic prediction: A communication-efficient approach. IEEE Transactions on Network and Service Management, 19(2), 1582–1593.

10. Bommasani, R., Hudson, D. A., Adeli, E., Altman, R., Arora, S., von Arx, S., Bernstein, M. S., Bohg, J., Bosselut, A., Brunskill, E., Brynjolfsson, E., Buch, S., Card, D., Castellon, R., Chatterji, N., Chen, A., Creel, K., Davis, J. Q., Demszky, D., … Liang, P. (2021). On the opportunities and risks of foundation models. arXiv preprint arXiv:2108.07258.

11. Woo, G., Liu, C., Sahoo, D., Kumar, A., & Hoi, S. (2024). Unified training of universal time series forecasting transformers. In Proceedings of the 27th International Conference on Artificial Intelligence and Statistics.

12. Tang, J., Yang, Y., Wei, W., Shi, L., Su, L., ChengTang, J., Yang, Y., Wei, W., Shi, L., Su, L., & Cheng, Y. (2024). GraphGPT: Graph instruction tuning for large language models. arXiv preprint arXiv:2311.06083.

13. Zhang, C., Zhang, H., Qiao, J., Li, Z., & Alouini, M. S. (2025). TIDES: Traffic Intelligence with DeepSeek Enhanced Spatial Temporal Prediction. IEEE Journal on Selected Areas in Communications.

14. Strubell, E., Ganesh, A., & McCallum, A. (2019). Energy and policy considerations for deep learning in NLP. In Proceedings of the 57th Annual Meeting of the Association for Computational Linguistics (pp. 3645–3650). Association for Computational Linguistics.

15. Polese, M., Bonati, L., D’Oro, S., Basagni, S., & Melodia, T. (2023). Understanding O-RAN: Architecture, interfaces, algorithms, security, and research challenges. IEEE Communications Surveys & Tutorials, 25(2), 1376–1411.

16. Bender, E. M., Gebru, T., McMillan-Major, A., & Shmitchell, S. (2021). On the dangers of stochastic parrots: Can language models be too big? In Proceedings of the 2021 ACM Conference on Fairness, Accountability, and Transparency (pp. 610–623). Association for Computing Machinery.

17. Goodfellow, I. J., Shlens, J., & Szegedy, C. (2015). Explaining and harnessing adversarial examples. In International Conference on Learning Representations.

18. European Commission. (2021). Proposal for a regulation of the European Parliament and of the Council laying down harmonised rules on artificial intelligence (Artificial Intelligence Act) (COM(2021) 206 final). European Commission.

19. Ribeiro, M. T., Singh, S., & Guestrin, C. (2016). “Why should I trust you?” Explaining the predictions of any classifier. In Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining (pp. 1135–1144). Association for Computing Machinery.

Downloads

Published

2026-05-17

How to Cite

Graph Neural Network and Spatial-Temporal Foundation Model Integration for Intelligent Mobile Network Traffic Prediction. (2026). Journal of Advanced Artificial Intelligence Research, 1(1). https://www.jaair.org/index.php/home/article/view/113