arXiv:2605.00365cs.LGcs.CL2026-05被引 3

解决推理模型输出单一化问题,让正确答案更均衡分布。

Uniform-Correct Policy Optimization: Breaking RLVR's Indifference to Diversity

论文配图:Uniform-Correct Policy Optimization: Breaking RLVR's Indifference to Diversity
图 1 · 摘自论文原文
  • 在GRPO基础上加均匀性惩罚,引导模型均匀分配正确答案概率。
  • 在五项数学推理任务上,Pass@64提升最高达10%,正确解多样性提高45%。
  • 适合追求输出多样性的大模型推理优化,尤其适用于数学与逻辑任务。

强化学习中可验证奖励(RLVR)在单次尝试准确率(Pass@1)上取得显著进展,但常出现多样本覆盖率下降(Pass@K),表明输出多样性崩溃。我们发现其根源在于:常见RLVR目标如GRPO对正确解之间的概率分布无差别,结合随机训练动态,导致概率质量集中于少数正确输出,抑制其他有效解。我们形式化了这一崩溃机制,并基于鲁棒性与熵正则最优性准则,证明均匀正确策略(Uniform-Correct Policy)为唯一最优。据此提出统一正确策略优化(UCPO),在GRPO中加入条件均匀性惩罚,使梯度信号向低频正确响应转移,实现正确解内概率均匀分配。在三个模型(1.5B-7B参数)和五个数学推理基准上,UCPO在保持竞争力的Pass@1的同时,显著提升Pass@K与多样性,在AIME24上Pass@64最高提升10%,正确解层面多样性最高提升45%。代码见https://github.com/AnamikaLochab/UCPO。

原文摘要 · Abstract (English)

Reinforcement Learning with Verifiable Rewards (RLVR) has achieved substantial gains in single-attempt accuracy (Pass@1) on reasoning tasks, yet often suffers from reduced multi-sample coverage (Pass@K), indicating diversity collapse. We identify a structural cause for this degradation: common RLVR objectives, such as GRPO, are indifferent to how probability mass is distributed among correct solutions. Combined with stochastic training dynamics, this indifference induces a self-reinforcing collapse, in which probability mass concentrates on a narrow subset of correct outputs while alternative valid solutions are suppressed. We formalize this collapse mechanism and further characterize the optimal policy structure under two complementary criteria: robustness and entropy-regularized optimality, which identify the Uniform-Correct Policy as uniquely optimal. Motivated by this analysis, we propose Uniform-Correct Policy Optimization (UCPO), a modification to GRPO that adds a conditional uniformity penalty on the policy's distribution over correct solutions. The penalty redistributes gradient signal toward underrepresented correct responses, encouraging uniform allocation of probability mass within the correct set. Across three models (1.5B-7B parameters) and five mathematical reasoning benchmarks, UCPO improves Pass@K and diversity while maintaining competitive Pass@1, achieving up to +10\% absolute improvement on AIME24 at Pass@64 and up to 45\% higher equation-level diversity within the correct set. The code is available at https://github.com/AnamikaLochab/UCPO.

强化学习推理优化多样性提升数学推理

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