arXiv:2602.04919cs.LG2026-02

通过渐进式压缩让大模型推理能力不降反升

Gradually Compacting Large Language Models for Reasoning Like a Boiling Frog

  • 分步迭代剪枝+微调,像煮青蛙般逐步瘦身
  • 压缩至原大小一半,推理性能几乎不变
  • 适配多种剪枝/微调方法,通用性强

大语言模型虽具备强大推理能力,但体积庞大导致计算开销高。传统剪枝直接删参数常引发性能骤降,且需大量后续训练恢复。本文提出渐进压缩方法(PTL),将压缩过程拆分为多个细粒度阶段,在每阶段执行剪枝-微调循环,使模型在逐步瘦身过程中持续恢复性能,避免突然失效。实验表明,该方法可将模型压缩至原始大小的约50%,仅需轻量级后训练,推理表现与原模型相当。PTL对神经元剪枝、层剪枝等多种策略及持续预训练、强化学习等微调方式均有效,且在数学推理之外的代码生成等任务中也表现良好,具有广泛适用性。

原文摘要 · Abstract (English)

Large Language Models (LLMs) have demonstrated impressive reasoning capabilities, but their substantial size often demands significant computational resources. To reduce resource consumption and accelerate inference, it is essential to eliminate redundant parameters without compromising performance. However, conventional pruning methods that directly remove such parameters often lead to a dramatic drop in model performance in reasoning tasks, and require extensive post-training to recover the lost capabilities. In this work, we propose a gradual compacting method that divides the compression process into multiple fine-grained iterations, applying a Prune-Tune Loop (PTL) at each stage to incrementally reduce model size while restoring performance with finetuning. This iterative approach-reminiscent of the "boiling frog" effect-enables the model to be progressively compressed without abrupt performance loss. Experimental results show that PTL can compress LLMs to nearly half their original size with only lightweight post-training, while maintaining performance comparable to the original model on reasoning tasks. Moreover, PTL is flexible and can be applied to various pruning strategies, such as neuron pruning and layer pruning, as well as different post-training methods, including continual pre-training and reinforcement learning. Additionally, experimental results confirm the effectiveness of PTL on a variety of tasks beyond mathematical reasoning, such as code generation, demonstrating its broad applicability.

模型压缩推理增强渐进剪枝

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