Spatial-Temporal Graph Convolutional Gated Recurrent Network for Traffic Forecasting

by   Le Zhao, et al.

As an important part of intelligent transportation systems, traffic forecasting has attracted tremendous attention from academia and industry. Despite a lot of methods being proposed for traffic forecasting, it is still difficult to model complex spatial-temporal dependency. Temporal dependency includes short-term dependency and long-term dependency, and the latter is often overlooked. Spatial dependency can be divided into two parts: distance-based spatial dependency and hidden spatial dependency. To model complex spatial-temporal dependency, we propose a novel framework for traffic forecasting, named Spatial-Temporal Graph Convolutional Gated Recurrent Network (STGCGRN). We design an attention module to capture long-term dependency by mining periodic information in traffic data. We propose a Double Graph Convolution Gated Recurrent Unit (DGCGRU) to capture spatial dependency, which integrates graph convolutional network and GRU. The graph convolution part models distance-based spatial dependency with the distance-based predefined adjacency matrix and hidden spatial dependency with the self-adaptive adjacency matrix, respectively. Specially, we employ the multi-head mechanism to capture multiple hidden dependencies. In addition, the periodic pattern of each prediction node may be different, which is often ignored, resulting in mutual interference of periodic information among nodes when modeling spatial dependency. For this, we explore the architecture of model and improve the performance. Experiments on four datasets demonstrate the superior performance of our model.


page 3

page 4


Attention-based Spatial-Temporal Graph Convolutional Recurrent Networks for Traffic Forecasting

Traffic forecasting is one of the most fundamental problems in transport...

Modeling Spatial-Temporal Dynamics for Traffic Prediction

Spatial-temporal prediction has many applications such as climate foreca...

Spatial-Temporal Gating-Adjacency GCN for Human Motion Prediction

Predicting future motion based on historical motion sequence is a fundam...

3D Randomized Connection Network with Graph-based Label Inference

In this paper, a novel 3D deep learning network is proposed for brain MR...

PSTN: Periodic Spatial-temporal Deep Neural Network for Traffic Condition Prediction

Accurate forecasting of traffic conditions is critical for improving saf...

Meta Attentive Graph Convolutional Recurrent Network for Traffic Forecasting

Traffic forecasting is a fundamental problem in intelligent transportati...

Graph WaveNet for Deep Spatial-Temporal Graph Modeling

Spatial-temporal graph modeling is an important task to analyze the spat...

Please sign up or login with your details

Forgot password? Click here to reset