1
Return

Auto-correlation based spatio-temporal adaptive transformer traffic flow prediction

delete2026-04-01
delete1
PRE
AI
W
Wang, Hongyan *
Z
Zhang, Hong
L
Linlong Chen
C
Chen, Linbiao
DOI:10.1177/09544070251341953delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Traffic flow prediction is a crucial technology in intelligent transportation systems. To effectively handle intricate spatio-temporal relationships and dynamic features of traffic flow, an Auto-Correlation Based Spatio-Temporal Adaptive Transformer Prediction Model (Auto-STAT) is established, which considers the periodicity of traffic flow. Auto-STAT encompasses such components as Auto-Correlation, Encoder-Decoder, Dynamic Halting, and Cross-Attention. Auto-Correlation is employed to capture the periodic characteristics of traffic flow. The encoder-decoder architecture incorporates Spatial-Adaptive Transformer (SA-Trans) and Temporal-Adaptive Transformer (TA-Trans) to extract intricate spatio-temporal dynamics. Dynamic Halting is integrated into the encoder to enhance computational efficiency. Cross-Attention module is constructed to mitigate error propagation between the encoder-decoder. Furthermore, two decoders are utilized to simultaneously tackle the Historical Traffic Reconstruction (HTR) task and the Future Traffic Forecasting (FTF) task to recollect historical traffic patterns and predict future traffic patterns. Experimental results demonstrate the proposed Auto-STAT achieves exceptional prediction performance on two datasets.
Keywords:
Traffic flow forecasting
transformer
spatio-temporal dynamics
periodicity

Journal

P
PROCEEDINGS OF THE INSTITUTION OF MECHANICAL ENGINEERS PART D-JOURNAL OF AUTOMOBILE ENGINEERING
IF:
1.5
Papers:
442
Citations:
0

Organization

L
lanzhou university of technology
Scholars:
1.1W
Papers: 6.7K
Citations: 4
Cited Papers

Cited Papers

Citing Papers

Citing Papers