Spatial-Temporal Tensor Graph Convolutional Network for Traffic Speed Prediction

doi:10.1109/TITS.2022.3215613

Spatial-Temporal Tensor Graph Convolutional Network for Traffic Speed Prediction

2023-01-01

10

PRE

AI

DOI：10.1109/TITS.2022.3215613

Original

Original request for help

Save

Abstract

En 中文

Accurate traffic speed prediction is crucial for the guidance and management of urban traffic, which at the same time requires a model with a satisfactory computational burden and memory space in applications. In this paper, we propose a factorized Spatial-Temporal Tensor Graph Convolutional Network for traffic speed prediction. Traffic networks are modeled and unified into a graph tensor that integrates spatial and temporal information simultaneously. We extend graph convolution into tensor space and propose a tensor graph convolution network to extract more discriminating features from spatial-temporal graph data. We further introduce Tucker decomposition and derive a factorized tensor convolution to reduce the computational burden, 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 the three real-world datasets demonstrate that our method is more effective than traditional prediction methods, and achieves state-of-the-art performance.

Keywords:

Traffic speed prediction

tensor decomposition

spatial-temporal graph convolutional network

higher-order principal components analysis

Spatial-Temporal Tensor Graph Convolutional Network for Traffic Speed Prediction

Abstract

Journal

Organization