Abstract
Accurate traffic flow prediction plays a critical role in intelligent transportation systems, providing essential support for urban planning, traffic control, and congestion mitigation. To address the challenges of spatial heterogeneity and temporal dynamics inherent in traffic data, this paper proposes AST-GNNFormer, an adaptive spatio-temporal graph neural network that integrates graph attention mechanisms with temporal convolution. The model introduces three key components to enhance predictive accuracy and generalization: (1) a Layer-aware Information Preservation mechanism that mitigates over-smoothing in deep GNNs by retaining original node features across layers; (2) an Inter-Layer Attention Module that dynamically selects and weights informative layer-wise features to improve multi-layer fusion quality; and (3) an Adaptive Graph Learning Module that fuses prior adjacency knowledge with learnable structures, enabling dynamic topology adaptation. Additionally, a Temporal Convolution Module is incorporated to model multi-scale temporal dependencies efficiently. Extensive experiments on real-world benchmark datasets (PEMS04 and PEMS08) demonstrate that AST-GNNFormer significantly outperforms existing state-of-the-art methods in both short-term and long-term traffic forecasting tasks. Ablation studies further confirm the effectiveness of each proposed component.
Keywords
graph neural network
traffic flow prediction
adaptive graph learning
inter-layer attention mechanism
Data Availability Statement
Data will be made available on request.
Funding
This work was supported in part by the National Natural Science Foundation of China under Grant 62173007, Grant 62203020, Grant 62473008, Grant 62433002, and Grant 62476014; in part by the Beijing Nova Program under Grant 20240484710; in part by the Project of Humanities and Social Sciences (Ministry of Education in China, MOC) under Grant 22YJCZH006; in part by the Beijing Scholars Program under Grant 099; in part by the Project of ALL China Federation of Supply and Marketing Cooperatives under Grant 202407; in part by the Project of Beijing Municipal University Teacher Team Construction Support Plan under Grant BPHR20220104.
Conflicts of Interest
The authors declare no conflicts of interest.
Ethical Approval and Consent to Participate
Not applicable.
Cite This Article
APA Style
Zheng, Y., & Jin, X. (2025). AST-GNNFormer: Adaptive Spatio-Temporal Graph Neural Network with Layer-Aware Preservation for Traffic Flow Prediction. ICCK Transactions on Emerging Topics in Artificial Intelligence, 2(4), 203–219. https://doi.org/10.62762/TETAI.2025.387543
Publisher's Note
ICCK stays neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and Permissions

Copyright © 2025 by the Author(s). Published by Institute of Central Computation and Knowledge. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license (
https://creativecommons.org/licenses/by/4.0/), which permits use, sharing, adaptation, distribution and reproduction in any medium or format, as long as you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons licence, and indicate if changes were made.