基于自适应图注意力门控网络的交通流量预测
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

U491.1+4

基金项目:

国家自然科学基金(52462050)。


Traffic Flow Prediction Based on Adaptive Graph Attention Gated Recurrent Network
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    为提高交通流预测的时空建模能力,提出一种基于自适应图注意力门控网络(Adaptive Graph Attention GRU,AGA-GRU)的预测方法。该方法以可学习参数构建邻接矩阵,通过 softplus与行归一化获得概率型结构,并引入逐行Top-K保留与重归一化实现硬稀疏化,同时对原始结构参数施加L1稀疏正则以提升结构可解释性与鲁棒性;在空间维度采用多头图注意力(GAT)抑制噪声边、突出关键关联,在时间维度采用GRU刻画时序依赖,二者级联形成高效的时空建模框架。训练目标由预测误差与稀疏正则联合构成。实验在所用数据集上,以 ARIMA、Bi-LSTM、GNN-LSTM、ST-GCN、MTGNN 为对比基线,使用 MAE、MSE、RMSE 等指标进行评估。结果表明,AGA-GRU 在多步预测场景下整体优于对比方法,并能学习到稀疏且可解释的图结构(如自环与少量关键跨路段连接明显),兼顾预测精度与模型可解释性,适用于城市路网的交通流量预测。

    Abstract:

    To enhance the spatio-temporal modeling capability of traffic flow prediction, this paper proposes a prediction method based on an Adaptive Graph Attention Gated Recurrent Unit network (AGA-GRU). The method constructs a learnable adjacency matrix and obtains a probabilistic structure through softplus activation and row normalization. A row-wise Top-K retention and renormalization strategy is introduced to achieve hard sparsification, while an L1 sparsity regularization is applied to the structural parameters to improve interpretability and robustness. In the spatial dimension, multi-head Graph Attention (GAT) is employed to suppress noisy edges and highlight key correlations; in the temporal dimension, GRU is used to capture sequential dependencies. The two components are cascaded to form an efficient spatio-temporal modeling framework. The training objective combines prediction error and sparsity regularization. Experiments conducted on real-world datasets, with ARIMA, Bi-LSTM, GNN-LSTM, ST-GCN, and MTGNN as baseline models, are evaluated using MAE, MSE, and 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.

    参考文献
    相似文献
    引证文献
引用本文

贾现广,郭梦毅,吕英英,等. 基于自适应图注意力门控网络的交通流量预测[J]. 科学技术与工程, 2026, 26(24): 10592-10605.
Jia Xianguang, Guo Mengyi, Lü Yingying, et al. Traffic Flow Prediction Based on Adaptive Graph Attention Gated Recurrent Network[J]. Science Technology and Engineering,2026,26(24):10592-10605.

复制
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2025-10-20
  • 最后修改日期:2026-06-16
  • 录用日期:2026-01-19
  • 在线发布日期: 2026-09-02
  • 出版日期:
×
2026年会通知 | “技术经济学驱动智能经济生态构建与治理变革”——中国技术经济学会第三十三届学术年会(2026)会议通知暨征文启事(第一轮)
亟待确认版面费归属稿件,敬请作者关注