用大模型协同小模型,自动完成轨迹建模任务。
TrajAgent: An LLM-Agent Framework for Trajectory Modeling via Large-and-Small Model Collaboration
- 大模型指挥小模型协作,统一接口处理多种轨迹数据
- 在5个任务上比基线提升2.38%至69.91%
- 适合需要自动化轨迹分析的研究者与城市规划人员
轨迹建模(包括轨迹数据模式挖掘与未来预测)广泛应用于生活服务、城市交通和公共管理等领域。现有方法虽多样,但因数据异构性和任务多样性,仍难实现高效可靠的建模。本文提出TrajAgent,一个基于大语言模型的智能体框架,通过大模型与小模型协同,实现自动化轨迹建模。框架首先构建UniEnv,提供统一的数据与模型接口;在此基础上设计自动化工作流,支持跨任务、跨数据集的建模。同时引入大模型智能体与小型专用模型间的协同学习机制,显著提升整体性能。在四个真实数据集上的五个任务实验表明,该框架相比基线方法性能提升2.38%至69.91%。代码与数据已开源。
原文摘要 · Abstract (English)
Trajectory modeling, which includes research on trajectory data pattern mining and future prediction, has widespread applications in areas such as life services, urban transportation, and public administration. Numerous methods have been proposed to address specific problems within trajectory modeling. However, the heterogeneity of data and the diversity of trajectory tasks make effective and reliable trajectory modeling an important yet highly challenging endeavor, even for domain experts. In this paper, we propose TrajAgent, an agent framework powered by large language models, designed to facilitate robust and efficient trajectory modeling through automation modeling. This framework leverages and optimizes diverse specialized models to address various trajectory modeling tasks across different datasets effectively. In TrajAgent, we first develop UniEnv, an execution environment with a unified data and model interface, to support the execution and training of various models. Building on UniEnv, we introduce an agentic workflow designed for automatic trajectory modeling across various trajectory tasks and data. Furthermore, we introduce collaborative learning schema between LLM-based agents and small speciallized models, to enhance the performance of the whole framework effectively. Extensive experiments on five tasks using four real-world datasets demonstrate the effectiveness of TrajAgent in automated trajectory modeling, achieving a performance improvement of 2.38%-69.91% over baseline methods. The codes and data can be accessed via https://github.com/tsinghua-fib-lab/TrajAgent.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。