Abstract
Recently, the surge in vehicle ownership has led to a corresponding increase in the complexity of traffic data. Consequently, accurate traffic flow prediction has become crucial for effective traffic management. While the advancements in intelligent transportation system (ITS) and internet of things (IoT) technology have facilitated traffic flow prediction, many existing methods overlook the influence of the training process on model accuracy. Traditional approaches often fail to account for this critical aspect. Hence, a new approach to traffic flow prediction is introduced in this paper: a spatial–temporal attention time-gated convolutional network based on particle swarm optimization (PSO-STATG). This method uses the particle swarm algorithm to dynamically optimize the learning rate and epoch parameters throughout the training process. Firstly, spatial–temporal correlations are extracted through spatial map convolution and time-gated convolution, facilitated by an attention mechanism. Subsequently, the learning rate and epoch parameters are dynamically adjusted during the training phase via the particle swarm optimization algorithm. Finally, experiments are conducted with real-world datasets, and the results are compared with those from several existing methods. The experimental results indicate that the accuracy and stability of our proposed model in predicting traffic flow are superior.
Similar content being viewed by others
Explore related subjects
Discover the latest articles, news and stories from top researchers in related subjects.Data availability
The original highway datasets are derived from [37].The raw data used in this work are available from the corresponding author upon request.
References
Liao L, Hu Z, Zheng Y, Bi S, Zou F, Qiu H, Zhang M (2022) An improved dynamic Chebyshev graph convolution network for traffic flow prediction with spatial-temporal attention. Appl Intell 52(14):16104–16116
Belhadi A, Djenouri Y, Djenouri D, Lin JC-WJAI (2020) A recurrent neural network for urban long-term traffic flow forecasting. Appl Intell 50:3252–3265
Kumar SV (2017) Traffic flow prediction using Kalman filtering technique. Procedia Eng 187:582–587
Shahriari S, Ghasri M, Sisson SA, Rashidi T (2020) Ensemble of ARIMA: combining parametric and bootstrapping technique for traffic flow prediction. Transportmetrica A: Transport Sci 16(3):1552–1573
Cong Y, Wang J, Li X (2016) Traffic flow forecasting by a least squares support vector machine with a fruit fly optimization algorithm. Procedia Eng 137:59–68
Lin G, Lin A, Gu D (2022) Using support vector regression and K-nearest neighbors for short-term traffic flow prediction based on maximal information coefficient. Inf Sci 608:517–531
Xu D, Wang Y, Peng P, Beilun S, Deng Z, Guo H (2020) Real-time road traffic state prediction based on kernel-KNN. Transportmetrica A: Transport Sci 16(1):104–118
Liu F, Wei Z, Huang Z, Lu Y, Hu X, Shi L (2019) A multi-grouped ls-svm method for short-term urban traffic flow prediction. In: 2019 IEEE Global Communications Conference (GLOBECOM). IEEE, pp 1–6
Dai G, Tang J, Luo W (2023) Short-term traffic flow prediction: An ensemble machine learning approach. Alex Eng J 74:467–480
Guo Y, Peng Y, Hao R, Tang X (2023) Capturing spatial–temporal correlations with Attention based Graph Convolutional Network for network traffic prediction. J Netw Comput Appl 220:103746
Bai D, Xia D, Huang D, Hu Y, Li Y, Li H (2023) Spatial-temporal graph neural network based on gated convolution and topological attention for traffic flow prediction. Appl Intell 53(24):30843–30864
Geng Z, Xu J, Wu R, Zhao C, Wang J, Li Y, Zhang C (2024) STGAFormer: Spatial–temporal gated attention transformer based graph neural network for traffic flow forecasting. Information Fusion 105:102228
Wei F, Li X, Guo Y, Wang Z, Li Q, Ma X (2023) Flow direction level traffic flow prediction based on a GCN-LSTM combined model. Intell Automation Soft Computing 37(2):2001–2018
Zheng H, Li X, Li Y, Yan Z, Li T (2022) GCN-GAN: integrating graph convolutional network and generative adversarial network for traffic flow prediction. IEEE Access 10:94051–94062
Bao Y, Liu J, Shen Q, Cao Y, Ding W, Shi Q (2023) PKET-GCN: prior knowledge enhanced time-varying graph convolution network for traffic flow prediction. Inf Sci 634:359–381
Guo K, Hu Y, Qian Z, Sun Y, Gao J, Yin B (2020) Dynamic graph convolution network for traffic forecasting based on latent network of Laplace matrix estimation. IEEE Trans Intell Transp Syst 23(2):1009–1018
Zhang W, Zhu F, Lv Y, Tan C, Liu W, Zhang X, Wang F-Y (2022) AdapGL: An adaptive graph learning algorithm for traffic prediction based on spatiotemporal neural networks. Transportation Res Part C: Emerging Technologies 139:103659
Sun L, Liu M, Liu G, Chen X, Yu X (2024) FD-TGCN: fast and dynamic temporal graph convolution network for traffic flow prediction. Inf Fusion 106:102291
Liu S, He M, Wu Z, Lu P, Gu W (2024) Spatial–temporal graph neural network traffic prediction based load balancing with reinforcement learning in cellular networks. Information Fusion 103:102079
Xing H, Chen A, Zhang X (2023) RL-GCN: Traffic flow prediction based on graph convolution and reinforcement learning for smart cities. Displays 80:102513
Liu Z, Ding F, Dai Y, Li L, Chen T, Tan H (2024) Spatial-temporal graph convolution network model with traffic fundamental diagram information informed for network traffic flow prediction. Expert Syst Appl 249:123543
Bao Y, Shen Q, Cao Y, Ding W, Shi Q (2024) Residual attention enhanced Time-varying Multi-Factor Graph Convolutional Network for traffic flow prediction. Eng Appl Artif Intell 133:108135
Chen J, Zheng L, Hu Y, Wang W, Zhang H, Hu X (2024) Traffic flow matrix-based graph neural network with attention mechanism for traffic flow prediction. Information Fusion 104:102146
Luo Y, Zheng J, Wang X, Tao Y, Jiang X (2024) GT-LSTM: A spatio-temporal ensemble network for traffic flow prediction. Neural Netw 171:251–262
Feng F, Li Y-B, Chen Z-H, Wu W-T, Peng J-Z, Mei M (2024) Rapid optimization for inner thermal layout in horizontal annuli using genetic algorithm coupled graph convolutional neural network. Int Commun Heat Mass Transfer 150:107210
Xu S, Xu X, Jia W, Liu W, Li J, Li D (2023) Microstructure-property mapping modeling for AZ31 alloy rolling deformation using improved PSO-BP neural network. J Market Res 25:2127–2139
Pei Z, Liu K, Zhang S, Chen X (2023) Optimized EKF algorithm using TSO-BP neural network for lithium battery state of charge estimation. J Energy Storage 73:108882
Guo J, Chen C, Wen H, Cai G, Liu Y (2024) Prediction model of goaf coal temperature based on PSO-GRU deep neural network. Case Studies in Thermal Eng 53:103813
Li F, Zuo W, Zhou K, Li Q, Huang Y (2024) State of charge estimation of lithium-ion batteries based on PSO-TCN-Attention neural network. Journal of Energy Storage 84:110806
Sahib M-M, Kovács G (2024) Multi-objective optimization of composite sandwich structures using Artificial Neural Networks and Genetic Algorithm. Results Eng 21:101937
Lee J, Park D, Park K, Song H, Kim T-S, Ryu S (2024) Optimization of grid composite configuration to maximize toughness using integrated hierarchical deep neural network and genetic algorithm. Mater Des 238:112700
Naheliya B, Redhu P, Kumar K (2024) MFOA-Bi-LSTM: An optimized bidirectional long short-term memory model for short-term traffic flow prediction. Physica A 634:129448
Redhu P, Kumar K (2023) Short-term traffic flow prediction based on optimized deep learning neural network: PSO-Bi-LSTM. Physica A 625:129001
Du W, Zhang Q, Chen Y, Ye Z (2021) An urban short-term traffic flow prediction model based on wavelet neural network with improved whale optimization algorithm. Sustain Cities Soc 69:102858
Yan H, Qi Y, Yu D-J (2022) Short-term traffic flow prediction based on a hybrid optimization algorithm. Appl Math Model 102:385–404
Ke Q, Siłka J, Wieczorek M, Bai Z, Woźniak M (2022) Deep neural network heuristic hierarchization for cooperative intelligent transportation fleet management. IEEE Trans Intell Transp Syst 23(9):16752–16762
Guo S, Lin Y, Feng N, Song C, Wan H (2019) Attention based spatial-temporal graph convolutional networks for traffic flow forecasting. Proceedings of the AAAI Conference Artificial Intell 33:922–929
Xiong L, Yuan X, Hu Z, Huang X, Huang P (2024) Gated Fusion Adaptive Graph Neural Network for Urban Road Traffic Flow Prediction. Neural Process Lett 56(1):9
Cortes C, Vapnik V (1995) Support-vector networks Machine learning 20:273–297
Hochreiter S, Schmidhuber J (1997) Long short-term memory. Neural Comput 9(8):1735–1780
Yu B, Yin H, Zhu Z (2017) Spatio-temporal graph convolutional networks: a deep learning framework for traffic forecasting. arXiv preprint arXiv:1709.04875
Ni Q, Zhang M (2022) STGMN: A gated multi-graph convolutional network framework for traffic flow prediction. Appl Intell 52(13):15026–15039
Bai L, Yao L, Li C, Wang X, Wang C (2020) Adaptive graph convolutional recurrent network for traffic forecasting. Adv Neural Inf Process Syst 33:17804–17815
Acknowledgements
This work was supported by the National Natural Science Foundation of China, grant number (52272367).
Author information
Authors and Affiliations
Contributions
Zhongxing Li: Supervision, Formal Analysis, Writing-Review and Editing. Zenan Li: Methodology, Visualization, Software, Writing-Original Draft. Chaofeng Pan: Funding Acquisition, Resources, Project Administration. Jian Wang: Conceptualization, Writing-Review and Editing.
Corresponding author
Ethics declarations
Declarations
Ethical and informed consent for data used.
The data used in this study come from the California State Traffic Performance Measurement System (PEMS) public datasets, designed for transportation research. They do not contain any personally identifiable information or sensitive data. No specific informed consent was required for this study as the data used were obtained from publicly accessible sources which do not contain information related to individual human subjects.
Competing interests
The authors declare that they have no known competing financial interests or personal relationships that could have appeared to influence the work reported in this paper.
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
About this article
Cite this article
Li, Z., Li, Z., Pan, C. et al. Short-term traffic flow prediction based on spatial–temporal attention time gated convolutional network with particle swarm optimization. Appl Intell 55, 214 (2025). https://doi.org/10.1007/s10489-024-06117-2
Accepted:
Published:
DOI: https://doi.org/10.1007/s10489-024-06117-2