AI-Driven Network Traffic Prediction and Optimization: A Systematic Review of Machine Learning and Deep Learning Approaches
Abstract
Efficient network traffic management is essential for maintaining high performance, reliability, and quality of service in modern communication infrastructures. The increasing volume and variability of network traffic make conventional statistical and rule-based prediction methods insufficient for many contemporary networking environments. Machine Learning (ML) and Deep Learning (DL) have therefore gained considerable attention for predicting traffic patterns and optimizing network resources. This review systematically examines AI-based approaches for network traffic prediction, including regression models, decision trees, support vector machines, recurrent neural networks, long short-term memory networks, convolutional neural networks, transformers, and hybrid architectures. Existing studies are analyzed with respect to traffic characteristics, datasets, prediction horizons, evaluation metrics, and deployment environments. The review also investigates how traffic prediction can support routing optimization, congestion control, bandwidth allocation, load balancing, and quality-of-service management. Particular attention is given to the challenges of concept drift, highly dynamic traffic, data imbalance, computational complexity, and model interpretability. The paper concludes by identifying opportunities for transformer-based models, federated learning, and real-time AI-enabled traffic engineering.
References
Wang, M., Cui, Y., Wang, X., Xiao, S., & Jiang, J. (2018). Machine learning for networking: Workflow, advances and opportunities. IEEE Network, 32(2), 92–99. https://doi.org/10.1109/MNET.2017.1700200
Satish, E. G., Mounika, B. V. S., Anand, A., Venkata, A. K. P., & Chandrashekar, G. (2025, October). Adaptive Sdn-Based Lightweight Cryptographic Framework for Post-Quantum Secure Smart Home Iot Networks. In 2025 2nd International Conference on Software, Systems and Information Technology (SSITCON) (pp. 01-08). IEEE
Zhang, C., Patras, P., & Haddadi, H. (2019). Deep learning in mobile and wireless networking: A survey. IEEE Communications Surveys & Tutorials, 21(3), 2224–2287. https://doi.org/10.1109/COMST.2019.2904897
Mao, Q., Hu, F., & Hao, Q. (2018). Deep learning for intelligent wireless networks: A comprehensive survey. IEEE Communications Surveys & Tutorials, 20(4), 2595–2621. https://doi.org/10.1109/COMST.2018.2846401
Mao, B., Tang, F., Kawamoto, Y., & Kato, N. (2021). Optimizing computation offloading in satellite-terrestrial networks through supermodular game. IEEE Wireless Communications, 28(5), 18–25.
Luong, N. C., Hoang, D. T., Gong, S., Niyato, D., Wang, P., Liang, Y.-C., & Kim, D. I. (2019). Applications of deep reinforcement learning in communications and networking: A survey. IEEE Communications Surveys & Tutorials, 21(4), 3133–3174. https://doi.org/10.1109/COMST.2019.2916583
Tang, F., Mao, B., Kawamoto, Y., & Kato, N. (2021). Survey on machine learning for intelligent end-to-end communication toward 6G: From network access, routing to traffic control and streaming adaption. IEEE Communications Surveys & Tutorials, 23(3), 1578–1598. https://doi.org/10.1109/COMST.2021.3073009
Letaief, K. B., Shi, Y., Lu, J., & Lu, J. (2019). The roadmap to 6G: AI empowered wireless networks. IEEE Communications Magazine, 57(8), 84–90. https://doi.org/10.1109/MCOM.2019.1900271
Saad, W., Bennis, M., & Chen, M. (2020). A vision of 6G wireless systems: Applications, trends, technologies, and open research problems. IEEE Network, 34(3), 134–142. https://doi.org/10.1109/MNET.001.1900287
Anand, A. (2025). Self-Healing Network Infrastructure using AI-based Intent Recognition. Available at SSRN 5726222
Zhang, Z., Xiao, Y., Ma, Z., Xiao, M., Ding, Z., Lei, X., Karagiannidis, G. K., & Fan, P. (2019). 6G wireless networks: Vision, requirements, architecture, and key technologies. IEEE Vehicular Technology Magazine, 14(3), 28–41. https://doi.org/10.1109/MVT.2019.2921477
Dang, S., Amin, O., Shihada, B., & Alouini, M.-S. (2020). What should 6G be? Nature Electronics, 3, 20–29. https://doi.org/10.1038/s41928-019-0355-6
Akyildiz, I. F., Kak, A., & Nie, S. (2020). Internet of Things (IoT): 5G and 6G vision, architecture, and applications. Microprocessors and Microsystems, 77, 103237. https://doi.org/10.1016/j.micpro.2020.103237
Shafin, R., Chen, L., Matinmikko-Blue, M., & Saad, W. (2020). Artificial intelligence-enabled cellular networks: A survey. IEEE Communications Surveys & Tutorials, 22(1), 795–819.
Sirohi, D., Kumar, N., Rana, P. S., Tanwar, S., Iqbal, R., & Hijjii, M. (2023). Federated learning for 6G-enabled secure communication systems: A comprehensive survey. Artificial Intelligence Review, 56, 1–93. https://doi.org/10.1007/s10462-023-10417-3
Niknam, S., Dhillon, H. S., & Reed, J. H. (2020). Federated learning for wireless communications: Motivation, opportunities and challenges. IEEE Communications Magazine, 58(6), 46–51.
Lim, W. Y. B., Luong, N. C., Hoang, D. T., Jiao, Y., Liang, Y.-C., Yang, Q., Niyato, D., & Miao, C. (2020). Federated learning in mobile edge networks: A comprehensive survey. IEEE Communications Surveys & Tutorials, 22(3), 2031–2063.
Nguyen, D. C., Ding, M., Pathirana, P. N., Seneviratne, A., Li, J., Niyato, D., & Poor, H. V. (2022). Federated learning for Internet of Things: A comprehensive survey. IEEE Communications Surveys & Tutorials, 23(3), 1622–1658.
Rahmani, S., Baghbani, A., Bouguila, N., & Patterson, Z. (2023). Graph neural networks for intelligent transportation systems: A survey. IEEE Transactions on Intelligent Transportation Systems, 24(8), 8846–8885. https://doi.org/10.1109/TITS.2023.3257759
Ouyang, Y., Zhao, Z., & others. (2025). AI-enabled routing in next generation networks: A survey. Alexandria Engineering Journal, 120, 449–474. https://doi.org/10.1016/j.aej.2025.01.095
Shankar, S. S., MuhsnHasan, M., Anand, A., Singh, S., & Kavitha, K. R. (2025, August). Trust-Aware Fuzzy Clustering and Hippopotamus Optimization Algorithm for Secure and Energy-Efficient Routing in Wireless Sensor Networks. In 2025 2nd International Conference on Intelligent Algorithms for Computational Intelligence Systems (IACIS) (pp. 1-8). IEEE
Tang, F., Mao, B., Kawamoto, Y., & Kato, N. (2021). Survey on machine learning for intelligent end-to-end communication toward 6G: From network access, routing to traffic control and streaming adaption. IEEE Communications Surveys & Tutorials, 23(3), 1578–1598. https://doi.org/10.1109/COMST.2021.3073009
Rahmani, M., Norouzi, S., Chen, J., Ahmadi, H., Braun, T., Chowdhury, K., & Burr, A. G. (2026). Hybrid GNN-centric architectures for AI-native 6G wireless networks: A comprehensive survey. IEEE Communications Surveys & Tutorials, 28, 5678–5712. https://doi.org/10.1109/COMST.2026.3681198