Multi-View SpatialTemporal Graph Convolutional Network for Traffic Prediction

被引:6
作者
Wei, Shuqing [1 ]
Feng, Siyuan [2 ]
Yang, Hai [3 ,4 ]
机构
[1] Hong Kong Univ Sci & Technol, Div Emerging Interdisciplinary Area, Hong Kong, Peoples R China
[2] Hong Kong Polytech Univ, Dept Logist & Maritime Studies, Hong Kong, Peoples R China
[3] Hong Kong Univ Sci & Technol, Dept Civil & Environm Engn, Hong Kong, Peoples R China
[4] Hong Kong Univ Sci & Technol Guangzhou, Intelligent Transportat Thrust, Guangzhou 511453, Peoples R China
关键词
Roads; Correlation; Public transportation; Convolutional neural networks; Predictive models; Data models; Computational modeling; Attention mechanism; multi-view learning; spatial-temporal graph convolutional network; traffic prediction;
D O I
10.1109/TITS.2024.3364759
中图分类号
TU [建筑科学];
学科分类号
0813 ;
摘要
Multi-step traffic speed prediction is a challenging issue due to the multiple spatial-temporal dependencies among roads. Some spatial dependencies, especially those formed by different traffic modes, are not fully exploited, and how to simultaneously consider spatial and temporal dependencies and effectively integrate them within a single prediction framework needs further exploration. To tackle the above issues, we propose a multi-view spatial-temporal graph convolutional framework MVSTG, which adequately exploits the multi-view spatial-temporal dependencies and their interactions to improve the accuracy of traffic prediction. Multi-view temporal learning captures the multiple temporal trends by temporal convolution from multi-granularity historical data, and multi-view spatial learning handles the multiple spatial correlations by graph convolution from multiple graphs. In addition, view-wise attention-based fusion is proposed to adaptively identify the importance of each upstream view, fuse the multi-view information, and generate integrated results for downstream views. The experiments on two real-world urban traffic datasets demonstrate that the multi-view data and the proposed model framework enhance performance on the accuracy of speed prediction, especially in mid-term and long-term prediction.
引用
收藏
页码:9572 / 9586
页数:15
相关论文
共 65 条
[1]   A Neural Network Architecture Combining Gated Recurrent Unit (GRU) and Support Vector Machine (SVM) for Intrusion Detection in Network Traffic Data [J].
Agarap, Abien Fred M. .
PROCEEDINGS OF 2018 10TH INTERNATIONAL CONFERENCE ON MACHINE LEARNING AND COMPUTING (ICMLC 2018), 2018, :26-30
[2]  
Ahmed A. R., 1979, Analysis of Freeway Traffic Time-SeriesData By Using Box-Jenkins Techniques, V722
[3]  
[Anonymous], 2019, Res. C, Emerg. Technol., V105, P1929
[4]  
Ba J, 2014, ACS SYM SER
[5]   A3T-GCN: Attention Temporal Graph Convolutional Network for Traffic Forecasting [J].
Bai, Jiandong ;
Zhu, Jiawei ;
Song, Yujiao ;
Zhao, Ling ;
Hou, Zhixiang ;
Du, Ronghua ;
Li, Haifeng .
ISPRS INTERNATIONAL JOURNAL OF GEO-INFORMATION, 2021, 10 (07)
[6]  
Bai SJ, 2018, Arxiv, DOI [arXiv:1803.01271, DOI 10.48550/ARXIV.1803.01271]
[7]  
Cao V. O. K., 2020, IEEE 91 VEH TECHNOL, P1
[8]  
Chen WQ, 2020, AAAI CONF ARTIF INTE, V34, P3529
[9]   Short-term traffic forecasting: An adaptive ST-KNN model that considers spatial heterogeneity [J].
Cheng, Shifen ;
Lu, Feng ;
Peng, Peng ;
Wu, Sheng .
COMPUTERS ENVIRONMENT AND URBAN SYSTEMS, 2018, 71 :186-198
[10]  
Cho KYHY, 2014, Arxiv, DOI arXiv:1409.1259