Spatial-Temporal Tensor Graph Convolutional Network for Traffic Prediction

10 Mar 2021  ·  Xuran Xu, Tong Zhang, Chunyan Xu, Zhen Cui, Jian Yang ·

Accurate traffic prediction is crucial to the guidance and management of urban traffics. However, most of the existing traffic prediction models do not consider the computational burden and memory space when they capture spatial-temporal dependence among traffic data. In this work, we propose a factorized Spatial-Temporal Tensor Graph Convolutional Network to deal with traffic speed prediction. Traffic networks are modeled and unified into a graph that integrates spatial and temporal information simultaneously. We further extend graph convolution into tensor space and propose a tensor graph convolution network to extract more discriminating features from spatial-temporal graph data. To reduce the computational burden, we take Tucker tensor decomposition and derive factorized a tensor convolution, which performs separate filtering in small-scale space, time, and feature modes. Besides, we can benefit from noise suppression of traffic data when discarding those trivial components in the process of tensor decomposition. Extensive experiments on two real-world traffic speed datasets demonstrate our method is more effective than those traditional traffic prediction methods, and meantime achieves state-of-the-art performance.

PDF Abstract

Datasets


Results from the Paper


Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Traffic Prediction SZ-Taxi factorized ST-TGCN MAE @ 15min 2.0198 # 1
MAE @ 30min 2.2951 # 1
MAE @ 45min 2.3689 # 1
MAE @ 60min 2.4476 # 1

Methods