AI-Driven Network Traffic Prediction and Optimization: A Systematic Review of Machine Learning and Deep Learning Approaches

Authors

  • Dr. Murli Manohar

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

Published

2026-09-04

Issue

Section

Articles