Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

3D Graph Convolutional Networks with Temporal Graphs: A Spatial Information Free Framework For Traffic Forecasting

About

Spatio-temporal prediction plays an important role in many application areas especially in traffic domain. However, due to complicated spatio-temporal dependency and high non-linear dynamics in road networks, traffic prediction task is still challenging. Existing works either exhibit heavy training cost or fail to accurately capture the spatio-temporal patterns, also ignore the correlation between distant roads that share the similar patterns. In this paper, we propose a novel deep learning framework to overcome these issues: 3D Temporal Graph Convolutional Networks (3D-TGCN). Two novel components of our model are introduced. (1) Instead of constructing the road graph based on spatial information, we learn it by comparing the similarity between time series for each road, thus providing a spatial information free framework. (2) We propose an original 3D graph convolution model to model the spatio-temporal data more accurately. Empirical results show that 3D-TGCN could outperform state-of-the-art baselines.

Bing Yu, Mengzhang Li, Jiyong Zhang, Zhanxing Zhu• 2019

Related benchmarks

TaskDatasetResultRank
Traffic speed forecastingPeMSD7 Large (test)
MAE2.27
19
Traffic speed forecastingPeMSD7 Medium-scale (test)
MAE2.23
19
Traffic Speed PredictionPEMS-BAY 60 min
MAE2.07
12
Traffic Speed PredictionPEMS-BAY 15 min
MAE1.34
7
Traffic Speed PredictionPEMS-BAY 30 min
MAE1.69
6
Showing 5 of 5 rows

Other info

Follow for update