Skip to main content
Log in

Probabilistic spatio-temporal graph convolutional network for traffic forecasting

  • Published:
Applied Intelligence Aims and scope Submit manuscript

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

This is a preview of subscription content, log in via an institution to check access.

Access this article

Subscribe and save

Springer+ Basic
$34.99 /Month
  • Get 10 units per month
  • Download Article/Chapter or eBook
  • 1 Unit = 1 Article or 1 Chapter
  • Cancel anytime
Subscribe now

Buy Now

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Fig. 1
Fig. 2
Fig. 3
Algorithm 1
Fig. 4
Fig. 5
Fig. 6
Fig. 7
Fig. 8
Fig. 9
Fig. 10

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

  1. 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

  2. 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)

  3. 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

  4. 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

    Article  Google Scholar 

  5. 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

    Article  Google Scholar 

  6. 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

    Article  Google Scholar 

  7. 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

    Article  Google Scholar 

  8. 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

  9. 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

    Google Scholar 

  10. 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

  11. 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

    Article  Google Scholar 

  12. Kumar SV (2017) Traffic flow prediction using kalman filtering technique. Procedia Eng 187:582–587

    Article  Google Scholar 

  13. 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

  14. 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

    Article  Google Scholar 

  15. 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

    Article  Google Scholar 

  16. 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

    Article  Google Scholar 

  17. 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

  18. 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

  19. 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

  20. 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

  21. Wu Y, Tan H (2016) Short-term traffic flow forecasting with spatial-temporal correlation in a hybrid deep learning framework. arXiv:1612.01022

  22. 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

    Article  Google Scholar 

  23. Defferrard M, Bresson X, Vandergheynst P (2016) Convolutional neural networks on graphs with fast localized spectral filtering. Adv Neural Inf Process Syst 29

  24. Kipf TN, Welling M (2016) Semi-supervised classification with graph convolutional networks. arXiv:1609.02907

  25. 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

    Article  Google Scholar 

  26. 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

    Article  Google Scholar 

  27. 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

  28. Li Y, Yu R, Shahabi C et al (2017) Diffusion convolutional recurrent neural network: Data-driven traffic forecasting. arXiv:1707.01926

  29. 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

  30. 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

    Article  Google Scholar 

  31. 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

  32. 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

    Article  Google Scholar 

  33. Afrin T, Yodo N (2021) A probabilistic estimation of traffic congestion using bayesian network. Measurement 174:109051

    Article  Google Scholar 

  34. 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

  35. 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

  36. Chen C, Petty K, Skabardonis A et al (2001) Freeway performance measurement system: mining loop detector data. Transportation Res Record 1748(1):96–102

    Article  Google Scholar 

  37. 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

    Article  Google Scholar 

Download references

Funding

No funds, grants, or other support was received for conducting the study.

Author information

Authors and Affiliations

Authors

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

Correspondence to Atkia Akila Karim.

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.

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

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

Download citation

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s10489-024-05562-3

Keywords

Navigation