通过逐步校准优化大模型智能体的决策路径,提升复杂任务完成率。
STeCa: Step-level Trajectory Calibration for LLM Agent Learning
- 在探索中逐步对比奖励,识别低效动作
- 用大模型反思生成改进后的决策轨迹
- 适合需要长期规划与鲁棒性的任务学习
基于大语言模型(LLM)的智能体在动态交互中展现处理复杂任务的潜力。现有方法主要依赖专家示范的行为克隆或通过探索轨迹采样进行偏好学习,但在长周期任务中常因逐步累积的次优行为导致偏离正确路径。为此,我们强调及时校准的重要性,并提出一种自动构建校准轨迹的方法。本文提出一步级轨迹校准(STeCa)框架:在探索过程中通过逐步奖励比较识别次优动作,利用大模型驱动的反思生成校准轨迹,使智能体从优化后的决策过程学习。最终将校准轨迹与成功轨迹联合用于强化训练。大量实验表明,STeCa显著优于现有方法。进一步分析显示,及时校准可增强智能体在任务执行中的鲁棒性。代码与数据已公开于 https://github.com/WangHanLinHenry/STeCa。
原文摘要 · Abstract (English)
Large language model (LLM)-based agents have shown promise in tackling complex tasks by interacting dynamically with the environment. Existing work primarily focuses on behavior cloning from expert demonstrations or preference learning through exploratory trajectory sampling. However, these methods often struggle to address long-horizon tasks, where suboptimal actions accumulate step by step, causing agents to deviate from correct task trajectories. To address this, we highlight the importance of timely calibration and the need to automatically construct calibration trajectories for training agents. We propose Step-Level Trajectory Calibration (STeCa), a novel framework for LLM agent learning. Specifically, STeCa identifies suboptimal actions through a step-level reward comparison during exploration. It constructs calibrated trajectories using LLM-driven reflection, enabling agents to learn from improved decision-making processes. We finally leverage these calibrated trajectories with successful trajectories for reinforced training. Extensive experiments demonstrate that STeCa significantly outperforms existing methods. Further analysis highlights that timely calibration enables agents to complete tasks with greater robustness. Our code and data are available at https://github.com/WangHanLinHenry/STeCa.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。