arXiv:2502.03460cs.LGcs.AI2025-02被引 10

自适应结构剪枝让小模型训练更高效,性能接近从头预训练。

Adapt-Pruner: Adaptive Structural Pruning for Efficient Small Language Model Training

  • 按层动态剪枝,每次只删约5%神经元,边训边剪提升效率。
  • 在多个基准上比传统剪枝方法高1%-7%,恢复小模型性能至600M级。
  • 适合资源受限场景,尤其适合想快速优化小模型的研究者。

小型语言模型(SLMs)因在边缘设备上的广泛应用而备受关注。传统方法要么从头预训练,成本高昂;要么压缩大模型,性能下降。本文提出自适应结构剪枝(Adapt-Pruner),发现按层动态剪枝在大模型中极为有效,结合训练后性能可媲美从头预训练。通过分步剪枝(每次仅移除约5%神经元),实现显著性能提升。在LLaMA-3.1-8B上,相比LLM-Pruner、FLAP和SliceGPT等方法,平均准确率提高1%-7%。该方法将MobileLLM-125M的性能恢复至600M水平,仅需200倍更少的训练数据;并发现一新1B模型,在多个基准上超越LLaMA-3.2-1B。代码已开源。

原文摘要 · Abstract (English)

Small language models (SLMs) have attracted considerable attention from both academia and industry due to their broad range of applications in edge devices. To obtain SLMs with strong performance, conventional approaches either pre-train the models from scratch, which incurs substantial computational costs, or compress/prune existing large language models (LLMs), which results in performance drops and falls short in comparison to pre-training. In this paper, we investigate the family of acceleration methods that involve both structured pruning and model training. We found 1) layer-wise adaptive pruning (Adapt-Pruner) is extremely effective in LLMs and yields significant improvements over existing pruning techniques, 2) adaptive pruning equipped with further training leads to models comparable to those pre-training from scratch, 3) incremental pruning brings non-trivial performance gain by interleaving pruning with training and only removing a small portion of neurons ($\sim$5%) at a time. Experimental results on LLaMA-3.1-8B demonstrate that Adapt-Pruner outperforms conventional pruning methods, such as LLM-Pruner, FLAP, and SliceGPT, by an average of 1%-7% in accuracy on commonsense benchmarks. Additionally, Adapt-Pruner restores the performance of MobileLLM-125M to 600M on the MMLU benchmark with 200$\times$ fewer tokens via pruning from its larger counterparts, and discovers a new 1B model that surpasses LLaMA-3.2-1B in multiple benchmarks. The official code is released at https://github.com/research4pan/AdaptPruner.

模型剪枝小模型高效训练

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