让大模型学会从失败中吸取经验,提升强化学习的探索效率
Experience is the Best Teacher: Motivating Effective Exploration in Reinforcement Learning for LLMs
- 用失败轨迹和未达标评分作为回顾性经验,指导模型探索新行为
- 在多个基准测试中显著优于基线方法,提升推理能力
- 适合希望改进大模型推理能力的研究者与开发者
基于评分的强化学习在提升大语言模型通用推理能力方面取得显著进展,但仍受限于当前策略分布下的低效探索。本文将RL优化视为引导策略向理想分布演进的过程,提出一种前瞻性经验引导的强化学习框架HeRL。HeRL将失败轨迹及其未满足的评分作为回顾性经验,作为上下文提示引导策略探索超出当前分布的期望响应。同时引入奖励增量,激励模型生成在该提示下具有更大改进潜力的回应。该方法使模型能有效学习高质量样本,避免重复试错,理论上更准确估计期望梯度。大量实验表明,HeRL在多个基准上均优于基线,且可在测试时通过经验引导自提升进一步增益。代码已开源。
原文摘要 · Abstract (English)
Reinforcement Learning (RL) with rubric-based rewards has recently shown remarkable progress in enhancing general reasoning capabilities of Large Language Models (LLMs), yet still suffers from ineffective exploration confined to curent policy distribution. In fact, RL optimization can be viewed as steering the policy toward an ideal distribution that maximizes the rewards, while effective exploration should align efforts with desired target. Leveraging this insight, we propose HeRL, a Hindsight experience guided Reinforcement Learning framework to bootstrap effective exploration by explicitly telling LLMs the desired behaviors specified in rewards. Concretely, HeRL treats failed trajectories along with their unmet rubrics as hindsight experience, which serves as in-context guidance for the policy to explore desired responses beyond its current distribution. Additionally, we introduce a bonus reward to incentivize responses with greater potential for improvement under such guidance. HeRL facilitates effective learning from desired high quality samples without repeated trial-and-error from scratch, yielding a more accurate estimation of the expected gradient theoretically. Extensive experiments across various benchmarks demonstrate that HeRL achieves superior performance gains over baselines, and can further benefit from experience guided self-improvement at test time. Our code is available at https://github.com/sikelifei/HeRL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。