LLM2D
RED:基于全面信息的有效轨迹表示学习
RED: Effective Trajectory Representation Learning with Comprehensive Information
作者: Silin Zhou, Shuo Shang, Lisi Chen, Christian S. Jensen, Panos Kalnis
发布日期: 11/25/2024
arXiv ID: oai:arXiv.org:2411.15096v1

摘要

轨迹表示学习 (TRL) 将轨迹映射到向量,然后可用于各种下游任务,包括轨迹相似度计算、轨迹分类和行程时间估计。然而,现有的 TRL 方法生成的向量在下游任务中往往精度不足。一个关键原因是它们未能利用轨迹包含的全面信息。我们提出了一种名为 RED 的自监督 TRL 框架,它有效地利用了多种类型的轨迹信息。总体而言,RED 采用 Transformer 作为主干模型,并对构成轨迹的路径进行掩码以训练掩码自动编码器 (MAE)。特别是,RED 通过采用保留轨迹关键路径的道路感知掩码策略来考虑轨迹的移动模式,从而保留轨迹的关键信息。RED 还采用时空用户联合嵌入方案,在准备轨迹作为模型输入时对全面信息进行编码。为了进行训练,RED 采用双目标任务学习:Transformer 编码器预测轨迹中的下一段,而 Transformer 解码器重建整个轨迹。RED 还通过修改 Transformer 的注意力机制来考虑轨迹的时空相关性。我们在 3 个真实世界数据集上针对 4 个下游任务将 RED 与 9 种最先进的 TRL 方法进行了比较,发现 RED 通常可以将最佳基线的精度提高 5% 以上。