首页|期刊导航|交通运输工程与信息学报|融合自适应图与时空Transformer的交通流预测模型

融合自适应图与时空Transformer的交通流预测模型OA

Traffic flow prediction model integrating adaptive graph and spatio-temporal transformer

中文摘要英文摘要

[背景]随着城市现代化进程的推进,智能交通系统已成为其必不可少的一部分,而通过使用准确的交通流预测来降低城市道路的拥堵则是智能交通系统发挥效能的关键所在.[目标]综合考虑交通数据的时空特性,动态捕获交通数据复杂的空间相关性以及时间相关性,有效提高交通预测任务的准确性.[方法]通过分析交通流数据时间和空间信息的相关性,实现时空特征的融合和交互,本文提出一种融合自适应图(DGC)与Transformer的预测模型,旨在动态捕获交通数据的时空相关性.模型首先利用多层感知机(MLP)投影和时间嵌入来捕捉周期性时间模式.在Sandwich块中,一个Transformer编码器负责捕捉长距离时间依赖性;随后,DGC模块捕捉数据驱动的隐藏空间依赖性;接着,图卷积网络(GCN)模块利用自适应邻接矩阵聚合空间信息;最后,第二个Transformer模块对融合了空间上下文的特征进行再次时间建模.整个架构堆叠两个Sandwich块,通过残差连接增强模型表达能力并确保训练稳定性,最后通过MLP投影层输出预测结果.[数据]加州交通局性能测量系统(PeMS)收集的四个广泛使用的交通预测数据集.[结果]在四个公开的数据集上,DGC-Transformer模型的平均绝对误差(MAE)、均方根误差(RMSE)和平均绝对百分比误差(MAPE)几乎全面优于所对比的五个基线以及十个模型,表明动态捕获交通数据时空相关性的重要性,使交通流预测效果得到显著提升.

[Background]With the advancement of urban modernization,intelligent transportation systems have become an indispensable part of modern cities.The key to the effectiveness of intelli-gent transportation systems lies in reducing urban road congestion through accurate traffic flow pre-diction.[Objective]To comprehensively consider the spatiotemporal characteristics of traffic data,dynamically capture the complex spatial and temporal correlations of traffic data,and effectively im-prove the accuracy of traffic prediction tasks.[Method]We propose a novel DGC-Transformer mod-el that integrates dynamic graph construction(DGC)with Transformer to dynamically capture spatio-temporal correlations in traffic data.The model employs multilayer perceptron(MLP)projection and temporal embeddings to capture periodic temporal patterns.Within the core Sandwich block,a Trans-former encoder first captures long-range temporal dependencies.Subsequently,a DGC module con-structs a data-driven adaptive adjacency matrix to model hidden spatial dependencies.A graph con-volutional network(GCN)module then aggregates spatial information using this adaptive matrix,followed by a second Transformer module that re-models the temporally contextualized features with spatial information.The architecture stacks two Sandwich blocks with residual connections to enhance expressive power and ensure training stability,outputting predictions through an MLP pro-jection layer.[Data]The model's performance was evaluated on four widely used traffic prediction datasets from the California department of transportation's performance measurement system(PeMS).[Result]Across the four public datasets,the DGC-Transformer model consistently outper-formed five baseline and ten state-of-the-art models in terms of mean absolute error(MAE),root mean square error(RMSE),and mean absolute percentage error(MAPE),underscoring the impor-tance of dynamically capturing spatio-temporal correlations for significantly improved traffic flow prediction accuracy.

殷炽磊;林之喆;周腾;谢海;曹春杰

海南大学,网络空间安全学院,海口 570228海南大学,网络空间安全学院,海口 570228海南大学,网络空间安全学院,海口 570228海南大学,网络空间安全学院,海口 570228海南大学,网络空间安全学院,海口 570228

交通工程

智能交通交通流预测注意力机制图卷积神经网络

intelligent transportationtraffic flow predictionattention mechanismgraph convolu-tional neural networks

《交通运输工程与信息学报》 2026 (1)

90-101,12

国家自然科学基金项目(62462021)

10.19961/j.cnki.1672-4747.2025.09.002

评论