arXiv:2510.15110cs.LGcs.AI2025-10被引 28

用强化学习让大模型每字更聪明,输出变短70%仍更准。

DLER: Doing Length pEnalty Right - Incentivizing More Intelligence per Token via Reinforcement Learning

  • 用批量奖励归一化等四策略改进强化学习优化,解决长度惩罚失效问题。
  • 输出长度减少70%以上,准确率超越所有此前基线模型。
  • 适合追求高效推理、资源受限或缺乏训练数据的场景。

推理型语言模型如OpenAI-o1、DeepSeek-R1和Qwen通过长思维链取得优异表现,但常生成冗余内容。提升每令牌的智能度——即准确率与响应长度之比——仍是未解难题。本文重新审视以最简长度惩罚(截断)为基础的强化学习,发现准确率下降并非源于惩罚机制不足,而是强化学习优化不充分所致。我们识别出三大挑战:(i)优势估计偏差大,(ii)熵崩溃,(iii)奖励信号稀疏。为此提出DLER训练方案,结合批量奖励归一化、更高裁剪阈值、动态采样及简单截断长度惩罚。DLER实现当前最优的准确率-效率权衡:输出长度减少超70%,准确率全面超越先前基线。同时提升测试时扩展性:相比DeepSeek-R1-7B,DLER-7B可并行生成多个简洁回答,准确率提高28%且延迟更低。进一步提出基于难度感知的DLER,对简单问题自适应收紧截断以获更高效率。还提出一种选择性更新合并方法,在保留原始模型准确率的同时,继承DLER模型的简洁推理能力,适用于强化学习数据稀缺场景。

原文摘要 · Abstract (English)

Reasoning language models such as OpenAI-o1, DeepSeek-R1, and Qwen achieve strong performance via extended chains of thought but often generate unnecessarily long outputs. Maximizing intelligence per token--accuracy relative to response length--remains an open problem. We revisit reinforcement learning (RL) with the simplest length penalty--truncation--and show that accuracy degradation arises not from the lack of sophisticated penalties but from inadequate RL optimization. We identify three key challenges: (i) large bias in advantage estimation, (ii) entropy collapse, and (iii) sparse reward signal. We address them with Doing Length pEnalty Right (DLER), a training recipe combining batch-wise reward normalization, higher clipping, dynamic sampling, and a simple truncation length penalty. DLER achieves state-of-the-art accuracy--efficiency trade-offs, cutting output length by over 70 percent while surpassing all previous baseline accuracy. It also improves test-time scaling: compared to DeepSeek-R1-7B, DLER-7B generates multiple concise responses in parallel with 28 percent higher accuracy and lower latency. We further introduce Difficulty-Aware DLER, which adaptively tightens truncation on easier questions for additional efficiency gains. We also propose an update-selective merging method that preserves baseline accuracy while retaining the concise reasoning ability of the DLER model, which is useful for scenarios where RL training data is scarce.

强化学习推理优化高效生成模型压缩

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