arXiv:2605.18331cs.LG2026-05

通过动态更新与逐层剪枝,实现大模型极致压缩且保持性能

Prune, Update and Trim: Robust Structured Pruning for Large Language Models

论文配图:Prune, Update and Trim: Robust Structured Pruning for Large Language Models
图 1 · 摘自论文原文
  • 先更新未剪枝参数,再按顺序剪枝,减少误差累积
  • 可实现极端稀疏度(如90%以上)剪枝仍保持模型精度
  • 支持注意力头级剪枝,适用于GQA等新型架构

大型语言模型(LLM)近年来发展迅速,但推理成本高昂,尤其在长上下文或资源受限设备上。为此,我们提出Putri,一种后训练结构化剪枝方法,相比现有技术有三项改进:首先,对未剪枝的前馈网络(FFN)权重进行更新以补偿剪枝带来的误差;其次,按顺序逐层剪枝,考虑先前层的更新影响;第三,不剪整个注意力层,而是剪除个别注意力头,并可扩展至分组查询注意力(GQA)。在多个模型、多种稀疏度范围和不同数据集上的实验验证了Putri的通用性。特别地,与以往方法不同,Putri可在极高的稀疏率下有效剪枝。代码已开源:https://github.com/Coello-dev/Putri。

原文摘要 · Abstract (English)

Large Language Models (LLMs) have experienced significant growth and development in recent years. However, performing inference on LLMs remains costly, especially for long-context inference or in resource-constrained devices. This motivates the development of new post-training pruning (PTP) methods. These methods reduce LLMs' requirements by removing a substantial part of the model's parameters. The discarded weights are selected depending on their impact on the models performance. Current PTP methods prune the models by removing the less informative hidden nodes from the FFN layers, and the least important attention layers. We propose Putri, a PTP method that introduces three changes to the State-of-the-art. First, we update the un-pruned weights of the FFN to compensate for the introduced pruning error. Second, the FFN layers are pruned sequentially, taking into account the updates done to the previous layers. Third, instead of removing full attention layers, we remove individual attention-heads. We extend this method such that it can also address Grouped-Query Attention. In summary, Putri is a structure pruning method which remains simple while showing SOTA performance. Pruning experiments on multiple models with a wide variety of sparsity ranges and on different datasets, validate the generality of Putri. Notably, we demonstrate that, unlike previous methods, Putri can prune LLMs on extreme sparsity ratios. The code is available at: https://github.com/Coello-dev/Putri.

模型剪枝大模型优化结构化剪枝注意力头

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