EAAI Journal 2025 Journal Article
ASTTN: An Adaptive Spatial–Temporal Transformer Network for traffic flow prediction
- Zijie Xue
- Linyu Huang
- Qian Ning
Traffic flow prediction as a representative challenge in spatial–temporal modeling tasks, playing a fundamental role in intelligent transportation systems (ITSs). The complexity of spatial–temporal features from road networks and unexpected events complicates accurate predictions. To address the above problems, we propose a novel Adaptive Spatial–Temporal Transformer Network (ASTTN) designed to effectively capture dynamic spatial–temporal features and improve the model’s predictive adaptability under different systems. Considering the comprehensive temporal features, we design a Multi-View Temporal Attention Module to extract long-term and short-term temporal correlation. Additionally, by the limitation of the fixed adjacency matrix, our model proposes an Adaptive Spatial Graph Convolutional Network Module to dynamically capture spatial dependence by the adaptive mechanism based on self-attention. An adaptive Gated Fusion module is employed to achieve the dynamic fusion of temporal and spatial features. Experiments on four real-world traffic datasets show that our model significantly outperforms existing state-of-the-art baselines.