让大模型自己设计剪枝方案,解决高压缩下的性能下降问题。
Automatic Pruning Discovery for Large Language Models
- 用大模型自动生成剪枝算法,无需人工设计
- 提出动态稀疏分配机制,缓解异常值导致的性能下降
- 在主流模型上实现比现有方法更优的压缩效果
大型语言模型(LLMs)在众多任务中表现优异,但其巨大规模阻碍了实际部署。现有的剪枝方法(如 Wanda)依赖人工设计,成本高且需专业知识。本文首次揭示了高剪枝率下性能急剧下降的根本原因——均匀稀疏性引发的异常值问题,并提出自主剪枝方法 AutoPrune。该方法利用大模型自身生成最优剪枝算法,突破专家知识限制;通过图驱动的思维链(GCoT)优化提示,提升推理能力与可解释性;并引入偏斜感知动态稀疏分配(SDSA)机制,有效缓解异常值影响。在主流 LLM 基准测试中,AutoPrune 持续优于现有先进方法。
原文摘要 · Abstract (English)
Large language models (LLMs) have achieved remarkable performance on a wide range of tasks, hindering real-world deployment due to their massive size. Existing pruning methods (e.g., Wanda) tailored for LLMs rely heavily on manual design pruning algorithms, thereby leading to huge labor costs and requires expert knowledge. Furthermore, we are the first to identify the serious outlier value issue behind dramatic performance degradation under high pruning ratios that are caused by uniform sparsity, raising an additional concern about how to design adaptive pruning sparsity ideal for LLMs. Can LLMs prune by themselves? In this work, we introduce an affirmative answer by proposing a novel pruning method called AutoPrune, which first overcomes expert knowledge limits by leveraging LLMs to design optimal pruning algorithm for themselves automatically without any expert knowledge. Specifically, to mitigate the black-box nature of LLMs, we propose a Graph-driven Chain-of-Thought (GCoT) to optimize prompts, significantly enhancing the reasoning process in learning the pruning algorithm and enabling us to generate pruning algorithms with superior performance and interpretability in the next generation. Finally, grounded in insights of outlier value issue, we introduce Skew-aware Dynamic Sparsity Allocation (SDSA) to overcome the outlier value issue, mitigating performance degradation under high pruning ratios. We conduct extensive experiments on mainstream LLMs benchmarks, demonstrating the superiority of AutoPrune, which consistently excels state-of-the-art competitors.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。