通过熵调制梯度,让大模型长程决策更稳定高效
Harnessing Uncertainty: Entropy-Modulated Policy Gradients for Long-Horizon LLM Agents
- 根据每步不确定性动态调整梯度大小,优化学习信号
- 在3个复杂任务上显著超越现有基线,成功率提升15%以上
- 适合需要长期规划与稳定探索的智能体系统
在长时序任务中,基于大语言模型(LLM)的智能体面临奖励稀疏的问题,难以对中间步骤进行有效信用分配。以往方法主要通过构建密集奖励信号来引导学习,如逆强化学习或过程奖励模型。本文揭示了LLM学习动态中的根本问题:策略梯度大小与熵值天然耦合,导致对自信正确动作更新过小,而对不确定动作可能引发不稳定的大更新。为此,我们提出熵调制策略梯度(EMPG),根据步骤级不确定性与最终任务结果重新校准学习信号。EMPG增强自信正确动作的更新,惩罚自信错误,并抑制不确定步骤的更新以稳定探索。此外引入未来清晰度奖励项,鼓励智能体选择更可预测的解题路径。在WebShop、ALFWorld和Deep Search三个挑战性任务上的实验表明,EMPG实现显著性能提升,显著优于多个强基线方法。
原文摘要 · Abstract (English)
In long-horizon tasks, recent agents based on Large Language Models (LLMs) face a significant challenge that sparse, outcome-based rewards make it difficult to assign credit to intermediate steps. Previous methods mainly focus on creating dense reward signals to guide learning, either through traditional reinforcement learning techniques like inverse reinforcement learning or by using Process Reward Models for step-by-step feedback. In this paper, we identify a fundamental problem in the learning dynamics of LLMs: the magnitude of policy gradients is inherently coupled with the entropy, which leads to inefficient small updates for confident correct actions and potentially destabilizes large updates for uncertain ones. To resolve this, we propose Entropy-Modulated Policy Gradients (EMPG), a framework that re-calibrates the learning signal based on step-wise uncertainty and the final task outcome. EMPG amplifies updates for confident correct actions, penalizes confident errors, and attenuates updates from uncertain steps to stabilize exploration. We further introduce a bonus term for future clarity that encourages agents to find more predictable solution paths. Through comprehensive experiments on three challenging agent tasks, WebShop, ALFWorld, and Deep Search, we demonstrate that EMPG achieves substantial performance gains and significantly outperforms strong policy gradient baselines. Project page is at https://empgseed-seed.github.io/
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。