-
CiteScore
-
Impact Factor
Volume 2, Issue 4, ICCK Transactions on Emerging Topics in Artificial Intelligence
Volume 2, Issue 4, 2025
Submit Manuscript Edit a Special Issue
Article QR Code
Article QR Code
Scan the QR code for reading
Popular articles
ICCK Transactions on Emerging Topics in Artificial Intelligence, Volume 2, Issue 4, 2025: 203-219

Open Access | Research Article | 26 October 2025
AST-GNNFormer: Adaptive Spatio-Temporal Graph Neural Network with Layer-Aware Preservation for Traffic Flow Prediction
1 School of Computer Science and Artificial Intelligence, Beijing Technology and Business University, Beijing 100048, China
* Corresponding Author: Xuebo Jin, [email protected]
Received: 07 July 2025, Accepted: 23 October 2025, Published: 26 October 2025  
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.

Graphical Abstract
AST-GNNFormer: Adaptive Spatio-Temporal Graph Neural Network with Layer-Aware Preservation for Traffic Flow Prediction

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.

References
  1. Lv, Y., Duan, Y., Kang, W., Li, Z., & Wang, F. Y. (2014). Traffic flow prediction with big data: A deep learning approach. IEEE transactions on intelligent transportation systems, 16(2), 865-873.
    [CrossRef]   [Google Scholar]
  2. Tedjopurnomo, D. A., Bao, Z., Zheng, B., Choudhury, F. M., & Qin, A. K. (2020). A survey on modern deep neural network for traffic prediction: Trends, methods and challenges. IEEE Transactions on Knowledge and Data Engineering, 34(4), 1544-1561.
    [CrossRef]   [Google Scholar]
  3. Hochreiter, S., & Schmidhuber, J. (1997). Long short-term memory. Neural computation, 9(8), 1735-1780.
    [CrossRef]   [Google Scholar]
  4. Ma, X., Tao, Z., Wang, Y., Yu, H., & Wang, Y. (2015). Long short-term memory neural network for traffic speed prediction using remote microwave sensor data. Transportation Research Part C: Emerging Technologies, 54, 187-197.
    [CrossRef]   [Google Scholar]
  5. Ke, J., Zheng, H., Yang, H., & Chen, X. M. (2017). Short-term forecasting of passenger demand under on-demand ride services: A spatio-temporal deep learning approach. Transportation research part C: Emerging technologies, 85, 591-608.
    [CrossRef]   [Google Scholar]
  6. Li, Y., Yu, R., Shahabi, C., & Liu, Y. (2017). Diffusion convolutional recurrent neural network: Data-driven traffic forecasting. arXiv preprint arXiv:1707.01926.
    [Google Scholar]
  7. Yu, B., Yin, H., & Zhu, Z. (2017). Spatio-temporal graph convolutional networks: A deep learning framework for traffic forecasting. arXiv preprint arXiv:1709.04875.
    [Google Scholar]
  8. Yan, S., Xiong, Y., & Lin, D. (2018, April). Spatial temporal graph convolutional networks for skeleton-based action recognition. In Proceedings of the AAAI conference on artificial intelligence (Vol. 32, No. 1).
    [CrossRef]   [Google Scholar]
  9. Guo, S., Lin, Y., Feng, N., Song, C., & Wan, H. (2019, July). Attention based spatial-temporal graph convolutional networks for traffic flow forecasting. In Proceedings of the AAAI conference on artificial intelligence (Vol. 33, No. 01, pp. 922-929).
    [CrossRef]   [Google Scholar]
  10. Zheng, C., Fan, X., Wang, C., & Qi, J. (2020, April). Gman: A graph multi-attention network for traffic prediction. In Proceedings of the AAAI conference on artificial intelligence (Vol. 34, No. 01, pp. 1234-1241).
    [CrossRef]   [Google Scholar]
  11. Wu, Z., Pan, S., Long, G., Jiang, J., & Zhang, C. (2019). Graph wavenet for deep spatial-temporal graph modeling. arXiv preprint arXiv:1906.00121.
    [Google Scholar]
  12. Wu, Z., Pan, S., Long, G., Jiang, J., Chang, X., & Zhang, C. (2020, August). Connecting the dots: Multivariate time series forecasting with graph neural networks. In Proceedings of the 26th ACM SIGKDD international conference on knowledge discovery & data mining (pp. 753-763).
    [CrossRef]   [Google Scholar]
  13. Bai, L., Yao, L., Li, C., Wang, X., & Wang, C. (2020). Adaptive graph convolutional recurrent network for traffic forecasting. Advances in neural information processing systems, 33, 17804-17815.
    [Google Scholar]
  14. Zheng, Y., Yi, L., & Wei, Z. (2025). A survey of dynamic graph neural networks. Frontiers of Computer Science, 19(6), 196323.
    [CrossRef]   [Google Scholar]
  15. Dai, B. A., Ye, B. L., & Li, L. (2024). A novel hybrid time-varying graph neural network for traffic flow forecasting. arXiv preprint arXiv:2401.10155.
    [Google Scholar]
  16. Lu, S., Chen, H., & Teng, Y. (2024). Multi-scale non-local spatio-temporal information fusion networks for multi-step traffic flow forecasting. ISPRS International Journal of Geo-Information, 13(3), 71.
    [CrossRef]   [Google Scholar]
  17. Xiong, L., Yuan, X., Hu, Z., Huang, X., & Huang, P. (2024). Gated fusion adaptive graph neural network for urban road traffic flow prediction. Neural Processing Letters, 56(1), 9.
    [CrossRef]   [Google Scholar]
  18. Yaqub, M., Ahmad, S., Manan, M. A., Pathan, M. S., & He, L. (2025). Predicting traffic flow with federated learning and graph neural with asynchronous computations network. Array, 100411.
    [CrossRef]   [Google Scholar]
  19. Jiang, W., & Luo, J. (2022). Graph neural network for traffic forecasting: A survey. Expert systems with applications, 207, 117921.
    [CrossRef]   [Google Scholar]
  20. Bai, S., Kolter, J. Z., & Koltun, V. (2018). An empirical evaluation of generic convolutional and recurrent networks for sequence modeling. arXiv preprint arXiv:1803.01271.
    [Google Scholar]
  21. Smith, B. L., Williams, B. M., & Oswald, R. K. (2002). Comparison of parametric and nonparametric models for traffic flow forecasting. Transportation Research Part C: Emerging Technologies, 10(4), 303-321.
    [CrossRef]   [Google Scholar]
  22. Williams, B. M., & Hoel, L. A. (2003). Modeling and forecasting vehicular traffic flow as a seasonal ARIMA process: Theoretical basis and empirical results. Journal of transportation engineering, 129(6), 664-672.
    [CrossRef]   [Google Scholar]
  23. Kipf, T. N. (2016). Semi-supervised classification with graph convolutional networks. arXiv preprint arXiv:1609.02907.
    [Google Scholar]
  24. Li, M., & Zhu, Z. (2021, May). Spatial-temporal fusion graph neural networks for traffic flow forecasting. In Proceedings of the AAAI conference on artificial intelligence (Vol. 35, No. 5, pp. 4189-4196).
    [CrossRef]   [Google Scholar]

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

Article Metrics
Citations:

Crossref

0

Scopus

0

Web of Science

0
Article Access Statistics:
Views: 128
PDF Downloads: 21

Publisher's Note
ICCK stays neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Rights and Permissions
CC BY 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.
ICCK Transactions on Emerging Topics in Artificial Intelligence

ICCK Transactions on Emerging Topics in Artificial Intelligence

ISSN: 3068-6652 (Online)

Email: [email protected]

Portico

Portico

All published articles are preserved here permanently:
https://www.portico.org/publishers/icck/