arXiv:2603.08083cs.CL2026-03

用输出分布熵提升大模型剪枝精度,不需额外教师模型。

High-Fidelity Pruning for Large Language Models

  • 以输出分布熵替代交叉熵,更全面评估神经元重要性。
  • 在LLaMA和Qwen系列模型上,剪枝后性能优于现有方法。
  • 无需额外训练,计算开销小,适合实际部署场景。

大语言模型在各类任务中表现优异,但其巨大的计算与内存需求给部署带来挑战。传统基于损失函数泰勒展开的剪枝方法依赖独热交叉熵,仅关注预测下一个词的概率,忽略模型其他潜在预测,评估片面。虽可用自蒸馏缓解,但需额外教师模型,增加计算负担。为此,我们提出一种简单有效的方法:使用模型输出分布的信息熵作为重要性评估标准,无需额外教师模型即可实现高效剪枝。相比传统交叉熵,该方法能全局评估神经元对模型预测的影响,显著提升剪枝后模型的保真度。在多个零样本基准测试中,本方法在LLaMA和Qwen系列模型上均持续优于现有剪枝方法。代码与权重已开源。

原文摘要 · Abstract (English)

Large Language Models (LLMs) have demonstrated exceptional performance across a wide range of tasks, yet their significant computational and memory requirements present major challenges for deployment. A common approach uses Taylor expansion on the loss function to estimate neuron importance. However, its reliance on one-hot cross entropy loss, a key limitation is that it narrowly assesses importance based only on the probability assigned to the single predicted next token, thereby ignoring the other potential predictions of the original model. An intuitive solution to address this is to employ self distillation criterion for importance evaluation. However, this approach introduces significant computational overhead by requiring a separate teacher model for supervision. To this end, we propose a simple but effective criterion, information entropy of the model's output distribution, to efficiently evaluate importance scores of neurons with Taylor pruning without requirement of additional teacher. Compared to plain cross entropy criterion, it provides a more holistic criterion for Taylor pruning to prune neurons with the least impact on the prediction of model in a global manner, thereby preserving the fidelity of the model's predictive capabilities. Experimental results on extensive zero-shot benchmarks demonstrate that our method consistently outperforms existing pruning methods across the LLaMA and Qwen series models. The source code and trained weights are availabel at https://github.com/visresearch/HFPrune.

大模型剪枝信息熵高效推理模型压缩

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