Dynamic Global-Local Spatial-Temporal Network for Traffic Speed Prediction

被引:15
作者
Feng, Dong [1 ,2 ,3 ]
Wu, Zhongcheng [1 ,2 ]
Zhang, Jun [1 ,2 ,3 ]
Wu, Ziheng [4 ]
机构
[1] Chinese Acad Sci, Hefei Inst Phys Sci, High Magnet Field Lab, Hefei 230031, Peoples R China
[2] Univ Sci & Technol China, Sch Hefei Inst Phys Sci, Hefei 230026, Peoples R China
[3] High Magnet Field Lab Anhui Prov, Hefei 230031, Peoples R China
[4] Anhui Univ Technol, Sch Elect & Informat Engn, Maanshan 243000, Peoples R China
关键词
Correlation; Convolution; Feature extraction; Predictive models; Forecasting; Market research; Data models; Traffic speed prediction; spatial-temporal network; graph convolutional network; dynamic graph learning; MODEL;
D O I
10.1109/ACCESS.2020.3038380
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Predicting traffic speed accurately is a very challenging task of the intelligent traffic system (ITS), due to the complex and dynamic spatial-temporal dependencies from both temporal and spatial aspects. There not only exits short-term local neighboring fluctuation and long-term global trend in temporal aspect, but also local and global correlations in spatial aspect. Most existing work focus on the local spatial-temporal dependencies, ignoring the global dynamic spatial-temporal corrections, which is comparably critical for traffic speed prediction. To address this problem, we propose a novel Dynamic Global-Local Spatial-Temporal Network(DGLSTNet) for traffic speed prediction, which consists of multiple spatial-temporal module considering the local and global information simultaneously from both temporal and spatial perspective. Each temporal module applies stacked dilated convolution block to exploit multi-scale local temporal information. Moreover, we empoly a global temporal attention block to capture global dependencies of temporal domain in an attention mechanism. In each spatial module, we not only learn the local but also focus on dynamic global spatial information learned by dymamic graph learning block. Combining the feature results from local and global perspective, the capability and expressiveness of traffic predicting model is improved. Experiment results on two real-world traffic datasets have demonstrated that our proposed model can effectively capture the comprehensive spatial-temporal dependencies and can achieve state-of-the-art prediction performance compared with the existing works.
引用
收藏
页码:209296 / 209307
页数:12
相关论文
共 37 条
[1]  
[Anonymous], 2019, ARXIV190600121
[2]  
[Anonymous], 2016, P SSW
[3]  
[Anonymous], ARXIV181010237
[4]  
[Anonymous], 2016, J. Mach. Learn. Res.
[5]   A spatiotemporal correlative k-nearest neighbor model for short-term traffic multistep forecasting [J].
Cai, Pinlong ;
Wang, Yunpeng ;
Lu, Guangquan ;
Chen, Peng ;
Ding, Chuan ;
Sun, Jianping .
TRANSPORTATION RESEARCH PART C-EMERGING TECHNOLOGIES, 2016, 62 :21-34
[6]   Dynamic travel time prediction with real-time and historic data [J].
Chien, SIJ ;
Kuchipudi, CM .
JOURNAL OF TRANSPORTATION ENGINEERING, 2003, 129 (06) :608-616
[7]  
Cui Z., 2018, ARXIV
[8]   Traffic Graph Convolutional Recurrent Neural Network: A Deep Learning Framework for Network-Scale Traffic Learning and Forecasting [J].
Cui, Zhiyong ;
Henrickson, Kristian ;
Ke, Ruimin ;
Wang, Yinhai .
IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2020, 21 (11) :4883-4894
[9]  
Defferrard M, 2016, ADV NEUR IN, V29
[10]  
Du S., 2018, ARXIV180302099