arXiv:2501.15296cs.CL2025-01ICLR被引 19

无需校准数据,15分钟压缩大模型,性能损失小。

You Only Prune Once: Designing Calibration-Free Model Compression With Policy Learning

  • 将剪枝转化为策略学习,仅凭模型内部特性判断参数重要性。
  • 30%压缩率下保留80%零样本性能,优于现有方法的75%。
  • 适合需要快速部署且对性能敏感的大模型压缩场景。

大型语言模型(LLMs)日益增长的规模带来了巨大的计算与内存压力。当前剪枝方法依赖外部校准数据来决定剪枝参数,限制了其在不同压缩率下的灵活性与可扩展性,且在高压缩率下常导致下游任务性能严重下降。本文提出PruneNet,一种新型模型压缩方法,将剪枝重构为策略学习过程,解耦剪枝与模型结构,无需校准数据。它通过学习随机剪枝策略,仅依据模型内在属性评估参数重要性,并保持谱结构以最小化信息损失。PruneNet可在15分钟内压缩LLaMA-2-7B模型,在30%压缩率下实现超过80%的零样本性能保留,显著优于仅保留75%性能的现有方法。在复杂多任务语言理解任务中,其性能最高可保持原模型的80%,展现出对传统结构化压缩技术的优越性。

原文摘要 · Abstract (English)

The ever-increasing size of large language models (LLMs) presents significant challenges for deployment due to their heavy computational and memory requirements. Current model pruning techniques attempt to alleviate these issues by relying heavily on external calibration datasets to determine which parameters to prune or compress, thus limiting their flexibility and scalability across different compression ratios. Moreover, these methods often cause severe performance degradation, particularly in downstream tasks, when subjected to higher compression rates. In this paper, we propose PruneNet, a novel model compression method that addresses these limitations by reformulating model pruning as a policy learning process. PruneNet decouples the pruning process from the model architecture, eliminating the need for calibration datasets. It learns a stochastic pruning policy to assess parameter importance solely based on intrinsic model properties while preserving the spectral structure to minimize information loss. PruneNet can compress the LLaMA-2-7B model in just 15 minutes, achieving over 80% retention of its zero-shot performance with a 30% compression ratio, outperforming existing methods that retain only 75% performance. Furthermore, on complex multitask language understanding tasks, PruneNet demonstrates its robustness by preserving up to 80% performance of the original model, proving itself a superior alternative to conventional structured compression techniques.

模型压缩大模型剪枝策略学习

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