GEnSHIN通过图结构增强提升交通流预测精度与稳定性。
GEnSHIN: Graphical Enhanced Spatio-temporal Hierarchical Inference Network for Traffic Flow Prediction
- 融合Transformer的图卷积循环单元,捕捉长期时间依赖
- 基于真实路网与数据驱动的异构图生成,更贴合实际交通特征
- 动态记忆库支持个性化节点表征,适合实时交通系统部署
随着城市化进程加快,智能交通系统对精准交通流预测的需求日益增长。本文提出一种新型图增强时空分层推理网络(GEnSHIN),以应对交通流中复杂的时空依赖关系。模型包含三项创新设计:1)引入Transformer模块的注意力增强型图卷积循环单元(GCRU),强化长期时间依赖建模能力;2)非对称双嵌入图生成机制,结合真实道路网络与数据驱动的潜在异构拓扑,生成更契合实际交通特性的图结构;3)动态记忆库模块,利用可学习的交通模式原型为每个传感器节点提供个性化表征,并在解码阶段引入轻量级图更新器,适应道路状态的动态变化。在公开数据集METR-LA上的大量实验表明,GEnSHIN在均值绝对误差(MAE)、均方根误差(RMSE)和平均绝对百分比误差(MAPE)等多个指标上达到或优于对比模型表现,尤其在早晚高峰时段展现出优异的预测稳定性。消融实验进一步验证了各核心模块的有效性及其对最终性能的贡献。
原文摘要 · Abstract (English)
With the acceleration of urbanization, intelligent transportation systems have an increasing demand for accurate traffic flow prediction. This paper proposes a novel Graph Enhanced Spatio-temporal Hierarchical Inference Network (GEnSHIN) to handle the complex spatio-temporal dependencies in traffic flow prediction. The model integrates three innovative designs: 1) An attention-enhanced Graph Convolutional Recurrent Unit (GCRU), which strengthens the modeling capability for long-term temporal dependencies by introducing Transformer modules; 2) An asymmetric dual-embedding graph generation mechanism, which leverages the real road network and data-driven latent asymmetric topology to generate graph structures that better fit the characteristics of actual traffic flow; 3) A dynamic memory bank module, which utilizes learnable traffic pattern prototypes to provide personalized traffic pattern representations for each sensor node, and introduces a lightweight graph updater during the decoding phase to adapt to dynamic changes in road network states. Extensive experiments on the public dataset METR-LA show that GEnSHIN achieves or surpasses the performance of comparative models across multiple metrics such as Mean Absolute Error (MAE), Root Mean Square Error (RMSE), and Mean Absolute Percentage Error (MAPE). Notably, the model demonstrates excellent prediction stability during peak morning and evening traffic hours. Ablation experiments further validate the effectiveness of each core module and its contribution to the final performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。