arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2510.04474cs.AIcs.LG
收藏

DRPO:通过解耦奖励策略优化实现高效推理

DRPO: Efficient Reasoning via Decoupled Reward Policy Optimization

Texas A&M University(德克萨斯A&M大学);The University of Virginia(弗吉尼亚大学)

Gang Li, Yan Chen, Ming Lin, Tianbao Yang

AI总结:

DRPO通过解耦奖励策略优化,有效解决大型推理模型过度思考问题,实现高效推理并减少响应延迟。

摘要:

近期基于强化学习算法(例如GRPO)驱动的大型推理模型(LRMs)在具有挑战性的推理任务上取得了显著的性能。然而,这些模型存在过度思考的问题,即使在简单的问题上也会生成不必要的长且冗余的推理过程,这大大增加了计算成本和响应延迟。尽管现有方法将长度奖励纳入GRPO以促进简洁推理,但它们会导致显著的性能下降。我们发现根本原因:当正确但较长的回放的奖励被惩罚时,GRPO的组相对优势函数可以将其分配为负优势,主动 discourage 有效的推理。为克服这一问题,我们提出了解耦奖励策略优化(DRPO),一种新的框架,将正确回放的基于长度的学习信号与不正确回放分离。DRPO确保正确回放的奖励信号仅在正组内被归一化,从而避免受到负样本的干扰。DRPO的目标基于将优化的正数据分布整合到一个判别目标中,该目标在KL正则化下最大化基于长度的奖励。我们推导了该分布的闭式解,使得能够仅使用on-policy数据和重要加权高效地计算目标及其梯度。值得注意的是,这种形式是通用的,可以整合其他正数据的偏好奖励,除了长度之外。在数学推理任务上的实验表明,DRPO在六个高效推理基线中表现显著优于。值得注意的是,使用1.5B模型时,我们的方法在简单的GSM8k数据集问题上实现了77%的长度减少,仅以1.1%的性能损失,而后续基线则牺牲了4.3%以换取68%的长度减少。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑