提出新正则化方法,让网络剪枝更公平、更稳定。
Catalyst: a Novel Regularizer for Structured Pruning with Auxiliary Extension of Parameter Space
- 通过扩展参数空间引入辅助变量,设计新正则化项
- 剪枝后模型性能保持稳定,且对微小变化不敏感
- 适合追求高鲁棒性与公平性的模型压缩场景
结构化剪枝通过移除整个卷积核或通道来降低深度神经网络的规模和计算开销。传统正则化方法如L1或组Lasso及其变体会导致基于权重大小的剪枝偏差,使小幅度滤波器更易被剪除。此外,这些方法常导致剪枝决策边界附近几乎无余量,使得滤波器幅度的微小扰动即可改变剪枝结果。本文识别出剪枝操作保持模型性能的精确代数条件,并据此在扩展参数空间中通过辅助催化剂变量构建了一种新型正则化。所提的Catalyst正则化确保每个滤波器具有理论上零偏差的剪枝机会,且通过大幅拉开保留与剪除滤波器间的幅度边界,实现稳健的剪枝行为。理论性质自然带来实际效果:实验验证了Catalyst剪枝算法在多个数据集和模型上的优越性,其剪枝结果显著优于当前最优滤波器剪枝方法,并证实了预测的鲁棒性与公平性特征。
原文摘要 · Abstract (English)
Structured pruning aims to reduce the size and computational cost of deep neural networks by removing entire filters or channels. The traditional regularizers such as L1 or Group Lasso and its variants lead to magnitude-biased pruning decisions, such that the filters with small magnitudes are likely to be pruned. Also, they often entail pruning results with almost zero margin around pruning decision boundary, such that tiny perturbation in a filter magnitude can flip the pruning decision. In this paper, we identify the precise algebraic condition under which pruning operations preserve model performance, and use the condition to construct a novel regularizer defined in an extended parameter space via auxiliary catalyst variables. The proposed Catalyst regularization ensures fair pruning chance for each filters with theoretically provable zero bias to their magnitude and robust pruning behavior achieved by wide-margin bifurcation of magnitudes between the preserved and the pruned filters. The theoretical properties naturally lead to real-world effectiveness, as shown by empirical validations of Catalyst Pruning algorithm. Pruning results on various datasets and models are superior to state-of-the-art filter pruning methods, and at the same time confirm the predicted robust and fair pruning characteristics of Catalyst pruning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。