Transformer-based Spatial-Temporal Graph Attention Network for Traffic Flow Prediction

Fangzhou Yan*, Qi Chen

*Corresponding author for this work

Research output: Chapter in Book or Report/Conference proceedingConference Proceedingpeer-review

Abstract

Traffic flow prediction, which plays an important role in intelligent traffic systems, has become a pressing problem to be addressed with the continuous development of smart cities. Currently, the fundamental obstacle lies in effectively modelling the complex spatial-temporal dependencies present in traffic flow data. Deep learning models such as Graph Neural Network based models and Transformer based models have shown promising results in this field. However, methods founded on a single model or framework have one significant limitation: Such methods cannot adequately represent the spatial and temporal features of traffic flow data, restricting the model's ability to learn the dynamics of urban transportation. In this paper, we propose a transformer-based spatial-temporal graph attention network model called TSTGAT for traffic flow prediction, which integrates Transformer and Graph Attention Network. Experiments on two real-world traffic datasets from the Caltrans Performance Measurement System (PeMS) demonstrate that the proposed TSTGAT model outperforms well-known baselines.

Original languageEnglish
Title of host publicationProceedings - 2023 International Conference on Cyber-Enabled Distributed Computing and Knowledge Discovery, CyberC 2023
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages132-135
Number of pages4
ISBN (Electronic)9798350308693
DOIs
Publication statusPublished - 2023
Event15th International Conference on Cyber-Enabled Distributed Computing and Knowledge Discovery, CyberC 2023 - Jiangsu, China
Duration: 2 Nov 20234 Nov 2023

Publication series

NameProceedings - 2023 International Conference on Cyber-Enabled Distributed Computing and Knowledge Discovery, CyberC 2023

Conference

Conference15th International Conference on Cyber-Enabled Distributed Computing and Knowledge Discovery, CyberC 2023
Country/TerritoryChina
CityJiangsu
Period2/11/234/11/23

Keywords

  • Traffic flow prediction
  • deep learning
  • graph neural network
  • transformer

Fingerprint

Dive into the research topics of 'Transformer-based Spatial-Temporal Graph Attention Network for Traffic Flow Prediction'. Together they form a unique fingerprint.

Cite this