Traffic Flow Prediction Based on Adaptive Graph Attention Gated Recurrent Network

Xian-guang JIA , Meng-yi GUO , Ying-ying LÜ , Jie QU , Wei-jie FANG , Jin-ke ZHANG , Dong LI

Science Technology and Engineering ›› 2026, Vol. 26 ›› Issue (24) : 10592 -10605.

PDF
Science Technology and Engineering ›› 2026, Vol. 26 ›› Issue (24) : 10592-10605 DOI: 10.12404/j.issn.1671-1815.2507328
research-article
Traffic Flow Prediction Based on Adaptive Graph Attention Gated Recurrent Network
Author information +
History +
PDF

Abstract

To enhance the spatio-temporal modeling capability of traffic flow prediction, a prediction method based on an adaptive graph attention gated recurrent unit network (AGA-GRU) was proposed. The method constructed a learnable adjacency matrix and obtains a probabilistic structure through softplus activation and row normalization. A row-wise Top-K retention and renormalization strategy was introduced to achieve hard sparsification, while an L1 sparsity regularization was applied to the structural parameters to improve interpretability and robustness. In the spatial dimension, multi-head graph attention (GAT) was employed to suppress noisy edges and highlight key correlations. In the temporal dimension, gate recurrent unit (GRU) was used to capture sequential dependencies. The two components were cascaded to form an efficient spatio-temporal modeling framework. The training objective combined prediction error and sparsity regularization. Experiments conducted on real-world datasets, with autoregressive integrated moving average model (ARIMA), bidirectional long short-term memory network (Bi-LSTM), graph neural network-long short-term memory network (GNN-LSTM), spatio-temporal graph convolutional network (ST-GCN), and multi-task graph neural network (MTGNN) as baseline models, were evaluated using mean absolute error (MAE), mean squared error (MSE), and root mean squared error (RMSE) metrics. The results show that AGA-GRU outperforms the compared methods in multi-step forecasting tasks and learns sparse and interpretable graph structures (e.g., self-loops and a few key cross-link connections), achieving a good balance between prediction accuracy and model interpretability. This method is well-suited for traffic flow prediction in urban road networks.

Graphical abstract

Keywords

intelligent transportation / traffic flow prediction / adaptive graph attention gated recurrent unit network (AGA-GRU) model / spatio-temporal dependency modeling / adaptive adjacency

Cite this article

Download citation ▾
Xian-guang JIA, Meng-yi GUO, Ying-ying LÜ, Jie QU, Wei-jie FANG, Jin-ke ZHANG, Dong LI. Traffic Flow Prediction Based on Adaptive Graph Attention Gated Recurrent Network. Science Technology and Engineering, 2026, 26(24): 10592-10605 DOI:10.12404/j.issn.1671-1815.2507328

登录浏览全文

4963

注册一个新账户 忘记密码

References

Funding

PDF (10141KB)

24

Accesses

0

Citation

Detail

Sections
Recommended

/

〈 〉