arXiv:2604.16972cs.AI2026-04

提升大模型推理能力,解决训练信号消失问题

MCPO: Mastery-Consolidated Policy Optimization for Large Reasoning Models

论文配图:MCPO: Mastery-Consolidated Policy Optimization for Large Reasoning Models
图 1 · 摘自论文原文
  • 针对已掌握题目设计铰链式KL正则,防止策略漂移
  • 优先优化多数正确但未完全正确的题目,增强学习巩固
  • 在数学基准上显著提升准确率,促进解法多样性

基于可验证奖励的强化学习(RLVR)已成为提升大语言模型(LLMs)推理能力的有力方法。其中,组相对策略优化(GRPO)及其变体表现出优异性能和高训练效率。然而,GRPO类目标在高准确率提示上存在两个问题:对于已掌握提示(回滚准确率=1),组相对优势消失,导致无训练信号,引发策略漂移与遗忘;对于多数正确提示(回滚准确率在(0.5,1)之间),诱导查询权重随准确率上升而减小,削弱从部分正确向完全掌握的巩固。为此,我们提出精通巩固策略优化(MCPO),引入(i)仅作用于已掌握提示的铰链式KL正则,限制连续梯度步间的有害策略漂移;(ii)一种加权机制,优先分配优化资源给多数正确提示。在三个数学基准上的大量实验表明,MCPO持续提升pass@1性能。出人意料的是,MCPO不仅未抑制探索,反而提升pass@k指标,表明精通巩固进一步催化了解法多样性。

原文摘要 · Abstract (English)

Reinforcement Learning with Verifiable Rewards (RLVR) has emerged as a promising approach to improve the reasoning abilities of Large Language Models (LLMs). Among RLVR algorithms, Group Relative Policy Optimization (GRPO) and its variants have demonstrated strong performance and high training efficiency. However, GRPO-style objectives exhibit two issues on high accuracy prompts including mastered prompts (rollout accuracy =1) and majority-correct prompts (rollout accuracy in (0.5,1)). For mastered prompts, group-relative advantages vanish, yielding no training signal and unconstrained policy drift that can cause forgetting. For majority-correct prompts, the induced query weight shrinks as accuracy increases, weakening consolidation from partial correctness to mastery. To alleviate this, we propose Mastery-Consolidated Policy Optimization (MCPO), which introduces (i) a hinge-KL regularizer applied exclusively to mastered prompts to bound harmful policy drift between successive gradient steps, and (ii) a weighting mechanism that prioritizes majority-correct prompts to better allocate optimization effort. Extensive experiments across three mathematical benchmarks demonstrate that MCPO consistently improves pass@1 performance. Counter-intuitively, rather than restricting exploration, MCPO boosts pass@k metrics, indicating that mastery consolidation further catalyzes solution diversity.

强化学习大模型推理策略优化数学推理

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