arXiv:2411.14768cs.LGcs.AI2024-11KDD被引 25

融合网格与道路轨迹,提升轨迹表示学习效果

Grid and Road Expressions Are Complementary for Trajectory Representation Learning

  • 双路编码:分别处理网格和道路轨迹信息
  • 对比与重建损失使两模态互补,提升表征质量
  • 适合需要精准轨迹理解的任务,如路径预测

轨迹表示学习(TRL)将轨迹映射为可支持多种下游任务的向量。现有方法仅使用网格轨迹(自由空间运动)或道路轨迹(道路网络运动)作为输入。我们观察到二者具有互补性:前者提供区域与位置信息,后者体现道路结构与运动规律。为此,提出GREEN方法,联合利用网格与道路轨迹表达进行有效表示学习。具体地,将原始GPS轨迹转换为网格与道路轨迹,分别设计编码器捕捉其特征;通过对比损失使同轨迹在两编码器下输出相似嵌入,并设计掩码语言模型(MLM)损失,用网格轨迹辅助重建被遮蔽的道路轨迹;最终通过双模态交互器,利用交叉注意力融合两编码器输出。在三个下游任务中,与7种先进方法对比,GREEN始终表现更优,平均提升最佳基线准确率15.99%。

原文摘要 · Abstract (English)

Trajectory representation learning (TRL) maps trajectories to vectors that can be used for many downstream tasks. Existing TRL methods use either grid trajectories, capturing movement in free space, or road trajectories, capturing movement in a road network, as input. We observe that the two types of trajectories are complementary, providing either region and location information or providing road structure and movement regularity. Therefore, we propose a novel multimodal TRL method, dubbed GREEN, to jointly utilize Grid and Road trajectory Expressions for Effective representatioN learning. In particular, we transform raw GPS trajectories into both grid and road trajectories and tailor two encoders to capture their respective information. To align the two encoders such that they complement each other, we adopt a contrastive loss to encourage them to produce similar embeddings for the same raw trajectory and design a mask language model (MLM) loss to use grid trajectories to help reconstruct masked road trajectories. To learn the final trajectory representation, a dual-modal interactor is used to fuse the outputs of the two encoders via cross-attention. We compare GREEN with 7 state-of-the-art TRL methods for 3 downstream tasks, finding that GREEN consistently outperforms all baselines and improves the accuracy of the best-performing baseline by an average of 15.99\%.

轨迹学习多模态表示学习

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