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.
Similar content being viewed by others
Explore related subjects
Discover the latest articles, news and stories from top researchers in related subjects.References
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
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
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
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
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
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
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
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
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
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
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
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
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
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
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
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
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)
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
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
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
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
Khairdoost N, Shirpour M, Bauer MA, Beauchemin SS (2020) Realtime driver maneuver prediction using LSTM. IEEE Trans Intell Veh 5(4):714–724
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
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
Park, C., et al. "STGRAT: a Spatio-temporal graph attention network for traffic forecasting." (2019)
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
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
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
Koschi M, Althoff M (2021) Set-based prediction of traffic participants considering occlusions and traffic rules. IEEE Trans Intell Veh 6(2):249–265
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
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
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
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
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
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
Corresponding author
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.
About this article
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
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s10489-024-05503-0