通过交替压缩扩展推理长度,让大模型更高效精准地思考。
SIRI: Scaling Iterative Reinforcement Learning with Interleaved Compression
- 训练时周期性缩短再放宽推理长度,动态调节思考效率。
- 三轮迭代后,准确率提升43.2%,令牌消耗减少46.9%。
- 适合追求高精度与低耗能推理的场景,如数学解题。
我们提出SIRI——一种基于交错压缩的迭代强化学习方法,用于大推理模型(LRMs),实现更高效且准确的推理。现有研究发现大模型存在重复思考现象,而抑制此类行为常以性能为代价。本文表明,通过在训练中迭代交替压缩与扩展推理预算(即动态调整最大回溯长度),可突破此权衡。压缩阶段缩短回溯长度,迫使模型在有限上下文中做出精准决策,有效减少冗余标记并提高推理密度;扩张阶段则放宽限制,使模型能在长程任务中探索与规划。令人惊讶的是,每完成一次压缩-扩张循环,模型性能持续提升,同时输出长度下降,逐步逼近性能-效率的帕累托最优前沿。在DeepSeek-R1-Distill-Qwen-1.5B上训练,SIRI-low在AIME24上性能提升43.2%,令牌使用量减少46.9%;SIRI-high相较所有其他方法达到最高准确率(图1)。结果揭示了训练期间周期性振荡输出截断长度的潜力,可动态平衡推理中的探索与效率,收敛至理想‘甜点’区域。模型已公开。
原文摘要 · Abstract (English)
We introduce SIRI, Scaling Iterative Reinforcement Learning with Interleaved Compression, a simple yet effective RL approach for Large Reasoning Models (LRMs) that enables more efficient and accurate reasoning. Existing studies have observed repetitive thinking patterns in LRMs, and attempts to reduce them often come at the cost of performance. In this paper, we show that this trade-off can be overcome through a training regime that iteratively alternates between compressing and expanding the reasoning budget, by dynamically adjusting the maximum rollout length during training. The compression phase cuts the rollout length, forcing the model to make precise and valuable decisions within a limited context, which effectively reduces redundant tokens and increases reasoning density. The expansion phase then relaxes the length limit, providing space for the model to explore and plan in long-horizon settings. Remarkably, we find that after each compression-expansion cycle, the model's performance improves even as its output length decreases, steadily pushing it closer to the Pareto frontier in the performance-efficiency trade-off. Training on DeepSeek-R1-Distill-Qwen-1.5B, SIRI-low improves performance on AIME24 by 43.2% while reducing token usage by 46.9% after three iterations, and SIRI-high achieves the highest accuracy compared to all other methods (Figure 1). Our findings shed light on the potential of periodically oscillating the LRM's output truncation length during training to dynamically balance exploration and efficiency in reasoning, converging towards an optimal "sweet spot" between the two. Our models are publicly available.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。