Multi-View Spatial-Temporal Graph Convolutional Network for Traffic Prediction

Shuqing Wei, Siyuan Feng, Hai Yang

Research output: Journal article publicationJournal articleAcademic researchpeer-review

9 Citations (Scopus)

Abstract

Multi-step traffic speed prediction is a challenging issue due to the multiple spatial-temporal dependencies among roads. Some spatial dependencies, especially those formed by different traffic modes, are not fully exploited, and how to simultaneously consider spatial and temporal dependencies and effectively integrate them within a single prediction framework needs further exploration. To tackle the above issues, we propose a multi-view spatial-temporal graph convolutional framework MVSTG, which adequately exploits the multi-view spatial-temporal dependencies and their interactions to improve the accuracy of traffic prediction. Multi-view temporal learning captures the multiple temporal trends by temporal convolution from multi-granularity historical data, and multi-view spatial learning handles the multiple spatial correlations by graph convolution from multiple graphs. In addition, view-wise attention-based fusion is proposed to adaptively identify the importance of each upstream view, fuse the multi-view information, and generate integrated results for downstream views. The experiments on two real-world urban traffic datasets demonstrate that the multi-view data and the proposed model framework enhance performance on the accuracy of speed prediction, especially in mid-term and long-term prediction.

Original languageEnglish
Pages (from-to)1-15
Number of pages15
JournalIEEE Transactions on Intelligent Transportation Systems
DOIs
Publication statusAccepted/In press - 2024

Keywords

  • Attention mechanism
  • Computational modeling
  • Convolutional neural networks
  • Correlation
  • Data models
  • multi-view learning
  • Predictive models
  • Public transportation
  • Roads
  • spatial–temporal graph convolutional network
  • traffic prediction

ASJC Scopus subject areas

  • Automotive Engineering
  • Mechanical Engineering
  • Computer Science Applications

Fingerprint

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

Cite this