arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2407.13331cs.LG
收藏

无需重新训练即可重建剪枝模型

Reconstruct the Pruned Model without Any Retraining

Shanghai Jiao Tong University(上海交通大学);Shanghai Artificial Intelligence Laboratory(上海人工智能实验室)

Pingjie Wang, Ziqing Fan, Shengchao Hu, Zhe Chen, Yanfeng Wang, Yu Wang

AI总结:

提出LIAR框架,通过线性插值重建剪枝后权重,无需再训练,兼容多种剪枝标准,使BERT在剪除50%参数后保持98%准确率,LLaMA在几分钟内达到顶尖性能。

摘要:

结构化剪枝是一种对大型语言模型(LLM)友好的硬件压缩技术,预期无需重新训练以避免巨大的再训练成本。这种免再训练范式涉及(1)定义架构的剪枝标准和(2)恢复性能的失真重建。然而,现有方法往往强调剪枝标准,同时使用特定于某些模块或标准的重建技术,导致泛化能力有限。为解决此问题,我们引入了基于线性插值的自适应重建(LIAR)框架,该框架既高效又有效。LIAR不需要反向传播或再训练,并且兼容各种剪枝标准和模块。通过对保留的权重应用线性插值,LIAR最小化重建误差并有效重建剪枝后的输出。我们在GLUE、SQuAD、WikiText和常识推理等基准上的评估表明,LIAR使BERT模型在移除50%的参数后仍能保持98%的准确率,并在几分钟内为LLaMA实现顶级性能。

英文摘要:

Structured pruning is a promising hardware-friendly compression technique for large language models (LLMs), which is expected to be retraining-free to avoid the enormous retraining cost. This retraining-free paradigm involves (1) pruning criteria to define the architecture and (2) distortion reconstruction to restore performance. However, existing methods often emphasize pruning criteria while using reconstruction techniques that are specific to certain modules or criteria, resulting in limited generalizability. To address this, we introduce the Linear Interpolation-based Adaptive Reconstruction (LIAR) framework, which is both efficient and effective. LIAR does not require back-propagation or retraining and is compatible with various pruning criteria and modules. By applying linear interpolation to the preserved weights, LIAR minimizes reconstruction error and effectively reconstructs the pruned output. Our evaluations on benchmarks such as GLUE, SQuAD, WikiText, and common sense reasoning show that LIAR enables a BERT model to maintain 98% accuracy even after removing 50% of its parameters and achieves top performance for LLaMA in just a few minutes.

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

↑