arXiv:2501.13347cs.LGcs.AI2025-01被引 6

一个模型搞定所有轨迹任务,效果比现有方法好13%以上。

One Fits All: General Mobility Trajectory Modeling via Masked Conditional Diffusion

  • 用掩码条件统一不同任务格式,通过上下文嵌入适应复杂条件。
  • 在轨迹生成任务中性能提升超13%,多任务测试均优于现有方法。
  • 适合需要通用轨迹建模的科研与城市规划场景。

轨迹数据在网络优化、城市规划等应用中至关重要。现有研究多为任务专用,仅适用于特定任务如生成、恢复或预测,难以跨任务迁移。尽管各类轨迹任务在输入、输出、目标和条件上差异显著,但共享共同的移动模式。基于此,我们提出一种通用轨迹建模框架——GenMove,通过掩码条件统一不同任务格式,并利用历史轨迹数据获取包含时空特征与用户偏好在内的上下文嵌入。通过无分类器引导将上下文嵌入融入扩散模型,使模型能灵活响应不同条件。大量实验表明,该模型在主流任务上显著优于当前最优基线,在生成任务中最高提升超过13%。

原文摘要 · Abstract (English)

Trajectory data play a crucial role in many applications, ranging from network optimization to urban planning. Existing studies on trajectory data are task-specific, and their applicability is limited to the specific tasks on which they have been trained, such as generation, recovery, or prediction. However, the potential of a unified model has not yet been fully explored in trajectory modeling. Although various trajectory tasks differ in inputs, outputs, objectives, and conditions, they share common mobility patterns. Based on these common patterns, we can construct a general framework that enables a single model to address different tasks. However, building a trajectory task-general framework faces two critical challenges: 1) the diversity in the formats of different tasks and 2) the complexity of the conditions imposed on different tasks. In this work, we propose a general trajectory modeling framework via masked conditional diffusion (named GenMove). Specifically, we utilize mask conditions to unify diverse formats. To adapt to complex conditions associated with different tasks, we utilize historical trajectory data to obtain contextual trajectory embeddings, which include rich contexts such as spatiotemporal characteristics and user preferences. Integrating the contextual trajectory embedding into diffusion models through a classifier-free guidance approach allows the model to flexibly adjust its outputs based on different conditions. Extensive experiments on mainstream tasks demonstrate that our model significantly outperforms state-of-the-art baselines, with the highest performance improvement exceeding 13% in generation tasks.

轨迹建模扩散模型通用框架

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