arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2502.03460cs.LGcs.AIcs.CL
收藏

Adapt-Pruner:面向高效小语言模型训练的自适应结构化剪枝

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

University of Illinois Urbana-Champaign(伊利诺伊大学厄巴纳-香槟分校);HKUST(香港科技大学)

Rui Pan, Shivanshu Shekhar, Boyao Wang, Shizhe Diao, Jipeng Zhang, Xingyuan Pan, Renjie Pi, Tong Zhang

AI总结:

针对小语言模型(SLM)训练的计算成本或性能下降问题,提出分层自适应剪枝方法Adapt-Pruner,结合增量剪枝与训练,在常识基准上准确率超传统方法1%-7%,还能恢复小模型性能并发现更优1B模型。

摘要:

小语言模型(SLM)因其在边缘设备中的广泛应用,已引起学术界和工业界的极大关注。为获得性能强劲的SLM,传统方法要么从头预训练模型(会产生高昂计算成本),要么压缩/剪枝现有大语言模型(LLM,会导致性能下降且效果不及预训练)。本文研究结合结构化剪枝与模型训练的加速方法家族,发现:1)分层自适应剪枝(Adapt-Pruner)在LLM中极其有效,相比现有剪枝技术有显著提升;2)自适应剪枝结合后续训练得到的模型,性能可与从头预训练的模型媲美;3)增量剪枝通过交替剪枝与训练(每次仅移除约5%的神经元),带来了重要性能增益。在LLaMA-3.1-8B上的实验结果表明,Adapt-Pruner在常识基准上的准确率平均超过LLM-Pruner、FLAP、SliceGPT等传统剪枝方法1%-7%。此外,Adapt-Pruner通过从更大模型剪枝,仅用200倍更少的token就将MobileLLM-125M在MMLU基准上的性能恢复到600M水平,并发现了一个在多个基准上超过LLaMA-3.2-1B的新1B模型。官方代码已发布于https://github.com/research4pan/AdaptPruner。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑