arXiv:2412.02839cs.LG2024-12被引 2

通过位置对齐注意力机制,提升交通事故预测精度。

Geographical Information Alignment Boosts Traffic Analysis via Transpose Cross-attention

  • 引入转置交叉注意力,融合节点特征与地理坐标信息。
  • 在多个城市数据集上,F1提升1.3%至10.9%,AUC提升0.3%至4.8%。
  • 适合作为通用模块嵌入现有图神经网络,计算高效。

交通事故预测对提升道路安全和缓解拥堵至关重要,近年来图神经网络(GNN)在建模基于图的交通数据方面展现出潜力。然而,现有基于GNN的方法往往忽略或未显式利用地理位置信息,而该信息在理解空间依赖性中起关键作用。我们观察到事故地点具有高度相关性。为此,提出一种可即插即用的模块——地理信息对齐(GIA),通过新型转置交叉注意力机制,高效融合节点特征与地理坐标信息。由于交通数据节点数量庞大,传统逐节点对齐的交叉注意力在资源受限环境下可能不可行。因此,采用对查询、键、值进行转置的操作,显著降低计算开销,同时保留充分信息。在大规模城市级数据集上的实验结果表明,该方法在交通事件预测和严重程度预测(基于事故数量区间划分的严重等级)任务中均有效。例如,F1分数提升1.3%至10.9%,AUC提升0.3%至4.8%。

原文摘要 · Abstract (English)

Traffic accident prediction is crucial for enhancing road safety and mitigating congestion, and recent Graph Neural Networks (GNNs) have shown promise in modeling the inherent graph-based traffic data. However, existing GNN- based approaches often overlook or do not explicitly exploit geographic position information, which often plays a critical role in understanding spatial dependencies. This is also aligned with our observation, where accident locations are often highly relevant. To address this issue, we propose a plug-in-and-play module for common GNN frameworks, termed Geographic Information Alignment (GIA). This module can efficiently fuse the node feature and geographic position information through a novel Transpose Cross-attention mechanism. Due to the large number of nodes for traffic data, the conventional cross-attention mechanism performing the node-wise alignment may be infeasible in computation-limited resources. Instead, we take the transpose operation for Query, Key, and Value in the Cross-attention mechanism, which substantially reduces the computation cost while maintaining sufficient information. Experimental results for both traffic occurrence prediction and severity prediction (severity levels based on the interval of recorded crash counts) on large-scale city-wise datasets confirm the effectiveness of our proposed method. For example, our method can obtain gains ranging from 1.3% to 10.9% in F1 score and 0.3% to 4.8% in AUC.

交通预测图神经网络注意力机制地理信息

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