DMGSTCN: Dynamic Multigraph Spatio-Temporal Convolution Network for Traffic Forecasting

被引:8
作者
Qin, Yanjun [1 ]
Tao, Xiaoming [1 ]
Fang, Yuchen [2 ]
Luo, Haiyong [3 ]
Zhao, Fang [2 ]
Wang, Chenxing [2 ]
机构
[1] Tsinghua Univ, Dept Elect Engn, Beijing 100084, Peoples R China
[2] Beijing Univ Posts & Telecommun, Sch Comp Sci, Natl Pilot Software Engn Sch, Beijing 100876, Peoples R China
[3] Chinese Acad Sci, Beijing Key Lab Mobile Comp & Pervas Device, Inst Comp Technol, Beijing 100190, Peoples R China
基金
中国国家自然科学基金;
关键词
Time series analysis; Convolution; Forecasting; Correlation; Task analysis; Roads; Predictive models; Graph convolution network (GCN); spatial-temporal data; traffic forecasting; PREDICTION; FLOW;
D O I
10.1109/JIOT.2024.3380746
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Traffic forecasting belongs to intelligent transportation systems and is helpful for public property and life safety. Therefore, to forecast traffic accurately, researchers pay great attention to dealing with complex problems by mining intricate spatial and temporal dependencies of the traffic. However, some challenges still hold back traffic forecasting: 1) Most studies mainly focus on modeling correlations of traffic time series of close distances on the road network and ignore correlations of remote but similar traffic time series; 2) Previous static graph-based methods failed to reflect the dynamic changed spatial relations of multiple time series in the evolving traffic system. To tackle the above issues, we design a new dynamic multigraph spatio-temporal convolution network (DMGSTCN) in this article, which utilizes the gated causal convolution with the dynamic multigraph convolution network (DMGCN) to simultaneously extract spatial and temporal information. Specifically, DMGCN uses not only distance-based graphs but also structure-based graphs to obtain spatial information from nearby and remote but similar traffic time series, respectively. Moreover, to dynamically model spatial correlations, DMGCN first splits neighbors of each traffic time series into different regions according to relative position relationships. Then DMGCN assigns different weights to different regions at different time slices. Empirical evaluations on four traffic forecasting benchmarks reveal that DMGSTCN outperforms existing methods.
引用
收藏
页码:22208 / 22219
页数:12
相关论文
共 62 条
[1]  
Atwood J, 2016, ADV NEUR IN, V29
[2]  
Bahdanau D, 2016, Arxiv, DOI arXiv:1409.0473
[3]  
Bai L, 2020, ADV NEUR IN, V33
[4]  
Berndt DonaldJ., 1994, KDD WORKSHOP, P359, DOI DOI 10.5555/3000850.3000887
[5]  
Bruna J., 2013, SPECTRAL NETWORKS LO
[6]  
Chen Ming., 2020, INT C MACHINE LEARNI, P1725
[7]  
Chen WQ, 2020, AAAI CONF ARTIF INTE, V34, P3529
[8]  
Chen Yiming, 2021, P MACHINE LEARNING R, V139
[9]  
Choi J, 2022, AAAI CONF ARTIF INTE, P6367
[10]   Towards Spatio-Temporal Aware Traffic Time Series Forecasting [J].
Cirstea, Razvan-Gabriel ;
Yang, Bin ;
Guo, Chenjuan ;
Tung Kieu ;
Pan, Shirui .
2022 IEEE 38TH INTERNATIONAL CONFERENCE ON DATA ENGINEERING (ICDE 2022), 2022, :2900-2913