解决交通流预测中节点间信息传播延迟问题,提升预测精度与效率。
Eliminating Propagation Delay: Attention-Based Spatial-Temporal Fusion Graph Convolution Network for Traffic Flow Prediction

- 引入基于掩码矩阵的多头自注意力机制,消除节点间传播延迟误差。
- 在五个真实数据集上优于8个基线模型,同时计算效率更高。
- 适合需要高时效性交通预测的应用场景,如智慧交通系统。
交通流预测对优化交通系统和提升城市出行效率至关重要。现有基于图卷积的模型虽能提取时空特征,但多聚焦于拓扑关系中的时空与语义关联,存在两大问题:一是卷积结构仅关注静态空间依赖与拓扑结构中的时空关系,忽视相邻节点间的信息传播延迟差异;二是模型常堆叠大量复杂结构,导致训练阶段计算时间显著增加,难以满足实时性需求。本文提出一种新型网络——注意力驱动的时空融合图卷积网络(A-STFGCN)。设计时空融合模块,在基于掩码矩阵的多头自注意力机制下,消除传播延迟误差,同时捕捉数据的长期与短期时间特征。在五个真实数据集上的大量实验表明,该方法在整体性能上优于8个基线模型,且具备优异的计算与数据利用效率。
原文摘要 · Abstract (English)
Predicting traffic flow is crucial to optimizing transportation systems and improving urban mobility. Many graph convolution-based models have been proposed to extract spatial-temporal features and predict traffic flow. However, most focus on spatial-temporal and semantic correlation in topological relationships. There are two primary problems to address. Firstly, the convolutional structure in the model focuses on utilizing static spatial dependencies and spatial-temporal relationships in topological structures, while neglecting the different information propagation delays between adjacent nodes in the convolution. Secondly, these methods often stack a large number of complex structures, resulting in a substantial increase in computational time during the model training phase, thereby disregarding the model's requirements for timeliness. In this paper, we propose a novel network called the Attention-Based Spatial-Temporal Fusion Graph Convolution Network (A-STFGCN). We design a spatial-temporal fusion block to extract the spatial-temporal feature correlations with propagation delay errors removed and to capture both long-term and short-term temporal characteristics of the data within a multi-head self-attention mechanism based on a mask matrix. Extensive experiments on five real-world datasets demonstrate that our method achieves the best overall performance while having good computation and data utilization efficiency compared with the eight baseline methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。