arXiv:2508.03346cs.AI2025-08中稿 · ICLR被引 28

通过步骤熵压缩大模型推理过程,80%冗余步骤可删减且准确率几乎不变。

Making Slow Thinking Faster: Compressing LLM Chain-of-Thought via Step Entropy

  • 用步骤熵衡量每步推理信息量,识别冗余步骤
  • 在多个模型上验证,80%低熵步骤可删减而精度损失小
  • 训练模型自动生成压缩推理路径,适合高效部署场景

使用链式思维(CoT)提示的大语言模型在复杂推理任务中表现优异,但生成的思考过程冗长且重复,导致推理成本高、效率低。本文提出一种基于步骤熵的新型CoT压缩框架,该指标用于量化单个推理步骤的信息贡献度以识别冗余。理论分析与在数学推理基准上的大量实证验证表明,低熵步骤具有高度冗余性。实验发现,在DeepSeek-R1-7B、14B和Qwen3-8B模型上,高达80%的低熵中间步骤可被剪枝,而最终答案准确率仅轻微下降。这一结果显著区别于随机或高熵剪枝方式,后者会严重损害推理性能。在此基础上,本文提出一种结合监督微调(SFT)与组相对策略优化(GRPO)的两阶段训练策略,使模型在推理时能自主学习生成压缩后的CoT,通过战略性插入[SKIP]标记实现。该方法大幅提升了大模型推理效率并保持准确率,为更高效的模型部署提供了可能,并深化了对模型内部推理机制的理解。代码与数据已公开于https://github.com/staymylove/COT_Compresstion_via_Step_entropy。

原文摘要 · Abstract (English)

Large Language Models (LLMs) using Chain-of-Thought (CoT) prompting excel at complex reasoning but generate verbose thought processes with considerable redundancy, leading to increased inference costs and reduced efficiency. We introduce a novel CoT compression framework based on step entropy, a metric that quantifies \emph{the informational contribution of individual reasoning steps} to identify redundancy. Through theoretical analysis and extensive empirical validation on mathematical reasoning benchmarks, we demonstrate that steps with low entropy are indeed highly redundant. Our experiments reveal that an astonishing 80\% of low-entropy intermediate steps can be pruned with minor degradation in the final answer accuracy across DeepSeek-R1-7B, 14B and Qwen3-8B. This finding sharply contrasts with random or high-entropy pruning, which severely impairs reasoning performance. Building on this, we propose a novel two-stage training strategy combining Supervised Fine-Tuning (SFT) and Group Relative Policy Optimization (GRPO) reinforcement learning. This approach enables LLMs to autonomously learn to generate compressed COTs during inference by strategically incorporating [SKIP] tokens. Our method significantly improves LLM inference efficiency while preserving accuracy, paving the way for more scalable LLM deployments and a better understanding of their internal reasoning. The code and data are released in https://github.com/staymylove/COT_Compresstion_via_Step_entropy.

大模型推理链式思维压缩技术步骤熵

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