Abstract
Forecasting traffic flow is crucial for Intelligent Traffic Systems (ITS), traffic control, and traffic management systems. Complex spatial and temporal interactions of traffic networks make traffic forecasting tasks challenging. Recently, Graph Convolutional Network (GCN) has attracted researchers’ attention as it can better represent graph-shaped road networks and extract spatial features of traffic. However, traditional GCN has some drawbacks since it uses a static adjacency matrix which is unable to capture the time-varying features of traffic propagation. To overcome this, we represent the traffic road network as a dynamic graph and use a probabilistic spatiotemporal adjacency matrix to identify the time-varying impacts of adjacent roads on target roads in GCN. In addition, to find the similarity among the nonadjacent nodes, we have employed node-specific learning in GCN rather than sharing parameters in traditional GCN. This node-specific learning helps our model to learn detailed characteristics of road networks. For temporal feature extractions, we used a Gated Recurrent Unit (GRU) that captures the local trend of traffic flow and an attention mechanism to capture the global trend of traffic flow. We compared the performance of our model with baseline models using two real-world datasets. Experimental results show that our model is effective in forecasting both short and long-term traffic flow. Source code of our model is available at https://github.com/atkia/PSTGCN
Similar content being viewed by others
Explore related subjects
Discover the latest articles, news and stories from top researchers in related subjects.Data availability and access
The datasets used are publicly available. We have added references to the source of datasets in the manuscript.
Code Availibility
Implementation of our model is available at https://github.com/atkia/PSTGCN.
References
Rahman MM, Nower N (2023) Attention based deep hybrid networks for traffic flow prediction using google maps data. In: Proceedings of the 2023 8th international conference on machine learning technologies, pp 74–81
Rahman MM, Jamil ARM, Nower N (2023) Uncertainty-aware traffic prediction using attention-based deep hybrid network with bayesian inference. Int J Adv Comput Sci Appl 14(6)
Yi H, Jung H, Bae S (2017) Deep neural networks for traffic flow prediction. In: 2017 IEEE International conference on big data and smart computing (BigComp), IEEE, pp 328–331
Zhao L, Song Y, Zhang C et al (2019) T-gcn: A temporal graph convolutional network for traffic prediction. IEEE Trans Intell Transportation Syst 21(9):3848–3858
Bai J, Zhu J, Song Y et al (2021) A3t-gcn: Attention temporal graph convolutional network for traffic forecasting. ISPRS Int J Geo-Inf 10(7):485
Zhu J, Wang Q, Tao C et al (2021) Ast-gcn: Attribute-augmented spatiotemporal graph convolutional network for traffic forecasting. IEEE Access 9:35973–35983
Chen Z, Lu Z, Chen Q et al (2022) Spatial-temporal short-term traffic flow prediction model based on dynamical-learning graph convolution mechanism. Inf Sci 611:522–539
Li B, Yang Q, Chen J et al (2023) A dynamic spatio-temporal deep learning model for lane-level traffic prediction. J Adv Transportation 2023
Bai L, Yao L, Li C et al (2020) Adaptive graph convolutional recurrent network for traffic forecasting. Adv Neural Inf Process Syst 33:17804–17815
Alghamdi T, Elgazzar K, Bayoumi M et al (2019) Forecasting traffic congestion using arima modeling. In: 2019 15th international wireless communications & mobile computing conference (IWCMC), IEEE, pp 1227–1232
Shahriari S, Ghasri M, Sisson S et al (2020) Ensemble of arima: combining parametric and bootstrapping technique for traffic flow prediction. Transportmetrica A: Transport Sci 16(3):1552–1573
Kumar SV (2017) Traffic flow prediction using kalman filtering technique. Procedia Eng 187:582–587
Liu Y, Wu H (2017) Prediction of road traffic congestion based on random forest. In: 2017 10th International symposium on computational intelligence and design (ISCID), IEEE, pp 361–364
Feng X, Ling X, Zheng H et al (2018) Adaptive multi-kernel svm with spatial-temporal correlation for short-term traffic flow prediction. IEEE Trans Intell Transportation Syst 20(6):2001–2013
Chen W, An J, Li R et al (2018) A novel fuzzy deep-learning approach to traffic flow prediction with uncertain spatial-temporal data features. Future Generation Comput Syst 89:78–88
Gu Y, Lu W, Xu X et al (2019) An improved bayesian combination model for short-term traffic prediction with deep learning. IEEE Trans Intell Transportation Syst 21(3):1332–1342
Yang L, Yang Q, Li Y et al (2019) K-nearest neighbor model based short-term traffic flow prediction method. In: 2019 18th International symposium on distributed computing and applications for business engineering and science (DCABES), IEEE, pp 27–30
Song Y, Lu J (2018) Rnn-based traffic flow prediction for dynamic reversible lane control decision. In: Data Science and knowledge engineering for sensing decision support: proceedings of the 13th international FLINS conference (FLINS 2018), World Scientific, pp 323–330
Guo J, Wang Z, Chen H (2017) On-line multi-step prediction of short term traffic flow based on gru neural network. In: Proceedings of the 2nd international conference on intelligent information processing, pp 1–6
Kang D, Lv Y, Chen Yy (2017) Short-term traffic flow prediction with lstm recurrent neural network. In: 2017 IEEE 20th international conference on intelligent transportation systems (ITSC), IEEE, pp 1–6
Wu Y, Tan H (2016) Short-term traffic flow forecasting with spatial-temporal correlation in a hybrid deep learning framework. arXiv:1612.01022
Duan Z, Yang Y, Zhang K et al (2018) Improved deep hybrid networks for urban traffic flow prediction using trajectory data. Ieee Access 6:31820–31827
Defferrard M, Bresson X, Vandergheynst P (2016) Convolutional neural networks on graphs with fast localized spectral filtering. Adv Neural Inf Process Syst 29
Kipf TN, Welling M (2016) Semi-supervised classification with graph convolutional networks. arXiv:1609.02907
Hou F, Zhang Y, Fu X et al (2021) The prediction of multistep traffic flow based on ast-gcn-lstm. J Adv Transportation 2021:1–10
Chen Z, Zhao B, Wang Y et al (2020) Multitask learning and gcn-based taxi demand prediction for a traffic road network. Sensors 20(13):3776
Guo K, Hu Y, Sun Y et al (2021) Hierarchical graph convolution network for traffic forecasting. In: Proceedings of the AAAI conference on artificial intelligence, pp 151–159
Li Y, Yu R, Shahabi C et al (2017) Diffusion convolutional recurrent neural network: Data-driven traffic forecasting. arXiv:1707.01926
Guo S, Lin Y, Feng N et al (2019) Attention based spatial-temporal graph convolutional networks for traffic flow forecasting. In: Proceedings of the AAAI conference on artificial intelligence, pp 922–929
Chen Z, Lu Z, Chen Q et al (2022) Spatial-temporal short-term traffic flow prediction model based on dynamical-learning graph convolution mechanism. Inf Sci 611:522–539
Roy A, Roy KK, Ali AA et al (2021) Unified spatio-temporal modeling for traffic forecasting using graph neural network. In: 2021 International joint conference on neural networks (IJCNN), IEEE, pp 1–8
Gu J, Jia Z, Cai T et al (2023) Dynamic correlation adjacency-matrix-based graph neural networks for traffic flow prediction. Sensors 23(6):2897
Afrin T, Yodo N (2021) A probabilistic estimation of traffic congestion using bayesian network. Measurement 174:109051
Sengupta A, Mondal S, Das A et al (2024) A bayesian approach to quantifying uncertainties and improving generalizability in traffic prediction models. Transportation Research Part C: Emerging Technologies 162:104585
Pal S, Ma L, Zhang Y et al (2021) Rnn with particle flow for probabilistic spatio-temporal forecasting. In: International conference on machine learning, PMLR, pp 8336–8348
Chen C, Petty K, Skabardonis A et al (2001) Freeway performance measurement system: mining loop detector data. Transportation Res Record 1748(1):96–102
Fafoutellis P, Vlahogianni EI (2023) Unlocking the full potential of deep learning in traffic forecasting through road network representations: A critical review. Data Sci Transportation 5(3):1–21
Funding
No funds, grants, or other support was received for conducting the study.
Author information
Authors and Affiliations
Contributions
Material preparation, data collection, and analysis were performed by Atkia Akila Karim and Naushin Nower. The first draft of the manuscript was written by Atkia Akila Karim and all authors commented on previous versions of the manuscript. All authors read and approved the final manuscript.
Corresponding author
Ethics declarations
Ethical and informed consent
Ethical and informed consent for data used not applicable.
Competing interests
The authors have no competing interests to declare that are relevant to the content of this article.
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
Karim, A.A., Nower, N. Probabilistic spatio-temporal graph convolutional network for traffic forecasting. Appl Intell 54, 7070–7085 (2024). https://doi.org/10.1007/s10489-024-05562-3
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s10489-024-05562-3