arXiv:2505.12672cs.LG2025-05NeurIPS被引 5

一个可跨区域和任务迁移的车辆轨迹学习模型,无需重新训练。

TransferTraj: A Vehicle Trajectory Learning Model for Region and Task Transferability

  • 通过多模态融合与空间上下文门控机制,提升区域迁移能力。
  • 统一输入输出结构,实现一次预训练适配多种下游任务。
  • 在零样本和少样本区域迁移场景下表现优异,适合实际交通应用。

车辆GPS轨迹蕴含丰富的移动信息,支撑多种下游任务。理想的轨迹学习模型应具备跨区域和跨任务迁移能力,避免维护多个专用模型及小样本下的性能下降。然而,各区域的空间特征和上下文差异显著,影响轨迹模式的泛化性;不同任务间输入输出结构差异也带来迁移挑战。现有方法多依赖轨迹嵌入向量,区域迁移效果差,任务迁移需重训练预测模块。为此,本文提出TransferTraj,其核心为RTTE模块,融合轨迹的空间、时间、POI及道路网络多模态信息,有效应对区域间空间分布差异;引入TRIE模块捕捉空间特征相对关系,设计空间上下文MoE模块处理多样化运动模式。针对任务迁移,提出可迁移的输入输出方案,将不同任务统一为模态与轨迹点的掩码恢复任务,实现一次预训练即可迁移到多种任务。在三个真实世界车辆轨迹数据集上,通过任务迁移、零样本和少样本区域迁移实验,验证了TransferTraj的有效性。

原文摘要 · Abstract (English)

Vehicle GPS trajectories provide valuable movement information that supports various downstream tasks and applications. A desirable trajectory learning model should be able to transfer across regions and tasks without retraining, avoiding the need to maintain multiple specialized models and subpar performance with limited training data. However, each region has its unique spatial features and contexts, which are reflected in vehicle movement patterns and difficult to generalize. Additionally, transferring across different tasks faces technical challenges due to the varying input-output structures required for each task. Existing efforts towards transferability primarily involve learning embedding vectors for trajectories, which perform poorly in region transfer and require retraining of prediction modules for task transfer. To address these challenges, we propose TransferTraj, a vehicle GPS trajectory learning model that excels in both region and task transferability. For region transferability, we introduce RTTE as the main learnable module within TransferTraj. It integrates spatial, temporal, POI, and road network modalities of trajectories to effectively manage variations in spatial context distribution across regions. It also introduces a TRIE module for incorporating relative information of spatial features and a spatial context MoE module for handling movement patterns in diverse contexts. For task transferability, we propose a task-transferable input-output scheme that unifies the input-output structure of different tasks into the masking and recovery of modalities and trajectory points. This approach allows TransferTraj to be pre-trained once and transferred to different tasks without retraining. Extensive experiments on three real-world vehicle trajectory datasets under task transfer, zero-shot, and few-shot region transfer, validating TransferTraj's effectiveness.

轨迹预测迁移学习多模态

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