Multistep traffic speed prediction from multiple time-scale spatiotemporal features using graph attention network | Applied Intelligence Skip to main content

Advertisement

Log in

Multistep traffic speed prediction from multiple time-scale spatiotemporal features using graph attention network

  • Published:
Applied Intelligence Aims and scope Submit manuscript

Abstract

Traffic forecasting using deep learning represents a crucial aspect of intelligent transportation systems, carrying substantial implications for congestion reduction and efficient route planning. Despite its significance, accurately predicting traffic states remains a challenge. Existing methodologies focus on capturing the temporal trends of traffic states and the spatial dependencies between roads to enhance prediction accuracy. However, two noteworthy limitations persist in these approaches: (1) Many models neglect the interaction between spatiotemporal features across varying time spans, hindering their ability to utilize traffic state information effectively for predicting future conditions. (2) Genuine correlations between roads are time-varying, making it inadequate to rely on static graphs or static pre-trained node embeddings to model dynamic correlations between roads. To address these challenges, we propose the Multiple Time-Scale Graph Attention Network (MTS-GATN), which comprises two key modules: the Multiple Time-Scale Spatiotemporal Features Extraction Module and the Feature Augmentation Module. The first module involves stacking multiple spatiotemporal extraction layers to discern traffic state information at different time scales. In the second module, we employ dynamic spatial semantic embedding for feature augmentation, providing nodes with dynamic representations over time. Subsequently, we leverage a multi-head spatiotemporal attention mechanism to comprehensively consider location information and real-time semantic data, facilitating the interaction of traffic state information across multiple time scales. Experimental results on two distinct traffic datasets validate the superior performance of MTS-GATN in medium-term and long-term forecasting scenarios.

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

Access this article

Subscribe and save

Springer+ Basic
¥17,985 /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 includes VAT (Japan)

Instant access to the full article PDF.

Fig. 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6
Fig. 7
Fig. 8

Similar content being viewed by others

Explore related subjects

Discover the latest articles, news and stories from top researchers in related subjects.

References

  1. Zhang J, Wang F, Wang K, Lin W, Xu X, Chen C (2011) Data-driven intelligent transportation systems: A survey. IEEE Trans Intell Transp Syst 12(4):1624–1639

    Article  Google Scholar 

  2. Guo S, Lin Y, Li S, Chen Z, Wan H (2019) Deep spatial–temporal 3D convolutional neural networks for traffic data forecasting. IEEE Trans Intell Transp Syst 20(10):3913–3926

    Article  Google Scholar 

  3. Hong W, Chakraborty I, Wang H, Tao G (2021) Co-optimization scheme for the powertrain and exhaust emission control system of hybrid electric vehicles using future speed prediction. IEEE Tran Intell Veh 6(3):533–545

    Article  Google Scholar 

  4. Guo G, Yuan W, Lv Y, Liu W, Liu J (2023) Traffic forecasting via dilated temporal convolution with peak-sensitive loss. IEEE Trans Intell Transp Syst 15(1):48–57

    Google Scholar 

  5. Kumar R, Swarnkar M, Singal G, Kumar N (2022) IoT network traffic classification using machine learning algorithms: an experimental analysis. IEEE Internet Things J 9(2):989–1008

    Article  Google Scholar 

  6. Singh R, Saluja D, Kumar S (2022) R-Comm: a traffic based approach for joint vehicular radar-communication. IEEE Trans Intell Veh 7(1):83–92

    Article  Google Scholar 

  7. Feng X, Ling X, Zheng H, Chen Z, Xu Y (2019) Adaptive multikernel SVM with spatial–temporal correlation for short-term traffic flow prediction. IEEE Trans Intell Transp Syst 20(6):2001–2013

    Article  Google Scholar 

  8. Zhou Q, Yang T, Zhong DC, Zhang N (2021) Variational graph neural networks for road traffic prediction in intelligent transportation systems. IEEE Trans Ind Informat 17(4):2802–2812

    Article  Google Scholar 

  9. Zhang J, Zheng Y, Qi D (2017) Deep spatio-temporal residual networks for citywide crowd flows prediction. In: Proc., 31st AAAI Conf.On artificial intelligence. Association for the Advancement of Artificial Intelligence, Palo Alto, CA. https://doi.org/10.1609/aaai.v31i1.10735

    Chapter  Google Scholar 

  10. Yan S, Xiong Y, Lin D (2018) Spatial temporal graph convolutional networks for skeleton-based action recognition. In: Proc., 32nd AAAI Conf. On artificial intelligence. Association for the Advancement of Artificial Intelligence, Palo Alto, CA. https://doi.org/10.1609/aaai.v32i1.12328

    Chapter  Google Scholar 

  11. Yao H, Wu F, Ke J, Tang X, Jia Y, Lu S, Gong P, Li Z, Ye J, Chuxing D (2018) Deep multi-view spatial-temporal network for taxi demand prediction. In: Proc., 32nd AAAI Conf. On artificial intelligence. Association for the Advancement of Artificial Intelligence, Palo Alto, CA. https://doi.org/10.1609/aaai.v32i1.11836

    Chapter  Google Scholar 

  12. Seo Y, Defferrard M, Vandergheynst P, Bresson X (2018) Structured sequence modeling with graph convolutional recurrent networks. In: Cheng L, Leung A, Ozawa S (eds) Vol. 11301 of Lecture notes in computer science. https://doi.org/10.1007/978-3-030-04167-0_33

    Chapter  Google Scholar 

  13. Li Y, Yu R, Shahabi C, Liu Y (2018) Diffusion convolutional recurrent neural network: Data-driven traffic forecasting. In: In Proc., 6th Int. Conf. on Learning Representations. International Conference on Learning Representations, La Jolla, CA

    Google Scholar 

  14. Yu B, Yin H, Zhu Z (2018) Spatio-temporal graph convolutional networks: A deep learning framework for traffic forecasting. In: Proc., IJCAI Int. Joint Conf. on Artificial Intelligence. https://arxiv.org/abs/1709.04875

    Google Scholar 

  15. Song C et al (2020) Spatial-temporal synchronous graph convolutional networks: a new framework for spatial-temporal network data forecasting. In: Proc., 34st AAAI Conf.On Artificial Intelligence pp, pp 914–921. https://doi.org/10.1609/aaai.v34i01.5438

    Chapter  Google Scholar 

  16. Zheng C et al (2020) GMAN: a graph multi-attention network for traffic prediction. In: Proc., 34st AAAI Conf.On Artificial Intelligence pp, pp 1234–1241. https://doi.org/10.1609/aaai.v34i01.5477

    Chapter  Google Scholar 

  17. Williams BM, Hoe LA (2003) Modeling and forecasting vehicular traffic flow as a seasonal ARIMA process: theoretical basis and empirical results. J Transp Eng. https://doi.org/10.1061/(ASCE)0733-947X(2003)129:6(664)

  18. Jeong YS, Byon YJ, Castro-Neto MM, Easa SM (2013) Supervised weighting-online learning algorithm for short-term traffic flow prediction. IEEE Trans Intell Transp Syst 14(4):1700–1707

    Article  Google Scholar 

  19. Fusco G, Colombaroni C, Isaenko N (2016) Short-term speed predictions exploiting big data on large urban road networks. Transport Res Part C: Emerg Technol 73:183–201. https://doi.org/10.1016/j.trc.2016.10.019

    Article  Google Scholar 

  20. Gräber T, Lupberger S, Unterreiner M, Schramm D (2019) A hybrid approach to side-slip angle estimation with recurrent neural networks and kinematic vehicle models. IEEE Trans Intell Veh 4(1):39–47

    Article  Google Scholar 

  21. Saleh K, Hossny M, Nahavandi S (2018) Intent prediction of pedestrians via motion trajectories using stacked recurrent neural networks. IEEE Trans Intell Veh 3(4):414–424

    Article  Google Scholar 

  22. Khairdoost N, Shirpour M, Bauer MA, Beauchemin SS (2020) Realtime driver maneuver prediction using LSTM. IEEE Trans Intell Veh 5(4):714–724

    Article  Google Scholar 

  23. Fu R, Zhang Z, Li L (2017) Using LSTM and GRU neural network methods for traffic flow prediction. In: 2016 31st Youth academic annual conference of Chinese association of automation (YAC), pp 324–328. https://doi.org/10.1109/YAC.2016.7804912

    Chapter  Google Scholar 

  24. Wu Z, Pan S, Long G, Jiang J, Zhang C (2019) Graph Wavenet for Deep Spatial-Temporal Graph Modeling. In: IJCAI International Joint Conference on Artificial Intelligence 2019-Augus: 1907–13. https://doi.org/10.24963/ijcai.2019/264

    Chapter  Google Scholar 

  25. Park, C., et al. "STGRAT: a Spatio-temporal graph attention network for traffic forecasting." (2019)

    Google Scholar 

  26. Wu X, Fang J, Liu Z et al (2021) Multistep traffic speed prediction from spatial-temporal dependencies using graph neural networks. J Transport Eng, Part A Syst 147(12):04021082

    Article  Google Scholar 

  27. Bloemheuvel S, Hoogen JVD, Jozinovi D et al (2022) Multivariate Time Series Regression with Graph Neural Networks. https://doi.org/10.48550/arXiv.2201.00818

  28. Bai L, Yao L, Li C, Wang X, Wang C (2020) Adaptive graph convolutional recurrent network for traffic forecasting. In: Advances in neural information processing systems 33: annual conference on neural information processing systems 2020. NeurIPS 2020

    Google Scholar 

  29. Koschi M, Althoff M (2021) Set-based prediction of traffic participants considering occlusions and traffic rules. IEEE Trans Intell Veh 6(2):249–265

    Article  Google Scholar 

  30. Lienke C, Wissing C, Keller M, Nattermann T, Bertram T (2019) Predictive driving: fusing prediction and planning for automated highway driving. IEEE Trans Intell Veh 4(3):456–467

    Article  Google Scholar 

  31. Wang Z, Su X, Ding Z (2021) Long-term traffic prediction based on LSTM encoder-decoder architecture. IEEE Trans Intell Transp Syst 22(10):6561–6571

    Article  Google Scholar 

  32. Li Y et al (2022) Dense skip attention based deep learning for day-ahead electricity price forecasting. IEEE Trans Power Syst. https://doi.org/10.1109/TPWRS.2022.3217579

  33. Yu B, Yin H, Zhu Z (2018) Spatio-temporal graph convolutional networks: a deep learning framework for Trafffc forecasting. In: Proc. the international joint conference on artificial intelligence (IJCAI) https://arxiv.org/abs/1709.04875

    Google Scholar 

  34. Meng L, Masuda N. Analysis of node2vec random walks on networks Proc Royal Soc A Eng Sci DOI:https://doi.org/10.1098/rspa.2020.0447, 2020

Download references

Acknowledgments

This work was supported by the National Natural Science Foundation of China(grant entitled “Connected vehicle big data driven expressway multi-objective coordinated control fusing deep learning and traffic flow model”, award number 466 71901070).

Author information

Authors and Affiliations

Authors

Corresponding author

Correspondence to Mengyun Xu.

Ethics declarations

The authors have no competing interests to declare that are relevant to the content of this article. The authors have no relevant financial or non-financial interests to disclose. All authors certify that they have no affliations with or involvement in any organization or entity with any financial interest or non-fnancial interest in the subject matter or materials discussed in this manuscript.

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

Fang, J., Wu, Z., Xu, M. et al. Multistep traffic speed prediction from multiple time-scale spatiotemporal features using graph attention network. Appl Intell 54, 7479–7492 (2024). https://doi.org/10.1007/s10489-024-05503-0

Download citation

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s10489-024-05503-0

Keywords

Navigation