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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Traffic speed forecasting | PeMSD7 Large (test) | MAE2.27 | 19 | |
| Traffic speed forecasting | PeMSD7 Medium-scale (test) | MAE2.23 | 19 | |
| Traffic Speed Prediction | PEMS-BAY 60 min | MAE2.07 | 12 | |
| Traffic Speed Prediction | PEMS-BAY 15 min | MAE1.34 | 7 | |
| Traffic Speed Prediction | PEMS-BAY 30 min | MAE1.69 | 6 |