动态图结构嵌入提升交通预测精度,捕捉节点间随时间变化的关联。
DWAFM: Dynamic Weighted Graph Structure Embedding Integrated with Attention and Frequency-Domain MLPs for Traffic Forecasting
- 构建可随时间动态调整的图结构嵌入,反映节点间关联强度变化。
- 在五个真实数据集上优于多个主流模型,最高提升达3.2%。
- 适合交通预测、智能交通系统研究者参考。
准确的交通预测是智能交通系统的关键任务,核心挑战在于精确建模交通数据中的复杂时空依赖关系。近年来,网络架构改进未能带来显著性能提升,而嵌入技术展现出巨大潜力。然而,现有嵌入方法常忽略图结构信息或仅依赖静态图结构,难以有效捕捉随时间演变的节点间动态关联。为此,本文提出一种新颖的动态加权图结构(DWGS)嵌入方法,其图结构能真实反映节点间动态关联强度的变化。通过将DWGS嵌入与时空自适应嵌入、时间嵌入及特征嵌入结合,并集成注意力机制与频域多层感知机(MLP),设计出新型交通预测模型DWAFM。在五个真实世界交通数据集上的实验表明,该模型性能优于部分先进方法。
原文摘要 · Abstract (English)
Accurate traffic prediction is a key task for intelligent transportation systems. The core difficulty lies in accurately modeling the complex spatial-temporal dependencies in traffic data. In recent years, improvements in network architecture have failed to bring significant performance enhancements, while embedding technology has shown great potential. However, existing embedding methods often ignore graph structure information or rely solely on static graph structures, making it difficult to effectively capture the dynamic associations between nodes that evolve over time. To address this issue, this letter proposes a novel dynamic weighted graph structure (DWGS) embedding method, which relies on a graph structure that can truly reflect the changes in the strength of dynamic associations between nodes over time. By first combining the DWGS embedding with the spatial-temporal adaptive embedding, as well as the temporal embedding and feature embedding, and then integrating attention and frequency-domain multi-layer perceptrons (MLPs), we design a novel traffic prediction model, termed the DWGS embedding integrated with attention and frequency-domain MLPs (DWAFM). Experiments on five real-world traffic datasets show that the DWAFM achieves better prediction performance than some state-of-the-arts.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。