arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.07375cs.CLcs.LG

大型语言模型高效后训练剪枝与统计修正

Efficient Post-Training Pruning of Large Language Models with Statistical Correction

Department of Computer Science and Engineering, Santa Clara University, Santa Clara, CA, USA(计算机科学与工程系,圣克拉拉大学);Futurewei Technologies, Inc., USA(未来韦艾技术公司)

Peiqi Yu, Jinhao Wang, Xinyi Sui, Nam Ling, Wei Wang, Wei Jiang

AI总结:

本文提出了一种基于模型权重和激活统计性质的轻量级后训练剪枝方法,通过统计修正提升剪枝性能,同时保持计算效率。

摘要:

后训练剪枝是一种有效的方法,用于减少大型语言模型(LLMs)的大小和推理成本,但现有方法常常面临剪枝质量与计算效率之间的权衡。启发式剪枝方法效率高但易受激活异常值影响,而基于重建的方法虽然能提高保真度,但计算成本高。本文提出了一种基于模型权重和激活的首次-order统计性质的轻量级后训练剪枝框架。在剪枝过程中,利用通道统计信息校准基于幅度的重要性评分,以减少由激活主导通道带来的偏差。剪枝后,应用解析能量补偿来纠正由于权重移除引起的分布扭曲。这两个步骤均不涉及重训练、梯度或二次信息。在多个LLM家族、稀疏模式和评估任务上的实验表明,所提出的方法在提高剪枝性能的同时,保持了与启发式方法相当的计算成本。结果表明,简单的统计修正对LLM的后训练剪枝是有效的。

英文摘要:

Post-training pruning is an effective approach for reducing the size and inference cost of large language models (LLMs), but existing methods often face a trade-off between pruning quality and computational efficiency. Heuristic pruning methods are efficient but sensitive to activation outliers, while reconstruction-based approaches improve fidelity at the cost of heavy computation. In this work, we propose a lightweight post-training pruning framework based on first-order statistical properties of model weights and activations. During pruning, channel-wise statistics are used to calibrate magnitude-based importance scores, reducing bias from activation-dominated channels. After pruning, we apply an analytic energy compensation to correct distributional distortions caused by weight removal. Both steps operate without retraining, gradients, or second-order information. Experiments across multiple LLM families, sparsity patterns, and evaluation tasks show that the proposed approach improves pruning performance while maintaining computational cost comparable to heuristic methods. The results suggest that simple statistical corrections can be effective for post-training pruning of LLMs.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑