Batch-Normalization-based Soft Filter Pruning for Deep Convolutional Neural Networks

Xiaozhou Xu,Qiming Chen,Lei Xie,Hongye Su
DOI: https://doi.org/10.1109/icarcv50220.2020.9305319
2020-01-01
Abstract:As convolutional neural network contains many redundant parameters, a lot of methods have been developed to compress the network for accelerating inference. Among these, network pruning, which is a kind of widely used approaches, can effectively decrease the memory capacity and reduce the computation cost. Herein, we propose a competitive pruning approach based on Soft Filter Pruning (SFP) by taking account of the scaling factors y of Batch Normalization (BN) layers as the criterion of filter selection strategy. During the soft pruning procedure, in each epoch only y values of BN layers less than threshold are set to zero instead of setting the weights of selected filters in convolutional layers to zero. Compared to the existing approaches, the proposed method can obtain a highly increased accuracy on image recognition. Notably, on CIFAR-10, the proposed method reduces the same 40.8% FLOPs as SFP on ResNet-110 with even 0.87% top-1 accuracy improvement.
What problem does this paper attempt to address?