arXiv:2506.23675cs.CV2025-06ICCV被引 3

针对视觉Transformer在迁移学习中剪枝失效问题,提出按块贡献率分配资源的新方法。

Pruning by Block Benefit: Exploring the Properties of Vision Transformer Blocks during Domain Adaptation

  • 基于块级贡献度动态分配参数资源,避免早期误判重要性
  • 在70%参数剪枝下仅损失0.64%准确率,性能稳定
  • 适合资源受限场景下的模型轻量化部署

视觉Transformer在多个任务中创下新纪录,但其高计算成本使其难以在资源受限硬件上应用。网络剪枝通过移除不重要操作降低复杂度,但在未见数据域上剪枝会误判权重重要性,导致资源分配不佳。本文发现任务敏感层在下游任务中初期无法提升特征表示,影响早期剪枝决策。为此提出剪枝按块收益(P3B)方法,利用块级别相对贡献全局分配参数资源,识别低影响组件减少分配,保留关键部分。传统剪枝掩码优化难以重新激活被掩去的元素,而P3B基于全局性能指标设定逐层保留率,确保后期收敛块可恢复。大量实验表明,P3B是当前最优剪枝方法,在迁移学习任务中表现突出。尤其在70%参数剪枝率下,准确率仅下降0.64%。

原文摘要 · Abstract (English)

Vision Transformer have set new benchmarks in several tasks, but these models come with the lack of high computational costs which makes them impractical for resource limited hardware. Network pruning reduces the computational complexity by removing less important operations while maintaining performance. However, pruning a model on an unseen data domain, leads to a misevaluation of weight significance, resulting in suboptimal resource assignment. In this work, we find that task-sensitive layers initially fail to improve the feature representation on downstream tasks, leading to performance loss for early pruning decisions. To address this problem, we introduce Pruning by Block Benefit (P3B), a pruning method that utilizes the relative contribution on block level to globally assign parameter resources. P3B identifies low-impact components to reduce parameter allocation while preserving critical ones. Classical pruning mask optimization struggles to reactivate zero-mask-elements. In contrast, P3B sets a layerwise keep ratio based on global performance metrics, ensuring the reactivation of late-converging blocks. We show in extensive experiments that P3B is a state of the art pruning method with most noticeable gains in transfer learning tasks. Notably, P3B is able to conserve high performance, even in high sparsity regimes of 70% parameter reduction while only losing 0.64% in accuracy.

视觉Transformer模型剪枝迁移学习

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