arXiv:2506.04513cs.CV2025-06被引 1

同时剪裁神经元和层,让模型更小更快还更准。

Pruning Everything, Everywhere, All at Once

  • 用相似性度量选最优剪枝路径,实现层与神经元协同剪裁。
  • 在ResNet56/110上分别达86.37%和95.82%的浮点运算削减。
  • 剪后模型抗干扰强,碳排放降低超83%,适合绿色部署。

深度学习在认知任务中占据主导地位,但模型日益复杂导致计算成本高昂,限制了其在资源受限场景的应用。现有剪枝方法多聚焦于移除神经元(如滤波器、注意力头)或层,但难以同时处理二者。为此,本文提出一种新方法:从层剪枝和神经元剪枝生成两个候选子网络后,基于中心核对齐(CKA)度量选择与原模型表示最相似的一个,迭代执行该过程以获得高度稀疏且保持原有预测能力的模型。在标准架构与基准测试上,该方法显著优于当前主流层剪枝与滤波器剪枝技术。在高精度压缩下,多数方法准确率大幅下降,而本方法反而提升或仅小幅下降。尤其在ResNet56和ResNet110上,分别实现86.37%和95.82%的浮点运算量减少。此外,剪枝后模型对对抗样本和分布外数据更具鲁棒性,推动绿色AI发展,最高可降低83.31%碳排放。整体上,本工作为剪枝开辟了新范式。

原文摘要 · Abstract (English)

Deep learning stands as the modern paradigm for solving cognitive tasks. However, as the problem complexity increases, models grow deeper and computationally prohibitive, hindering advancements in real-world and resource-constrained applications. Extensive studies reveal that pruning structures in these models efficiently reduces model complexity and improves computational efficiency. Successful strategies in this sphere include removing neurons (i.e., filters, heads) or layers, but not both together. Therefore, simultaneously pruning different structures remains an open problem. To fill this gap and leverage the benefits of eliminating neurons and layers at once, we propose a new method capable of pruning different structures within a model as follows. Given two candidate subnetworks (pruned models), one from layer pruning and the other from neuron pruning, our method decides which to choose by selecting the one with the highest representation similarity to its parent (the network that generates the subnetworks) using the Centered Kernel Alignment metric. Iteratively repeating this process provides highly sparse models that preserve the original predictive ability. Throughout extensive experiments on standard architectures and benchmarks, we confirm the effectiveness of our approach and show that it outperforms state-of-the-art layer and filter pruning techniques. At high levels of Floating Point Operations reduction, most state-of-the-art methods degrade accuracy, whereas our approach either improves it or experiences only a minimal drop. Notably, on the popular ResNet56 and ResNet110, we achieve a milestone of 86.37% and 95.82% FLOPs reduction. Besides, our pruned models obtain robustness to adversarial and out-of-distribution samples and take an important step towards GreenAI, reducing carbon emissions by up to 83.31%. Overall, we believe our work opens a new chapter in pruning.

模型剪枝高效推理绿色AI协同剪裁

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