通过环境观察回溯优化智能体自我蒸馏,提升多轮任务成功率。
HERO: Hindsight-Enhanced Reflection from Environment Observations for Agentic Self-Distillation
- 用后续环境观测作为局部对齐反馈,生成每一步的诊断信息。
- 在TauBench和WebShop上成功率达92.3%,减少17%无效轮次。
- 适合训练轮次受限、成功轨迹稀少的复杂交互场景。
强化学习通常依赖轨迹的最终结果来提升多轮智能体能力,但难以对中间步骤进行信用分配。近期的在线自我蒸馏方法通过自教师将特权反馈转化为密集的词元级监督,提供了一种有前景的替代方案。然而,我们发现将该范式直接扩展到多轮设置时会出现意外性能下降,其根源在于特权反馈(如成功轨迹或终端结果)与学生当前决策上下文之间缺乏对齐。为此,我们提出HERO——一种基于回溯增强的自我蒸馏框架,利用下一步环境观测作为局部对齐反馈。每轮结束后,HERO会回溯已完成的交互,将每个观测转化为紧凑的回合级诊断,捕捉关于原动作的可操作反馈,如必要性、有效性或失败原因。在TauBench和WebShop上,HERO相比仅依赖环境反馈的自蒸馏和GRPO方法显著提升了任务成功率,并减少了17%的无效轮次。尤其在训练轮次预算有限、成功回溯稀少的场景下表现更优,此时GRPO提供的奖励对比信号较弱。
原文摘要 · Abstract (English)
Reinforcement learning typically improves multi-turn agent capabilities through the terminal outcome of the trajectories, which makes it difficult to determine credit assignments for each intermediate turns. Recent on-policy self-distillation methods offer a promising alternative by converting privileged feedback into dense token-level supervision through a self-teacher. Our study is motivated by the unexpected performance degradation observed when naively extending this paradigm to multi-turn settings, which we attribute to a lack of alignment between privileged feedback, such as successful trajectories or terminal outcomes, and the student's current decision context. We introduce HERO, a hindsight-enhanced self-distillation framework that uses next environment observations as locally aligned feedback. After each rollout, HERO reflects on the completed interaction to convert each observation into a compact turn-level diagnosis, that captures actionable feedback about the original action such as its necessity, validity or failure cause. On TauBench and WebShop, HERO improves task success and reduces unnecessary turns over environment-feedback-only self-distillation and GRPO. It is especially effective under limited training turn budgets, where successful rollouts are rare and GRPO provides weak reward-contrast signals.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。