arXiv:2508.15828cs.LGcs.CL2025-08中稿 · AICCSA 2025被引 2

无需重训练,高效剪枝大模型提升推理效率

Z-Pruner: Post-Training Pruning of Large Language Models for Efficiency without Retraining

  • 结合权重更新幅度与激活模式识别冗余参数
  • 在多个主流模型上实现更低困惑度和更高零样本准确率
  • 方法通用、易实现,适合部署优化场景

近年来大型语言模型(LLMs)在自然语言处理任务中取得了显著进展,但模型规模不断增大,给部署、扩展性和能效带来挑战。为解决这一问题,后训练剪枝成为一种无需重训练即可缩减模型规模和推理延迟的有前景方法。然而,现有方法常导致性能显著下降或需计算成本高昂的微调。本文提出 Z-Pruner,一种新型后训练剪枝方法,可在不重训练的情况下对预训练 LLMs 进行稀疏化。不同于传统方法,Z-Pruner 同时利用权重更新幅度和激活模式,更有效地识别并移除冗余参数。该方法具备模型无关性、高效性和易实现性。我们在 LLaMA-2、LLaMA-3、OPT 等多种主流架构上进行了评估,涵盖多个标准语言基准。实验结果表明,Z-Pruner 在零样本准确率和困惑度方面均优于需大量权重更新的先进剪枝方法。代码已开源:https://github.com/sazzadadib/Z-Pruner。

原文摘要 · Abstract (English)

Large language models (LLMs) have rapidly advanced in recent years, achieving remarkable performance across a wide range of natural language processing tasks. However, this progress has come at the cost of increasingly large model sizes, which pose significant challenges for deployment, scalability, and energy efficiency. To address these limitations, post-training pruning has emerged as a promising approach for reducing model size and inference latency without the need for retraining. Despite these advantages, many existing pruning methods result in substantial performance degradation or require computationally expensive fine-tuning. In this work, we introduce Z-Pruner, a novel post-training pruning method designed to induce sparsity in pretrained LLMs without any retraining. Unlike conventional approaches, Z-Pruner leverages both weight update magnitudes and activation patterns to identify and eliminate redundant parameters more effectively. Our method is model-agnostic, efficient, and easy to implement. We evaluate Z-Pruner using multiple widely-used LLM architectures, including LLaMA-2, LLaMA-3, and OPT, across a diverse set of standard language benchmarks. Experimental results demonstrate that Z-Pruner surpasses state-of-the-art pruning methods that require intensive weight updates. Specifically, Z-Pruner achieves the lowest perplexity scores and the highest overall average score for zero-shot accuracy. We have made the corresponding codes publicly available at https://github.com/sazzadadib/Z-Pruner.

模型剪枝大模型优化推理效率无重训练

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