提出新方法让大模型推理更短更快,不牺牲准确率。
DRPO: Efficient Reasoning via Decoupled Reward Policy Optimization
- 分离正确推理的长度奖励,避免错误样本干扰
- 1.5B模型在GSM8k上减少77%长度,仅损失1.1%性能
- 适合追求高效推理的部署场景,如实时问答系统
近期基于强化学习的大规模推理模型(如GRPO)在复杂推理任务中表现优异,但存在过度思考问题,对简单问题生成冗长回答,显著增加计算开销和延迟。现有方法虽引入长度奖励以促进简洁推理,却导致性能下降。我们发现根本原因在于:正确但过长的推理路径被惩罚后,GRPO的组相对优势函数会赋予其负优势,从而抑制有效推理。为此,提出解耦奖励策略优化(DRPO),将正确推理路径的长度信号与错误路径分离。DRPO仅在正样本组内归一化正确路径奖励,避免负样本干扰。该方法通过最优正样本分布构建判别目标,结合KL正则化,在闭式解下实现仅使用在线策略数据和重要性加权的高效梯度计算。该框架具通用性,可拓展至其他正样本偏好奖励。在数学推理任务上的实验表明,相比六种基线方法,DRPO显著更优。以1.5B模型为例,在GSM8k上实现77%长度压缩,性能仅下降1.1%,而基线方法需牺牲4.3%性能换取68%长度缩减。
原文摘要 · Abstract (English)
Recent large reasoning models (LRMs) driven by reinforcement learning algorithms (e.g., GRPO) have achieved remarkable performance on challenging reasoning tasks. However, these models suffer from overthinking, generating unnecessarily long and redundant reasoning even for simple questions, which substantially increases computational cost and response latency. While existing methods incorporate length rewards to GRPO to promote concise reasoning, they incur significant performance degradation. We identify the root cause: when rewards for correct but long rollouts are penalized, GRPO's group-relative advantage function can assign them negative advantages, actively discouraging valid reasoning. To overcome this, we propose Decoupled Reward Policy Optimization (DRPO), a novel framework that decouples the length-based learning signal of correct rollouts from incorrect ones. DRPO ensures that reward signals for correct rollouts are normalized solely within the positive group, shielding them from interference by negative samples. The DRPO's objective is grounded in integrating an optimized positive data distribution, which maximizes length-based rewards under a KL regularization, into a discriminative objective. We derive a closed-form solution for this distribution, enabling efficient computation of the objective and its gradients using only on-policy data and importance weighting. Of independent interest, this formulation is general and can incorporate other preference rewards of positive data beyond length. Experiments on mathematical reasoning tasks demonstrate DRPO's significant superiority over six efficient reasoning baselines. Notably, with a 1.5B model, our method achieves 77\% length reduction with only 1.1\% performance loss on simple questions like GSM8k dataset, while the follow-up baseline sacrifices 4.3\% for 68\% length reduction.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。