解决大模型智能体在长序列稀疏奖励任务中的探索崩溃问题
EPO: Entropy-regularized Policy Optimization for LLM Agents Reinforcement Learning
- 引入熵正则化框架,动态调节探索强度
- 在ScienceWorld上性能提升152%,ALFWorld提升19.8%
- 适合长序列交互、稀疏反馈的LLM智能体训练场景
在多轮互动环境中训练大模型智能体,单个任务需30轮以上交互,稀疏奖励带来根本性挑战。我们识别出一种独特失败模式:探索-利用级联失效。早期策略过早收敛,导致低熵错误策略固化;后期熵正则化反而加剧混乱探索,引发训练崩溃。为此提出熵正则化策略优化(EPO),通过三项协同机制破局:(1) 多轮场景下熵正则化增强探索;(2) 熵平滑项约束策略熵在历史均值范围内,防止剧烈波动;(3) 自适应阶段加权,动态平衡探索与利用。理论分析表明EPO保证熵方差单调下降且可收敛。在ScienceWorld上最高提升152%,ALFWorld提升19.8%。研究证明,多轮稀疏奖励设置需不同于传统强化学习的熵控制机制,对大模型智能体训练具有广泛意义。
原文摘要 · Abstract (English)
Training LLM agents in multi-turn environments with sparse rewards, where completing a single task requires 30+ turns of interaction within an episode, presents a fundamental challenge for reinforcement learning. We identify a critical failure mode unique to this setting: the exploration-exploitation cascade failure. This cascade begins with early-stage policy premature convergence, where sparse feedback causes agents to commit to flawed, low-entropy strategies. Subsequently, agents enter late-stage policy collapse, where conventional entropy regularization becomes counterproductive, promoting chaotic exploration that destabilizes training. We propose Entropy-regularized Policy Optimization (EPO), a general framework that breaks this failure cycle through three synergistic mechanisms: (1) adopting entropy regularization in multi-turn settings to enhance exploration, (2) an entropy smoothing regularizer that bounds policy entropy within historical averages to prevent abrupt fluctuations, and (3) adaptive phase-based weighting that balances exploration and exploitation across training. Our analysis justifies that EPO guarantees monotonically decreasing entropy variance while maintaining convergence. EPO achieves up to 152% performance improvement on ScienceWorld and up to 19.8% on ALFWorld. Our work demonstrates that multi-turn sparse-reward settings require fundamentally different entropy control than traditional RL, with broad implications for LLM agent training.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。