DMGSTCN: Dynamic Multi-Graph Spatio-Temporal Convolution Network for Traffic Forecasting
Yanjun Qin,Xiaoming Tao,Yuchen Fang,Haiyong Luo,Fang Zhao,Chenxing Wang
DOI: https://doi.org/10.1109/jiot.2024.3380746
IF: 10.6
2024-01-01
IEEE Internet of Things Journal
Abstract: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 multi-graph spatio-temporal convolution network (DMGSTCN) in this paper, which utilizes the gated causal convolution with the dynamic multi-graph 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.
computer science, information systems,telecommunications,engineering, electrical & electronic