让提示能随推理模型错误自适应调整,提升强化学习训练效果。
Learning to Hint for Reinforcement Learning
- 提示生成与推理模型联合训练,动态适应当前错误
- 实验显示在多个基准上优于传统方法,提升稳定性和性能
- 引入提示依赖度,确保提示信号能有效迁移到无提示场景
组相对策略优化(GRPO)在具有可验证奖励的强化学习中广泛应用,但常面临优势坍塌问题:当一组采样轨迹均获相同奖励时,相对优势为零,导致无学习信号。例如,若问题过于困难,所有采样路径均错误且得分为零。近期工作通过添加提示或辅助结构来解决此问题,使推理模型产生混合结果并恢复更新信号。然而,现有提示多为固定,未随当前推理模型动态调整;且在带提示输入下有效的提示,未必能提升测试时无提示策略的表现。为此,本文提出提示学习框架HiLL,联合训练提示生成策略与推理策略。对于每个难题,提示策略基于当前推理模型的错误轨迹在线生成提示,实现提示生成对模型演进错误的自适应。我们进一步提出提示依赖度,衡量正确轨迹对提示的依赖强度,并推导出迁移性结果:提示依赖度越低,提示成功越可能迁移到无提示成功。据此设计转移加权奖励以训练提示策略。因此,HiLL不仅恢复有效学习信号,还偏好能提升原始无提示策略的提示。多基准实验表明,HiLL持续优于GRPO及已有提示基线,证明自适应与可迁移提示学习在强化学习中的价值。代码已开源。
原文摘要 · Abstract (English)
Group Relative Policy Optimization (GRPO) is widely used for reinforcement learning with verifiable rewards, but it often suffers from advantage collapse: when all rollouts in a group receive the same reward, the group yields zero relative advantage and thus no learning signal. For example, if a question is too hard for the reasoner, all sampled rollouts can be incorrect and receive zero reward. Recent work addresses this issue by adding hints or auxiliary scaffolds to such hard questions so that the reasoner produces mixed outcomes and recovers a non-zero update. However, existing hints are usually fixed rather than adapted to the current reasoner, and a hint that creates learning signal under the hinted input does not necessarily improve the no-hint policy used at test time. To this end, we propose Hint Learning for Reinforcement Learning (HiLL), a framework that jointly trains a hinter policy and a reasoner policy during RL. For each hard question, the hinter generates hints online conditioned on the current reasoner's incorrect rollout, allowing hint generation to adapt to the reasoner's evolving errors. We further introduce hint reliance, which measures how strongly correct hinted trajectories depend on the hint. We derive a transferability result showing that lower hint reliance implies stronger transfer from hinted success to no-hint success, and we use this result to define a transfer-weighted reward for training the hinter. Therefore, HiLL favors hints that not only recover informative GRPO groups, but also produce signals that are more likely to improve the original no-hint policy. Experiments across multiple benchmarks show that HiLL consistently outperforms GRPO and prior hint-based baselines, demonstrating the value of adaptive and transfer-aware hint learning for RL. The code is available at https://github.com/Andree-9/HiLL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。