A Deep Dive into AI-Based Network Traffic Prediction Using Heterogeneous Real Datasets
Source abstract
Abstract
Recent studies have highlighted that network traffic may be influenced by various external factors such as weather conditions and user behavior, making it challenging to achieve precise predictions using only historical traffic data. To address this limitation, this study proposes a multivariate time series prediction model that incorporates environmental variables, such as meteorological information, to improve the accuracy of network traffic forecasting. Five deep learning models—RNN, GRU, LSTM, CNN, and Transformer—were evaluated under the same experimental conditions. Performance was assessed using metrics such as MSE, RMSE, MAE, R2, and MAPE. In addition, ANOVA and Tukey HSD post hoc tests were conducted to analyze the statistical significance of performance differences between models, and the contribution of each environmental variable was evaluated using the Permutation Importance method, which demonstrated a significant impact on model performance. Experimental results indicated that the GRU and RNN models achieved the best overall prediction accuracy. Additionally, some weather variables, such as temperature and sunlight duration, positively impacted performance improvement. This study empirically demonstrates the generalization capabilities of simple recurrent architectures and the effectiveness of integrating environmental variables. Furthermore, it suggests future research directions, including cross-domain model adaptation and the application of large language model (LLM)-based time series forecasting frameworks.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Traffic Prediction and Management Techniques · Advanced Data and IoT Technologies · Traffic control and management
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
Department of Computer Science & Engineering, Kyung Hee University, Yongin-si 17104, Gyeonggi-do, Republic of Korea
References (67)
Show source-provided references
1. 5G traffic prediction based on deep learning10.1155/2022/3174530 ↗
2. Machine learning-based fifth-generation network traffic prediction using federated learning
3. TransMUSE: Transferable traffic prediction in multi-service edge networks10.1016/j.comnet.2022.109518 ↗
4. Comparative analysis of short-term demand predicting models using ARIMA and deep learning10.11591/ijece.v11i4.pp3319-3328 ↗
5. Ghojogh, B., and Ghodsi, A. (2023). Recurrent neural networks and long short-term memory networks: Tutorial and survey. arXiv.
6. Shiri, F.M., Perumal, T., Mustapha, N., and Mohamed, R. (2023). A comprehensive overview and comparative analysis on deep learning models: CNN, RNN, LSTM, GRU. arXiv.
7. Zerveas, G., Jayaraman, S., Patel, D., Bhamidipaty, A., and Eickhoff, C. (2021, January 14–18). A transformer-based framework for multivariate time series representation learning. Proceedings of the 27th ACM SIGKDD Conference on Knowledge Discovery & Data Mining, Virtual.10.1145/3447548.3467401 ↗
8. Attention Is All You Need
9. Review of deep learning: Concepts, CNN architectures, challenges, applications, future directions10.1186/s40537-021-00444-8 ↗
10. A survey of convolutional neural networks: Analysis, applications, and prospects10.1109/TNNLS.2021.3084827 ↗
11. A survey of transformers10.1016/j.aiopen.2022.10.001 ↗
12. A comprehensive survey on applications of transformers for deep learning tasks10.1016/j.eswa.2023.122666 ↗
13. Zhou, T., Ma, Z., Wen, Q., Wang, X., Sun, L., and Jin, R. (2022, January 17–23). FEDformer: Frequency Enhanced Decomposed Transformer for Long-Term Series Forecasting. Proceedings of the 39th International Conference on Machine Learning, PMLR 2022, Baltimore, MD, USA.
14. Oliveira, J.M., and Ramos, P. (2024). Evaluating the Effectiveness of Time Series Transformers for Demand Forecasting in Retail. Mathematics, 12.10.3390/math12172728 ↗
15. Zhou, H., Zhang, S., Peng, J., Zhang, S., Li, J., Xiong, H., and Zhang, W. (2021, January 2–6). Informer: Beyond efficient transformer for long sequence time-series forecasting. Proceedings of the AAAI Conference on Artificial Intelligence, Vancouver, BC, Canada.10.1609/aaai.v35i12.17325 ↗
16. California Department of Transportation (Caltrans) (2025, December 14). PeMS Data Source (Performance Measurement System), Available online: https://dot.ca.gov/about-caltrans.
17. Biswas, S., Grundlingh, N., Boardman, J., White, J., and Le, L. (2025). A Target Permutation Test for Statistical Significance of Feature Importance in Differentiable Models. Electronics, 14.10.3390/electronics14030571 ↗
18. Model-agnostic feature importance and effects with dependent features: A conditional subgroup approach10.1007/s10618-022-00901-9 ↗
19. Convolutional neural networks for time series classification10.21629/JSEE.2017.01.18 ↗
20. Pan, Z., Liang, Y., Wang, W., Yu, Y., Zheng, Y., and Zhang, J. (2019, January 4–8). Urban traffic prediction from spatio-temporal data using deep meta learning. Proceedings of the 25th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining, Anchorage, AL, USA.10.1145/3292500.3330884 ↗
21. Madariaga, D., Panza, M., and Bustos-Jiménez, J. (2018, January 26–29). I’m only unhappy when it rains: Forecasting mobile QoS with weather conditions. Proceedings of the 2018 Network Traffic Measurement and Analysis Conference (TMA), Vienna, Austria.10.23919/TMA.2018.8506509 ↗
22. Kuber, T., Seskar, I., and Mandayam, N. (April, January 29). Traffic prediction by augmenting cellular data with non-cellular attributes. Proceedings of the 2021 IEEE Wireless Communications and Networking Conference (WCNC), Nanjing, China.10.1109/WCNC49053.2021.9417547 ↗
23. Impact of Altitude and Weather Conditions on Cellular Networks: A Comprehensive Analysis of Quality of Service
24. Joshi, T., Semwal, S., and Rawat, A. (2024, January 22–24). Assessing Quality of Service in Cellular Networks: Effects of Altitude and Weather Conditions in Dehradun, Uttarakhand. Proceedings of the 2024 International Conference on Advances in Computing, Communication and Materials (ICACCM), Dehradun, India.10.1109/ICACCM61117.2024.11059176 ↗
25. Weather-aware fiber-wireless traffic prediction using graph convolutional networks10.1109/ACCESS.2022.3203420 ↗
26. Korea Meteorological Administration (KMA) (2025, December 14). Meteorological Data Open Portal. Available online: https://data.kma.go.kr/.
27. Lan, S., Ma, Y., Huang, W., Wang, W., Yang, H., and Li, P. (2022, January 17–23). DSTAGNN: Dynamic spatial-temporal aware graph neural network for traffic flow forecasting. Proceedings of the International Conference on Machine Learning (ICML), Baltimore, MD, USA.
28. Spatial-temporal attention wavenet: A deep learning framework for traffic prediction considering spatial-temporal dependencies10.1049/itr2.12044 ↗
29. Spatial-temporal cellular traffic prediction for 5G and beyond: A graph neural networks-based approach
30. Spatial–temporal dynamic graph convolutional network with interactive learning for traffic forecasting
31. Ji, J., Wang, J., Huang, C., Wu, J., Xu, B., Wu, Z., Zhang, J., and Zheng, Y. (2023, January 13–14). Spatio-temporal self-supervised learning for traffic flow prediction. Proceedings of the AAAI Conference on Artificial Intelligence, Washington, DC, USA.10.1609/aaai.v37i4.25555 ↗
32. Transfer learning with spatial–temporal graph convolutional network for traffic prediction10.1109/TITS.2023.3250424 ↗
33. Deep learning on network traffic prediction: Recent advances, analysis, and future directions10.1145/3703447 ↗
34. Song, C., Lin, R., Zheng, X., and Chen, Z. (2025, December 03). PeMSD7 and PeMSD4 Traffic Datasets. Available online: https://zenodo.org/records/7816008.
35. Li, Y., Yu, R., Shahabi, C., and Liu, Y. (2025, December 03). PEMS-BAY Traffic Dataset. Available online: https://zenodo.org/records/4263971.
36. Jagadish, V., Mori, U., Mendiburu, A., and Otaegi, A. (2025, December 03). METR-LA Traffic Dataset. Available online: https://zenodo.org/records/5146275.
37. Autoformer: Decomposition transformers with auto-correlation for long-term series forecasting
38. Zhou, H., Zhang, S., Peng, J., Zhang, S., Li, J., Xiong, H., and Zhang, W. (2021). Informer: Beyond efficient transformer for long sequence time-series forecasting. arXiv.10.1609/aaai.v35i12.17325 ↗
39. LSTTN: A long-short term transformer-based spatiotemporal neural network for traffic flow forecasting10.1016/j.knosys.2024.111637 ↗
40. MTESformer: Multi-scale temporal and enhance spatial transformer for traffic flow prediction10.1109/ACCESS.2024.3381987 ↗
41. ProSTformer: Progressive space-time self-attention model for short-term traffic flow forecasting10.1109/TITS.2024.3367754 ↗
42. Tedjopurnomo, D.A., Choudhury, F.M., and Qin, A.K. (2023). TrafFormer: A transformer model for predicting long-term traffic. arXiv.
43. (2025, December 03). TimeSeriesDatasets: Datasets for Time Series Forecasting. Available online: https://github.com/juyongjiang/TimeSeriesDatasets.
44. Shi, B., Wang, Z., Yan, J., Yang, Q., and Yang, N. (2024). A novel spatial–temporal deep learning method for metro flow prediction considering external factors and periodicity. Appl. Sci., 14.10.3390/app14051949 ↗
45. AST-GCN: Attribute-augmented spatiotemporal graph convolutional network for traffic forecasting10.1109/ACCESS.2021.3062114 ↗
46. Nadarajan, J., and Sivanraj, R. (2022). Attention-based multiscale spatiotemporal network for traffic forecast with fusion of external factors. ISPRS Int. J. Geo-Inf., 11.10.3390/ijgi11120619 ↗
47. Attention-based spatio-temporal graph convolutional network considering external factors for multi-step traffic flow prediction
48. Combining weather factors to predict traffic flow: A spatial-temporal fusion graph convolutional network-based deep learning approach10.1049/itr2.12401 ↗
49. Rajha, R., Shiode, S., and Shiode, N. (2025). Improving traffic-flow prediction using proximity to urban features and public space. Sustainability, 17.10.3390/su17010068 ↗
50. Large-scale traffic flow forecast with lightweight LLM in edge intelligence10.1109/IOTM.001.2400047 ↗
51. Patro, S.G., and Sahu, K.K. (2015). Normalization: A preprocessing stage. arXiv.10.17148/IARJSET.2015.2305 ↗
52. Investigating the impact of min-max data normalization on the regression performance of K-nearest neighbor with different similarity measurements
53. A Study on Network Traffic Prediction Based on Deep Learning Training Using Environmental In-formation10.5909/JBE.2025.30.3.461 ↗
54. Wen, Q., Zhou, T., Zhang, C., Chen, W., Ma, Z., Yan, J., and Sun, L. (2024). Transformers in time series: A survey. arXiv.10.24963/ijcai.2023/759 ↗
55. A Comparative Study of Multivariate Time Series Deep Learning Models for Network Traffic Pre-diction Incorporating Environmental Variables10.5909/JBE.2025.30.6.1093 ↗
56. Abadi, M., Barham, P., Chen, J., Chen, Z., Davis, A., Dean, J., Devin, M., Ghemawat, S., Irving, G., and Isard, M. (2016, January 2–4). TensorFlow: A system for large-scale machine learning. Proceedings of the 12th USENIX Symposium on Operating Systems Design and Implementation (OSDI 16), Savannah, GA, USA.
57. Zollanvari, A. (2023). Deep learning with Keras–TensorFlow. Machine Learning with Python: Theory and Implementation, Springer.10.1007/978-3-031-33342-2 ↗
58. Kingma, D.P., and Ba, J. (2017). Adam: A method for stochastic optimization. arXiv.
59. Ahn, K., Zhang, Z., Kook, Y., and Dai, Y. (2024, January 21–27). Understanding Adam optimizer via online learning of updates: Adam is FTRL in disguise. Proceedings of the 41st International Conference on Machine Learning, Vienna, Austria.
60. Lundberg, S.M., and Lee, S.-I. (2017). A Unified Approach to Interpreting Model Predictions. Advances in Neural Information Processing Systems, Curran Associates, Inc.
61. Greedy Function Approximation: A Gradient Boosting Machine10.1214/aos/1013203451 ↗
62. On the tractability of SHAP explanations10.1613/jair.1.13283 ↗
63. Arnastauskaitė, J., Ruzgas, T., and Bražėnas, M. (2021). An exhaustive power comparison of normality tests. Mathematics, 9.10.3390/math9070788 ↗
64. Performance of Levene’s Test with Various Residuals and Correction Factors for Homogeneity of Variance in Single and Factorial ANOVA—A Simulation Approach10.54364/cybersecurityjournal.2025.1107 ↗
65. Multiple comparison test by Tukey’s honestly significant difference (HSD): Do the confidence level control type I error10.22271/maths.2021.v6.i1a.636 ↗
66. Mohamad, D.A., Goeman, J.J., and van Zwet, E.W. (2017). An improvement of Tukey’s HSD with application to ranking institutions. arXiv.
67. Beyond Pearson’s Correlation: Modern Nonparametric Independence Tests for Psychological Research10.1080/00273171.2024.2347960 ↗