LSGCN: Long Short-Term Traffic Prediction with Graph Convolutional Networks

LSGCN: Long Short-Term Traffic Prediction with Graph Convolutional Networks

Rongzhou Huang, Chuyin Huang, Yubao Liu, Genan Dai, Weiyang Kong

Proceedings of the Twenty-Ninth International Joint Conference on Artificial Intelligence
Main track. Pages 2355-2361. https://doi.org/10.24963/ijcai.2020/326

Traffic prediction is a classical spatial-temporal prediction problem with many real-world applications such as intelligent route planning, dynamic traffic management, and smart location-based applications. Due to the high nonlinearity and complexity of traffic data, deep learning approaches have attracted much interest in recent years. However, few methods are satisfied with both long and short-term prediction tasks. Target at the shortcomings of existing studies, in this paper, we propose a novel deep learning framework called Long Short-term Graph Convolutional Networks (LSGCN) to tackle both traffic prediction tasks. In our framework, we propose a new graph attention network called cosAtt, and integrate both cosAtt and graph convolution networks (GCN) into a spatial gated block. By the spatial gated block and gated linear units convolution (GLU), LSGCN can efficiently capture complex spatial-temporal features and obtain stable prediction results. Experiments with three real-world traffic datasets verify the effectiveness of LSGCN.
Keywords:
Machine Learning: Deep Learning
Data Mining: Mining Spatial, Temporal Data