GSTNet: Global Spatial-Temporal Network for Traffic Flow Prediction

GSTNet: Global Spatial-Temporal Network for Traffic Flow Prediction

Shen Fang, Qi Zhang, Gaofeng Meng, Shiming Xiang, Chunhong Pan

Proceedings of the Twenty-Eighth International Joint Conference on Artificial Intelligence
Main track. Pages 2286-2293. https://doi.org/10.24963/ijcai.2019/317

Predicting traffic flow on traffic networks is a very challenging task, due to the complicated and dynamic spatial-temporal dependencies between different nodes on the network. The traffic flow renders two types of temporal dependencies, including short-term neighboring and long-term periodic dependencies. What's more, the spatial correlations over different nodes are both local and non-local. To capture the global dynamic spatial-temporal correlations, we propose a Global Spatial-Temporal Network (GSTNet), which consists of several layers of spatial-temporal blocks. Each block contains a multi-resolution temporal module and a global correlated spatial module in sequence, which can simultaneously extract the dynamic temporal dependencies and the global spatial correlations. Extensive experiments on the real world datasets verify the effectiveness and superiority of the proposed method on both the public transportation network and the road network.
Keywords:
Machine Learning: Data Mining
Machine Learning: Time-series;Data Streams
Machine Learning: Deep Learning
Machine Learning Applications: Networks
Multidisciplinary Topics and Applications: Transportation