Traffic Flow Prediction Based on Interactive Dynamic Spatio-Temporal Graph Convolution with a Probabilistic Sparse Attention Mechanism

Abstract

Accurate traffic flow prediction is of great practical significance to alleviate road congestion. Existing methods ignore the hidden dynamic associations between road nodes, and for the problem of difficulty capturing the dynamic spatio-temporal features of traffic flow in the prediction process, a novel model based on the interactive dynamic spatio-temporal graph convolutional probabilistic sparse attention mechanism (IDG-PSAtt) is proposed, which consists of an interactive dynamic graph convolutional network (IDGCN) structure with a spatio-temporal convolutional block (ST-Conv block) and a probabilistic sparse self-attention mechanism block (ProbSSAtt block). Among them, the IDGCN synchronizes the dynamic spatio-temporal features captured by interaction sharing, and the ST-Conv block is combined with the ProbSSAtt block to effectively capture the long short-term temporal features of the traffic flow. In addition, to effectively find the hidden dynamic associations between road network nodes, a dynamic graph convolutional network generated by the fusion of an adaptive neighbor matrix and a learnable neighbor matrix was constructed. Experimental results demonstrate that the prediction performance of the IDG-PSAtt model outperforms the baseline model under the evaluation criteria and experimental settings given in this paper. In the PEMS-BAY dataset, the mean absolute error and root mean square error of the IDG-PSAtt to 60 min are improved by 15.49% and 12.10%, compared with the state-of-the-art model, respectively.

Keywords

data and data science artificial intelligence and advanced computing applications deep learning machine learning (artificial intelligence)neural networks

Get full access to this article

View all access options for this article.

References

Zhao

. How to Build a Graph-Based Deep Learning Architecture in Traffic Domain: A Survey. IEEE Transactions on Intelligent Transportation Systems, Vol. 23, No. 5, 2020, pp. 3904–3924.

Lasenby

. Spatiotemporal Attention-Based Graph Convolution Network for Segment-Level Traffic Prediction. IEEE Transactions on Intelligent Transportation Systems, Vol. 23, No. 7, 2022, pp. 8337–8345.

Zhang

Chang

Meng

Xiang

Pan

. Spatio-Temporal Graph Structure Learning for Traffic Forecasting. Proc. AAAI Conference on Artificial Intelligence, Hawaii, Vol. 34, No. 1, 2019, pp. 1177–1185.

Zhu

. Spatial-Temporal Fusion Graph Neural Networks for Traffic Flow Forecasting. Proc. AAAI Conference on Artificial Intelligence, Vancouver, Canada, Vol. 35, No. 5, 2021, pp. 4189–4196.

Zhao

Tanin

Cui

Nassir

Sarvi

. A Graph and Attentive Multi-Path Convolutional Network for Traffic Prediction. IEEE Transactions on Knowledge and Data Engineering, Vol. 35, 2022, pp. 6548–6560.

Guo

Lin

Wan

Cong

. Learning Dynamics and Heterogeneity of Spatial-Temporal Graph Data for Traffic Forecasting. IEEE Transactions on Knowledge and Data Engineering, Vol. 99, 2021, pp. 5415–5428.

Bai

Yao

Wang

. Adaptive Graph Convolutional Recurrent Network for Traffic Forecasting. Advances in Neural Information Processing Systems, Vol. 33, 2020, pp. 17804–17815.

Tian

Zou

Zhang

. NA-DGRU: A Dual-GRU Traffic Speed Prediction Model Based on Neighborhood Aggregation and Attention Mechanism. Sustainability, Vol. 15, No. 4, 2023, p. 2927.

Tian

Zhang

. MAT-WGCN: Traffic Speed Prediction Using Multi-Head Attention Mechanism and Weighted Adjacency Matrix. Sustainability, Vol. 15, No. 7, 2023, p. 13080.

10.

Chen

Xie

Cao

Gao

Feng

. Multirange Attentive Bicomponent Graph Convolutional Network for Traffic Forecasting. Proc. AAAI Conference on Artificial Intelligence, Vol. 34, No. 04, 2020, pp. 3529–3536.

11.

Liu

Wang

Zhu

. Short-Term Traffic Speed Forecasting Based on Attention Convolutional Neural Network for Arterials. Computer-Aided Civil and Infrastructure Engineering, Vol. 33, No. 11, 2018, pp. 999–1016.

12.

Deng

Zhang

Tao

Zhao

Liu

. A Spatiotemporal Graph Convolution-Based Model for Daily Runoff Prediction in a River Network with Non-Euclidean Topological Structure. Stochastic Environmental Research and Risk Assessment, Vol. 37, 2023, pp. 1457–1478.

13.

Pan

Long

Jiang

Chang

Zhang

. Connecting the Dots: Multivariate Time Series Forecasting With Graph Neural Networks. Proc. 26th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining, New York, 2020, pp. 753–763.

14.

Yan

Wang

Jin

Zhang

. Spatial-Temporal Chebyshev Graph Neural Network for Traffic Flow Prediction in IoT-Based ITS. IEEE Internet of Things Journal, Vol. 9, No. 12, 2022, pp. 9266–9279.

15.

Chen

Zhong

Zhang

Xue

. Spatial-Temporal Short-Term Traffic Flow Prediction Model Based on Dynamical-Learning Graph Convolution Mechanism. Information Sciences, Vol. 611, 2022, pp. 522–539.

16.

Chen

Zhang

. Adaptive Rolling Smoothing with Heterogeneous Data for Traffic State Estimation and Prediction. IEEE Transactions on Intelligent Transportation Systems, Vol. 20, No. 4, 2019, pp. 1247–1258.

17.

Xiao

Yang

Mao

Wen

. An Improved Seasonal Rolling Grey Forecasting Model Using a Cycle Truncation Accumulated Generating Operation for Traffic Flow. Applied Mathematical Modelling, Vol. 51, 2017, pp. 386–404.

18.

Zheng

Chai

W. K.

Katos

. A Dynamic Spatial-Temporal Deep Learning Framework for Traffic Speed Prediction on Large-Scale Road Networks. Expert Systems with Application, Vol. 195, 2022, p. 116585.

19.

Zheng

Fan

Wang

. GMAN: A Graph Multi-Attention Network for Traffic Prediction. Proc. AAAI Conference on Artificial Intelligence, New YorkHilton Midtown, New York, 2020, pp. 1234–1241.

20.

Zhang

Chen

Cao

Zhao

Kan

. Research on Traffic Flow Forecasting Based on Spatio-Temporal Convolutional Network. Quest Journals, Vol. 7, 2021, pp. 6–19.

21.

Divyam

Saxena Singh

Rai

. Comparative Study of Static VAR Compensation Techniques-Thyristor Switched Reactor and Thyristor Switched Capacitor. International Conference on Communication and Electronics Systems (ICCES), montenegro, 2020, pp. 1–5.

22.

Tan

. Short-Term Traffic Flow Forecasting with Spatial-Temporal Correlation in a Hybrid Deep Learning Framework, ArXiv Preprint, 2016.

23.

Yan

. Learning Dynamic and Hierarchical Traffic Spatiotemporal Features with Transformer. IEEE Transactions on Intelligent Transportation Systems, Vol. 23, No. 11, 2021, pp. 22386–22399.

24.

Rio

Momey

Ducottet

Alata

. WaveNet Based Architectures for Denoising Periodic Discontinuous Signals and Application to Friction Signals. 2020 28th European Signal Processing Conference (EUSIPCO), Netherlands, Amsterdam, 2021, pp. 1580–1584.

25.

Song

Lin

Guo

Wan

. Spatial-Temporal Synchronous Graph Convolutional Networks: A New Framework for Spatial-Temporal Network Data Forecasting. Proc. AAAI Conference on Artificial Intelligence, Vol. 34, No. 01, 2020, pp. 914–921.