arXiv:2602.06717cs.LGcs.AI2026-02被引 9

让强化学习避开常见路径,关注稀有正确解。

F-GRPO: Don't Let Your Policy Learn the Obvious and Forget the Rare

  • 设计难度感知系数,降低高成功率样本的更新权重。
  • 在N=8时数学任务准确率提升至70.3~76.8,优于基线。
  • 适用于大模型推理优化,不增加计算开销。

基于可验证奖励的强化学习(RLVR)通常采用分组采样估计优势并稳定策略更新。实际中,因计算限制无法使用过大的分组,训练仅能强化暴露的正确行为。在有限分组下,更新可能遗漏稀有但正确的轨迹,同时仍包含混合奖励,导致概率集中在更常见的采样解上。我们推导出提示局部尾部遗漏事件的概率与分组大小的关系,发现其非单调;在分类抽象中,未采样正确解的占比甚至在总正确量增长时持续缩小。受此启发,我们提出一种受焦点损失启发的难度感知缩放系数,降低高成功率分组的更新权重。实验表明,类别模拟验证了该效应,迷宫测试提供单解验证,大语言模型实验涵盖典型GRPO分组大小扫描及固定N下的跨方法迁移。在Qwen2.5-7B模型、N=8条件下,平均数学通过率从64.1→70.3(GRPO)、69.3→72.5(DAPO)、73.2→76.8(CISPO),OOD通过率同样提升,且无需增大分组或计算成本。

原文摘要 · Abstract (English)

Reinforcement Learning with Verifiable Rewards (RLVR) is commonly based on group sampling to estimate advantages and stabilize policy updates. In practice, computational limits often rule out very large groups, so training proceeds with finite rollout sets that can reinforce only the correct behavior they expose. At practical group sizes, updates can miss rare-correct trajectories while still containing mixed rewards, concentrating probability on more common sampled solutions. We derive the probability of such prompt-local tail-miss events as a function of group size, showing non-monotonic behavior, and in the categorical abstraction characterize how unsampled-correct mass can shrink even as total correct mass grows. Motivated by this analysis, we propose a difficulty-aware scaling coefficient, inspired by Focal loss, that down-weights updates on high-success sampled groups. Empirically, categorical simulation illustrates the same effect in the categorical setting, Maze provides a single-solution test, and LLM experiments include a representative GRPO group-size sweep together with fixed-$N$ transfer across GRPO, DAPO, and CISPO. On Qwen2.5-7B at $N{=}8$, our method improves average math pass@256 from 64.1 $\rightarrow$ 70.3 (GRPO), 69.3 $\rightarrow$ 72.5 (DAPO), and 73.2 $\rightarrow$ 76.8 (CISPO); OOD pass@256 also improves in all three cases, without increasing group size or computational cost.

强化学习大模型推理奖励设计

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。