arXiv:2603.18533cs.LGcs.CL2026-03

针对大模型过思与自满问题,提出分难度优化策略提升推理效率与准确率。

Balancing the Reasoning Load: Difficulty-Differentiated Policy Optimization with Length Redistribution for Efficient and Robust Reinforcement Learning

  • 按任务难易度分别优化,简单题缩短输出,复杂题拓展探索空间。
  • 相比GRPO,平均答案长度减少12%,准确率提升1.85%。
  • 理论推导出最优长度分布条件,适合追求高效可靠的RL应用者。

大型推理模型(LRMs)虽具强大推理能力,但常出现过度思考现象,生成冗长重复内容;对超出能力范围的问题,则表现出自满倾向,生成过短且错误的答案,导致性能下降。为此,我们提出难度分化的策略优化(DDPO),一种高效的强化学习算法,根据自满现象将简单与复杂任务分开优化:对简单任务压缩输出长度而不损失准确率,对复杂任务扩展探索空间以提升表现。我们进一步推导出最大化期望准确率的理论条件,要求输出长度分布应尽量接近最优长度且高度集中。基于此,提出以难度级平均值作为长度优化的合理参考。在域内与域外基准上的大量实验验证了DDPO的优越性与有效性。相较于GRPO,DDPO在多个基准上平均答案长度减少12%,准确率提升1.85%,实现了准确率与长度间的更优权衡。代码已开源于https://github.com/Yinan-Xia/DDPO。

原文摘要 · Abstract (English)

Large Reasoning Models (LRMs) have shown exceptional reasoning capabilities, but they also suffer from the issue of overthinking, often generating excessively long and redundant answers. For problems that exceed the model's capabilities, LRMs tend to exhibit the overconfidence phenomenon, generating overly short but incorrect answers, which may contribute to suboptimal performance. To address these issues, we propose Difficulty-Differentiated Policy Optimization (DDPO), an efficient reinforcement learning algorithm that optimizes simple and complex tasks separately based on the overconfidence phenomenon. Specifically, it reduces the output length for simple tasks without compromising accuracy, while for complex tasks, it expands the exploration space to improve performance. We further derive the theoretical conditions for maximizing expected accuracy, which require the length distribution to closely approximate the optimal length and be as concentrated as possible. Based on these conditions, we propose using the difficulty-level average as a well-founded reference for length optimization. Extensive experiments on both in-domain and out-of-domain benchmarks validate the superiority and effectiveness of DDPO. Compared to GRPO, DDPO reduces the average answer length by 12% while improving accuracy by 1.85% across multiple benchmarks, achieving a better trade-off between accuracy and length. The code is available at https://github.com/Yinan-Xia/DDPO.

强化学习推理优化模型效率

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