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

PDF BibTeX

Abstract

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
Welcome Future Conferences AI Journal Division IJCAI Proceedings Travel Grant Program IJCAI Awards