Traffic Prediction Based on Multi-graph Spatio-Temporal Convolutional Network

Xiaomin Yao,Zhenguo Zhang,Rongyi Cui,Yahui Zhao
DOI: https://doi.org/10.1007/978-3-030-87571-8_13
2021-01-01
Abstract:In the Intelligent Traffic System (ITS), accurate prediction of the state of the traffic at the next moment is of great significance to transportation planning. Existing researches mainly focus on the research of the topological structure of the road network in space, and consider the temporal dependence at the same time. However, we have noticed that it is not only important to consider the dependence of time and space at the same time, but also other organizational relationships between the road network will also affect the forecast results. In this paper, we reconsider the correlation between roads and capture their correlation in both space and time. More specifically, we first encode the road network into two graphs (connect graph and similar graph) based on the connectivity and historical pattern similarity, and merge the graphs to make the connected edges carry more information. Then graph convolution is used for the fused graph. In order to capture the temporal dependence, we use one-dimensional convolution to first convolve the information before the graph convolution, and then perform the one-dimensional convolution on the information after the fusion graph, which can achieve fast spatial-state propagation from graph convolution through fast spatial-state propagation. We evaluate the predictive performance of our model by real-world traffic dataset and experiments prove that the addition of multi-graph is effective. Compared with the relatively new baseline, RMSE has dropped by about 10\(\%\).
What problem does this paper attempt to address?