arXiv:2501.00890cs.ROcs.LG2025-01被引 2

用时空注意力模型提升车联网中车辆轨迹预测精度

Spatial Temporal Attention based Target Vehicle Trajectory Prediction for Internet of Vehicles

  • 构建图注意力与Transformer融合的时空特征提取框架
  • 在两地出租车数据集上比Transformer高10.55%匹配率
  • 适合智能交通与物流路径规划场景使用

在智能交通系统中,准确预测复杂交通环境下车辆行为至关重要。本文提出基于时空注意力的目标车辆轨迹预测方法STATVTPred,结合GPS定位技术,利用完整的时空轨迹数据动态预测车辆未来路径。将车辆轨迹映射为有向图,通过图注意力网络(GAT)提取空间特征,使用Transformer捕捉时间序列特征,并融合局部道路网络结构图进行轨迹优化,生成平滑的预测序列。在T-Drive和成都出租车轨迹数据集上的实验表明,STATVTPred在北京和成都数据集上分别比Transformer模型提升6.38%和10.55%的平均匹配率(AMR),相比LSTM编码器-解码器模型分别提升37.45%和36.06%。该方法显著提升了轨迹预测准确性,适用于物流与交通系统的轨迹预测任务。

原文摘要 · Abstract (English)

Forecasting vehicle behavior within complex traffic environments is pivotal within Intelligent Transportation Systems (ITS). Though this technology plays a significant role in alleviating the prevalent operational difficulties in logistics and transportation systems, the precise prediction of vehicle trajectories still poses a substantial challenge. To address this, our study introduces the Spatio Temporal Attention-based methodology for Target Vehicle Trajectory Prediction (STATVTPred). This approach integrates Global Positioning System(GPS) localization technology to track target movement and dynamically predict the vehicle's future path using comprehensive spatio-temporal trajectory data. We map the vehicle trajectory onto a directed graph, after which spatial attributes are extracted via a Graph Attention Networks(GATs). The Transformer technology is employed to yield temporal features from the sequence. These elements are then amalgamated with local road network structure maps to filter and deliver a smooth trajectory sequence, resulting in precise vehicle trajectory prediction.This study validates our proposed STATVTPred method on T-Drive and Chengdu taxi-trajectory datasets. The experimental results demonstrate that STATVTPred achieves 6.38% and 10.55% higher Average Match Rate (AMR) than the Transformer model on the Beijing and Chengdu datasets, respectively. Compared to the LSTM Encoder-Decoder model, STATVTPred boosts AMR by 37.45% and 36.06% on the same datasets. This is expected to establish STATVTPred as a new approach for handling trajectory prediction of targets in logistics and transportation scenarios, thereby enhancing prediction accuracy.

轨迹预测时空注意力车联网图神经网络

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