arXiv:2410.09356cs.LG2024-10被引 1

通过融合矩阵提示增强时空交互,提升交通流量预测精度

Fusion Matrix Prompt Enhanced Self-Attention Spatial-Temporal Interactive Traffic Forecasting Framework

  • 构建融合矩阵提示的自注意力机制,动态建模节点间时空关联
  • 在六大数据集上优于现有模型,显著提升预测准确率
  • 适合需要高精度交通流预测的智慧城市与导航系统

近年来,随着交通管理和出行规划需求的增长,时空预测技术迅速发展。然而,现有交通预测模型仍存在局限:一方面,多数研究过度关注真实地理信息,忽略不同区域间的潜在交通关联;另一方面,时间建模中忽视不同时段的重要性。为此,我们提出融合矩阵提示增强的自注意力时空交互交通预测框架(FMPESTF),包含空间与时间模块对交通数据进行下采样。网络设计通过考虑时空异质性的交通融合矩阵作为查询,重建数据驱动的动态交通数据结构,精准揭示交通网络中节点间的流量关系。此外,引入注意力机制优化时间建模,并设计分层时空交互学习,使模型能适应多种交通场景。在六个真实世界交通数据集上的大量实验表明,该方法显著优于其他基线模型,展现出在交通预测任务中的高效性与准确性。

原文摘要 · Abstract (English)

Recently, spatial-temporal forecasting technology has been rapidly developed due to the increasing demand for traffic management and travel planning. However, existing traffic forecasting models still face the following limitations. On one hand, most previous studies either focus too much on real-world geographic information, neglecting the potential traffic correlation between different regions, or overlook geographical position and only model the traffic flow relationship. On the other hand, the importance of different time slices is ignored in time modeling. Therefore, we propose a Fusion Matrix Prompt Enhanced Self-Attention Spatial-Temporal Interactive Traffic Forecasting Framework (FMPESTF), which is composed of spatial and temporal modules for down-sampling traffic data. The network is designed to establish a traffic fusion matrix considering spatial-temporal heterogeneity as a query to reconstruct a data-driven dynamic traffic data structure, which accurately reveal the flow relationship of nodes in the traffic network. In addition, we introduce attention mechanism in time modeling, and design hierarchical spatial-temporal interactive learning to help the model adapt to various traffic scenarios. Through extensive experimental on six real-world traffic datasets, our method is significantly superior to other baseline models, demonstrating its efficiency and accuracy in dealing with traffic forecasting problems.

交通预测时空建模自注意力融合矩阵

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。