A Robust Spatio-Temporal Graph Attention Network for Traffic Flow Forecasting Under Sudden Disturbances
Main Article Content
Abstract
The prediction of traffic flow has evolved from an empirical extrapolation method to a spatiotemporal correlation-based method due to the development of artificial intelligence and intelligent algorithms. In order to remedy the problem that the disturbance in the future causes the prediction error to increase sharply, we put forward the robust spatiotemporal graph attention network. Based on graph attention and gated temporal modeling, we provide a framework which brings together a road network topological encoding, a disturbance factor mapping, a temporal dependency extraction and a strong loss constraint mechanism. Experimental results show that under normal conditions, the model achieves an MAE of 13.42%, an RMSE of 20.74%, and a MAPE of 8.31%, outperforming STGCN’s 14.19%, 21.86%, and 8.64%, respectively; under disturbance conditions, the MAE is 17.96%, with a robustness index of 0.874, demonstrating good prediction stability.
Downloads
Article Details
This work is licensed under a Creative Commons Attribution 4.0 International License.
Authors who publish with this journal agree to the following terms:
- Authors retain copyright and grant the journal right of first publication with the work simultaneously licensed under a Creative Commons Attribution License that allows others to share the work with an acknowledgement of the work's authorship and initial publication in this journal.
- Authors are able to enter into separate, additional contractual arrangements for the non-exclusive distribution of the journal's published version of the work (e.g., post it to an institutional repository or publish it in a book), with an acknowledgement of its initial publication in this journal.
- Authors are permitted and encouraged to post their work online (e.g., in institutional repositories or on their website) prior to and during the submission process, as it can lead to productive exchanges, as well as earlier and greater citation of published work (See The Effect of Open Access).
References
C. Zheng, X. Fan, C. Wang, et al., “Gman: A graph multi-attention network for traffic prediction,” in Proc. AAAI Conf. Artif. Intell., vol. 34, no. 1, pp. 1234–1241, 2020.
X. Wang, Y. Ma, Y. Wang, W. Jin, X. Wang, J. Tang, C. Jia, and J. Yu, “Traffic flow prediction via spatial temporal graph neural network,” in Proc. Web Conf., pp. 1082–1092, 2020.
C. Song, Y. Lin, S. Guo, and H. Wan, “Spatial-temporal synchronous graph convolutional networks: A new framework for spatial-temporal network data forecasting,” in Proc. AAAI Conf. Artif. Intell., vol. 34, no. 1, pp. 914–921, 2020.
L. Bai, L. Yao, C. Li, X. Wang, and C. Wang, “Adaptive graph convolutional recurrent network for traffic forecasting,” in Advances in Neural Information Processing Systems, pp. 17804–17815, 2020.
Z. Wu, S. Pan, G. Long, J. Jiang, X. Chang, and C. Zhang, “Connecting the dots: Multivariate time series forecasting with graph neural networks,” in Proc. 26th ACM SIGKDD Int. Conf. Knowledge Discovery & Data Mining, pp. 753–763, 2020.
M. Li and Z. Zhu, “Spatial-temporal fusion graph neural networks for traffic flow forecasting,” in Proc. AAAI Conf. Artif. Intell., vol. 35, no. 5, pp. 4189–4196, 2021.
X. Ta, Z. Liu, X. Hu, L. Yu, L. Sun, and B. Du, “Adaptive spatio-temporal graph neural network for traffic forecasting,” Knowledge-Based Systems, vol. 242, Art. no. 108199, 2022.
L. Chen, W. Shao, M. Lv, W. Chen, Y. Zhang, and C. Yang, “AARGNN: An attentive attributed recurrent graph neural network for traffic flow prediction considering multiple dynamic factors,” IEEE Trans. Intell. Transp. Syst., vol. 23, no. 10, pp. 17201–17211, 2022.
J. Huang, K. Luo, L. Cao, Y. Wen, and S. Zhong, “Learning multiaspect traffic couplings by multirelational graph attention networks for traffic prediction,” IEEE Trans. Intell. Transp. Syst., vol. 23, no. 11, pp. 20681– 20695, 2022.
Z. Qiu, T. Zhu, Y. Jin, L. Sun, and B. Du, “A graph attention fusion network for event-driven traffic speed prediction,” Information Sciences, vol. 622, pp. 405–423, 2023.
W. Weng, J. Fan, H. Wu, Y. Hu, H. Tian, F. Zhu, and J. Wu, “A decomposition dynamic graph convolutional recurrent network for traffic forecasting,” Pattern Recognition, vol. 142, Art. no. 109670, 2023.
C. Wang, R. Tian, J. Hu, et al., “A trend graph attention network for traffic prediction,” Information Sciences, vol. 623, pp. 275–292, 2023.
G. Jin, L. Liu, F. Li, and J. Huang, “Spatio-temporal graph neural point process for traffic congestion event prediction,” in Proc. AAAI Conf. Artif. Intell., vol. 37, no. 12, pp. 14268–14276, 2023.
J. Feng, Y. Zhang, X. Piao, Y. Hu, and B. Yin, “Traffic anomaly detection based on spatio-temporal hypergraph convolution neural networks,” Physica A: Statistical Mechanics and its Applications, vol. 646, Art. no. 129891, 2024.
S. Yang, Q. Wu, and M. Li, “Decoupled multi-spatio-temporal fusion graph convolutional recurrent network for traffic prediction,” Engineering Applications of Artificial Intelligence, vol. 163, Art. no. 112956, 2026.
X. Yang, S. Tan, and P. Krähenbühl, “Long-term traffic simulation with interleaved autoregressive motion and scenario generation,” in Proc. IEEE/CVF Int. Conf. Computer Vision, pp. 25305–25314, 2025.
L. Feng, Q. Li, Z. Peng, et al., “Trafficgen: Learning to generate diverse and realistic traffic scenarios,” arXiv preprint arXiv:2210.06609, 2022.
Q. Li, Z. M. Peng, L. Feng, et al., “Scenarionet: Open-source platform for large-scale traffic scenario simulation and modeling,” Advances in Neural Information Processing Systems, vol. 36, pp. 3894–3920, 2023.
P. Cao, Y. Wang, and X. Liu, “Trajectory-based method for dividing lanes of vehicle trajectories collected by roadside LiDAR,” Transportation Research Record, vol. 2678, no. 3, pp. 816–828, 2024.
G. Reyes, R. Tolozano-Benites, L. Lanzarini, et al., “Method for the identification and classification of zones with vehicular congestion,” ISPRS Int. J. Geo-Inf., vol. 13, no. 3, Art. no. 73, 2024.
R. Jain, K. Okumura, M. Amir, et al., “Graph attention-guided search for dense multi-agent pathfinding,” in Proc. AAAI Conf. Artif. Intell., vol. 40, no. 35, pp. 29504–29512, 2026.
W. Lan, J. Liu, B. Yuan, et al., “AutoSceCraft: Generate various driving scenarios from scratch for autonomous driving systems,” Tsinghua Science and Technology, vol. 31, no. 2, pp. 1282–1305, 2026.
T. Phan, T. Phan, and S. Koenig, “Generative curricula for multi-agent path finding via unsupervised and reinforcement learning,” Journal of Artificial Intelligence Research, vol. 82, pp. 2471–2534, 2025.
G. Hess, J. Jaxing, E. Svensson, et al., “Masked autoencoder for self-supervised pre-training on lidar point clouds,” in Proc. IEEE/CVF Winter Conf. Applications of Computer Vision, pp. 350–359, 2023.
B. Chen, Z. Cao, and Q. Bai, “Satf: A scalable attentive transfer framework for efficient multiagent reinforcement learning,” IEEE Trans. Neural Netw. Learn. Syst., vol. 36, no. 4, pp. 6627–6641, 2024.
R. A. Alshamsi, I. M. Al Jawarneh, L. Foschini, et al., “ApproxGe-oMap: An efficient system for generating approximate geo-maps from big geospatial data with quality of service guarantees,” Computers, vol. 14, no. 2, Art. no. 35, 2025.