arXiv:2505.07411cs.LGcs.CV2025-05中稿 · International Join…被引 2

提出高效剪枝流程,提速近10倍且不损失精度

ICE-Pruning: An Iterative Cost-Efficient Pruning Pipeline for Deep Neural Networks

  • 自动判断剪枝后何时微调,减少冗余计算
  • 冻结部分参数加速微调,提升训练效率
  • 自适应学习率调度与超参调优,适合大模型压缩

剪枝是压缩深度神经网络(DNN)的常用方法,通过移除不重要参数来减小模型规模。但剪枝会降低精度,通常需配合微调甚至权重重置等操作恢复性能。现有方法采用逐轮剪枝+微调的迭代策略,虽有效但计算开销大。本文提出ICE-Pruning,一种基于三个核心组件的迭代剪枝流程:i)自动决定剪枝后何时进行微调;ii)冻结策略以加快每轮微调速度;iii)定制化的剪枝感知学习率调度器,提升每轮精度并减少总耗时。此外,设计了高效的自动调参阶段,优化冻结比例等超参数。在多个DNN模型和数据集上评估,结果显示剪枝速度最高可提升9.61倍。代码已开源。

原文摘要 · Abstract (English)

Pruning is a widely used method for compressing Deep Neural Networks (DNNs), where less relevant parameters are removed from a DNN model to reduce its size. However, removing parameters reduces model accuracy, so pruning is typically combined with fine-tuning, and sometimes other operations such as rewinding weights, to recover accuracy. A common approach is to repeatedly prune and then fine-tune, with increasing amounts of model parameters being removed in each step. While straightforward to implement, pruning pipelines that follow this approach are computationally expensive due to the need for repeated fine-tuning. In this paper we propose ICE-Pruning, an iterative pruning pipeline for DNNs that significantly decreases the time required for pruning by reducing the overall cost of fine-tuning, while maintaining a similar accuracy to existing pruning pipelines. ICE-Pruning is based on three main components: i) an automatic mechanism to determine after which pruning steps fine-tuning should be performed; ii) a freezing strategy for faster fine-tuning in each pruning step; and iii) a custom pruning-aware learning rate scheduler to further improve the accuracy of each pruning step and reduce the overall time consumption. We also propose an efficient auto-tuning stage for the hyperparameters (e.g., freezing percentage) introduced by the three components. We evaluate ICE-Pruning on several DNN models and datasets, showing that it can accelerate pruning by up to 9.61x. Code is available at https://github.com/gicLAB/ICE-Pruning

模型压缩剪枝效率优化

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