arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2404.08021cs.LGcs.AIcs.RO

VeTraSS:基于图建模与表示学习的车辆轨迹相似性搜索

VeTraSS: Vehicle Trajectory Similarity Search Through Graph Modeling and Representation Learning

Dartmouth College(达特茅斯学院);Shanghai Jiao Tong University(上海交通大学);University of California, Irvine(加州大学欧文分校);Harvard University(哈佛大学)

Ming Cheng, Bowen Zhang, Ziyu Wang, Ziyi Zhou, Weiqi Feng, Yi Lyu, Xingjian Diao

AI总结:

针对现有车辆轨迹相似性搜索方法架构复杂、未利用轨迹动态特征的问题,提出基于多尺度图建模与多层注意力GNN的端到端框架VeTraSS,在两个数据集上达到SOTA,可支撑自动驾驶轨迹分析与安全导航。

摘要:

轨迹相似性搜索在自动驾驶中发挥着核心作用,它能让车辆分析不同轨迹的信息与特征,从而在动态环境中做出合理决策并安全导航。现有轨迹相似性搜索任务的研究主要采用序列处理算法或循环神经网络(RNNs),这类方法不可避免地存在架构复杂、训练成本高昂的问题。考虑到轨迹间存在复杂关联,使用图神经网络(GNNs)进行数据建模具备可行性,但多数方法直接采用现成的数学图结构作为输入,而非针对特定车辆轨迹数据构建专属图结构,忽略了这类数据的独特性与动态特征。为填补这一研究空白,我们提出VeTraSS——一种面向车辆轨迹相似性搜索的端到端流水线。具体而言,VeTraSS将原始轨迹数据建模为多尺度图,并通过一种新型的基于多层注意力的GNN生成全面的嵌入表示,所得嵌入可用于搜索相似车辆轨迹。在Porto和Geolife数据集上开展的大量实验验证了VeTraSS的有效性,我们的模型性能优于现有方法,达到了当前最优水平,展现了VeTraSS在现实世界自动驾驶车辆的轨迹分析与安全导航方面的应用潜力。

英文摘要:

Trajectory similarity search plays an essential role in autonomous driving, as it enables vehicles to analyze the information and characteristics of different trajectories to make informed decisions and navigate safely in dynamic environments. Existing work on the trajectory similarity search task primarily utilizes sequence-processing algorithms or Recurrent Neural Networks (RNNs), which suffer from the inevitable issues of complicated architecture and heavy training costs. Considering the intricate connections between trajectories, using Graph Neural Networks (GNNs) for data modeling is feasible. However, most methods directly use existing mathematical graph structures as the input instead of constructing specific graphs from certain vehicle trajectory data. This ignores such data's unique and dynamic characteristics. To bridge such a research gap, we propose VeTraSS -- an end-to-end pipeline for Vehicle Trajectory Similarity Search. Specifically, VeTraSS models the original trajectory data into multi-scale graphs, and generates comprehensive embeddings through a novel multi-layer attention-based GNN. The learned embeddings can be used for searching similar vehicle trajectories. Extensive experiments on the Porto and Geolife datasets demonstrate the effectiveness of VeTraSS, where our model outperforms existing work and reaches the state-of-the-art. This demonstrates the potential of VeTraSS for trajectory analysis and safe navigation in self-driving vehicles in the real world.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑