让AI推理更简洁:用短回答自动引导模型避免过度思考
Implicit Compression Regularization: Concise Reasoning via Internal Shorter Distributions in RL Post-Training

- 通过分析推理过程中的最短正确回答,生成压缩信号
- 在多个数学和知识类任务上缩短输出长度同时保持或提升准确率
- 适合需要高效精准推理的场景,如教育、智能客服
基于可验证奖励的强化学习能提升大模型推理能力,但常导致过度思考,即生成过长的推理链条。现有方法多依赖长度惩罚或提前退出策略,前者可能降低准确率并引发思考不足,后者假设推理中可安全截断大量内容。我们重新审视压缩训练动态发现:长度与准确率的相关性初始为负,随压缩逐渐变正,说明短响应初期更可能正确,但随策略演化会失去这一优势。据此定义过思考:负相关为过思考,正相关为思考不足。在过思考阶段,最短正确响应平均比群体平均更短,是天然的压缩目标。为此提出隐式压缩正则化(ICR),利用采样组内最短正确响应构造虚拟短分布,作为在线策略的压缩信号,引导模型走向简洁且正确的路径。训练动态显示,ICR在压缩过程中维持更好的长度-准确率相关性,确保短响应持续与正确性对齐。在三个推理模型及多个数学和知识密集型基准上,ICR一致缩短输出,同时保持或提升准确率,实现更强的准确率-长度帕累托前沿。
原文摘要 · Abstract (English)
Reinforcement learning with verifiable rewards improves LLM reasoning but often induces overthinking, where models generate unnecessarily long reasoning traces. Existing methods mainly rely on length penalties or early-exit strategies; however, the former may degrade accuracy and induce underthinking, whereas the latter assumes that substantial portions of reasoning traces can be safely truncated. To obtain a compression signal without these limitations, we revisit the training dynamics of existing compression methods. We observe that the length--accuracy correlation is initially negative but continually increases during compression, indicating that shorter responses are initially more likely to be correct but gradually lose this property as the policy moves toward underthinking. Based on this observation, we formalize overthinking: a negative correlation indicates an overthinking regime, while a positive one indicates underthinking. When overthinking, the shortest correct responses are shorter than the group-average response length in expectation, making them natural compression targets already present in on-policy rollouts. We therefore propose \emph{Implicit Compression Regularization} (ICR), an on-policy regularization method whose compression signal comes from a virtual shorter distribution induced by the shortest correct responses in rollout groups, guiding the policy toward concise yet correct trajectories. Training dynamics show that ICR maintains a better length--accuracy correlation during compression, indicating that short responses remain better aligned with correctness instead of drifting toward underthinking. Experiments on three reasoning backbones and multiple mathematical and knowledge-intensive benchmarks show that ICR consistently shortens responses while preserving or improving accuracy, achieving a stronger accuracy--length Pareto frontier.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。