arXiv:2510.06870cs.CL2025-10被引 12

让大模型自己学着判断哪些词重要,提升推理能力。

$λ$-GRPO: Unifying the GRPO Frameworks with Learnable Token Preferences

  • 引入可学习的权重参数λ,动态调整不同词的优化重要性。
  • 在多个数学推理任务上,准确率比传统方法提升1.0%~1.9%。
  • 无需改数据或加算力,适合实际部署的强化学习训练场景。

基于人类反馈的强化学习(RLHF)是提升大语言模型推理能力的主流方法。近期基于可验证奖励的强化学习(RLVR)通过规则验证器替代奖励与价值模型,简化了该范式。其中代表性方法为组相对策略优化(GRPO),但其存在长度偏差:相同优势值被均分给响应中所有词元,导致长文本在梯度更新中占更大权重。已有改进如DAPO和Dr. GRPO通过修改词元级损失聚合方式缓解此问题,但依赖启发式设计且缺乏可解释性。本文提出统一现有框架的新方法λ-GRPO,引入可学习参数λ以自适应控制词元权重。实验表明,在1.5B、3B、7B参数的Qwen2.5模型上,λ-GRPO相比GRPO分别实现+1.9%、+1.0%、+1.7%的平均准确率提升。这些增益不依赖训练数据修改或额外计算开销,凸显学习词元偏好机制的有效性与实用性。

原文摘要 · Abstract (English)

Reinforcement Learning with Human Feedback (RLHF) has been the dominant approach for improving the reasoning capabilities of Large Language Models (LLMs). Recently, Reinforcement Learning with Verifiable Rewards (RLVR) has simplified this paradigm by replacing the reward and value models with rule-based verifiers. A prominent example is Group Relative Policy Optimization (GRPO). However, GRPO inherently suffers from a length bias, since the same advantage is uniformly assigned to all tokens of a response. As a result, longer responses distribute the reward over more tokens and thus contribute disproportionately to gradient updates. Several variants, such as DAPO and Dr. GRPO, modify the token-level aggregation of the loss, yet these methods remain heuristic and offer limited interpretability regarding their implicit token preferences. In this work, we explore the possibility of allowing the model to learn its own token preference during optimization. We unify existing frameworks under a single formulation and introduce a learnable parameter $λ$ that adaptively controls token-level weighting. We use $λ$-GRPO to denote our method, and we find that $λ$-GRPO achieves consistent improvements over vanilla GRPO and DAPO on multiple mathematical reasoning benchmarks. On Qwen2.5 models with 1.5B, 3B, and 7B parameters, $λ$-GRPO improves average accuracy by $+1.9\%$, $+1.0\%$, and $+1.7\%$ compared to GRPO, respectively. Importantly, these gains come without any modifications to the training data or additional computational cost, highlighting the effectiveness and practicality of learning token preferences.

强化学习大模型推理增强可学习权重

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