arXiv:2411.00969stat.MLcs.LG2024-11被引 2

用混合高斯先验提升大模型剪枝效果,高压缩下仍保持性能

Magnitude Pruning of Large Pretrained Transformer Models with a Mixture Gaussian Prior

  • 引入混合高斯先验指导权重剪枝,保留模型表达能力
  • 在高压缩率下优于现有方法,跨多项NLP任务表现更优
  • 为稀疏Transformer的稳定性提供理论支持,适合部署优化

大型预训练Transformer模型在自然语言处理中取得了顶尖性能,但其庞大的参数量给实际部署带来挑战。为此,研究者常基于权重大小或敏感性剪枝以缩小模型。然而,以往研究表明,单纯基于大小的剪枝在现代NLP任务的迁移学习中存在局限。本文提出一种新的基于大小的剪枝算法——混合高斯先验剪枝(MGPP),利用混合高斯先验进行正则化,引导剪除非表达性权重,旨在保留模型表达能力。在自然语言理解、问答和生成等多类NLP任务上的广泛评估表明,MGPP在高压缩率下显著优于现有剪枝方法。此外,本文还提供了稀疏Transformer一致性的理论依据,揭示了该剪枝方法的有效性。

原文摘要 · Abstract (English)

Large pretrained transformer models have revolutionized modern AI applications with their state-of-the-art performance in natural language processing (NLP). However, their substantial parameter count poses challenges for real-world deployment. To address this, researchers often reduce model size by pruning parameters based on their magnitude or sensitivity. Previous research has demonstrated the limitations of magnitude pruning, especially in the context of transfer learning for modern NLP tasks. In this paper, we introduce a new magnitude-based pruning algorithm called mixture Gaussian prior pruning (MGPP), which employs a mixture Gaussian prior for regularization. MGPP prunes non-expressive weights under the guidance of the mixture Gaussian prior, aiming to retain the model's expressive capability. Extensive evaluations across various NLP tasks, including natural language understanding, question answering, and natural language generation, demonstrate the superiority of MGPP over existing pruning methods, particularly in high sparsity settings. Additionally, we provide a theoretical justification for the consistency of the sparse transformer, shedding light on the effectiveness of the proposed pruning method.

模型剪枝Transformer压缩高斯先验

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