arXiv:2510.08696cs.LG2025-10被引 12

让错误样本变有用:通过置信度加权重用无正确回答的训练组

Don't Waste Mistakes: Leveraging Negative RL-Groups via Confidence Reweighting

  • 基于最大似然目标,将错误生成的置信度融入梯度更新
  • 在MATH基准上显著提升模型性能,尤其在难题上增益明显
  • 无需额外标注,适合希望提升强化学习效率的研究者

基于可验证奖励的强化学习(RLVR)已成为提升大语言模型推理能力的标准方法,其中组相对策略优化(GRPO)被广泛使用。然而,GRPO会浪费大量计算资源:当某组中所有采样回复均不正确时,其优势为零,无法产生梯度。本文提出能否在不增加监督的前提下利用负组。从奖励建模的最大似然(MLE)目标出发,我们证明了MLE梯度等价于对一个修正价值函数的策略梯度。该价值函数对错误响应施加置信度加权惩罚,越自信的错误惩罚越大。我们称之为L-ENS(Likelihood Estimation with Negative Samples)。LENS修改GRPO,为错误生成赋予非零、依赖置信度的奖励,使负组具备信息量,将原本浪费的样本转化为有效梯度更新。在使用Llama-3.1-8B和Qwen-2.5-3B模型的MATH基准测试中,该方法持续优于基线GRPO,尤其在高难度题目上表现突出。结果表明,这是一种原理严谨且实用的“回收”负组方法,能有效提升RLVR的效率与性能。

原文摘要 · Abstract (English)

Reinforcement learning with verifiable rewards (RLVR) has become a standard recipe for improving large language models (LLMs) on reasoning tasks, with Group Relative Policy Optimization (GRPO) widely used in practice. Yet GRPO wastes substantial compute on negative groups: groups in which no sampled response is correct yield zero advantage and thus no gradient. We ask whether negative groups can be leveraged without extra supervision. Starting from a maximum-likelihood (MLE) objective in reward modeling, we show that the MLE gradient is equivalent to a policy gradient for a modified value function. This value function adds a confidence-weighted penalty on incorrect responses, imposing larger penalties on more confident mistakes. We refer to this as \textbf{L}ikelihood \textbf{E}stimation with \textbf{N}egative \textbf{S}amples (\textbf{LENS}). LENS modifies GRPO to assign non-zero, confidence-dependent rewards to incorrect generations, making negative groups informative and converting previously wasted samples into useful gradient updates. On the MATH benchmark with Llama-3.1-8B and Qwen-2.5-3B, the proposed variant consistently outperforms GRPO baseline, with significant gains on harder items. These results demonstrate a principled and practical way to "rescue" negative groups, improving efficiency and performance in RLVR.

强化学习大模型训练效率

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