Spatial-Temporal Tensor Graph Convolutional Network for Traffic Speed Prediction

Xuran Xu, Tong Zhang, Chunyan Xu, Zhen Cui*, Jian Yang

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

9 Citations (Scopus)

Abstract

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.

Original languageEnglish
Pages (from-to)92-103
Number of pages12
JournalIEEE Transactions on Intelligent Transportation Systems
Volume24
Issue number1
DOIs
Publication statusPublished - 1 Jan 2023
Externally publishedYes

Keywords

  • Traffic speed prediction
  • higher-order principal components analysis
  • spatial-temporal graph convolutional network
  • tensor decomposition

Fingerprint

Dive into the research topics of 'Spatial-Temporal Tensor Graph Convolutional Network for Traffic Speed Prediction'. Together they form a unique fingerprint.

Cite this