解决大模型推理中一致性导致梯度消失的问题,提升训练效率
COPO: Consistency-Aware Policy Optimization
- 设计基于结果一致性的全局奖励机制,避免相同输出时梯度消失
- 在多个数学推理基准上显著提升性能,最佳模型达92.3%准确率
- 适合需要稳定推理的复杂任务,如数学证明和逻辑推演
强化学习显著提升了大语言模型在复杂问题求解中的推理能力。近期DeepSeek R1的推出激发了对基于规则奖励作为低成本优势函数计算方式的兴趣。然而,多数复现与扩展工作中发现,当单个提示下的多个采样输出收敛至相同结果(无论正确与否)时,组内优势会退化为零,导致梯度消失,使样本无法用于学习,最终限制训练效率与下游表现。为此,我们提出一种一致性感知策略优化框架,引入基于结果一致性的结构化全局奖励。该全局损失确保即使模型输出具有高组内一致性,训练过程仍能获得有意义的学习信号,从全局视角鼓励生成正确且自一致的推理路径。此外,我们还引入基于熵的软融合机制,自适应平衡局部优势估计与全局优化,实现训练过程中探索与收敛的动态切换。我们在多个数学推理基准上验证了方法的有效性,展现出显著的性能提升与强泛化能力。代码已开源:https://github.com/hijih/copo-code.git。
原文摘要 · Abstract (English)
Reinforcement learning has significantly enhanced the reasoning capabilities of Large Language Models (LLMs) in complex problem-solving tasks. Recently, the introduction of DeepSeek R1 has inspired a surge of interest in leveraging rule-based rewards as a low-cost alternative for computing advantage functions and guiding policy optimization. However, a common challenge observed across many replication and extension efforts is that when multiple sampled responses under a single prompt converge to identical outcomes, whether correct or incorrect, the group-based advantage degenerates to zero. This leads to vanishing gradients and renders the corresponding samples ineffective for learning, ultimately limiting training efficiency and downstream performance. To address this issue, we propose a consistency-aware policy optimization framework that introduces a structured global reward based on outcome consistency, the global loss based on it ensures that, even when model outputs show high intra-group consistency, the training process still receives meaningful learning signals, which encourages the generation of correct and self-consistent reasoning paths from a global perspective. Furthermore, we incorporate an entropy-based soft blending mechanism that adaptively balances local advantage estimation with global optimization, enabling dynamic transitions between exploration and convergence throughout training. Our method introduces several key innovations in both reward design and optimization strategy. We validate its effectiveness through substantial performance gains on multiple mathematical reasoning benchmarks, highlighting the proposed framework's robustness and general applicability. Code of this work has been released at https://github.com/hijih/copo-code.git.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。