Fractional Step Discriminant Pruning: A Filter Pruning Framework for Deep Convolutional Neural Networks
Nikolaos Gkalelis, Vasileios Mezaris
-
SPS
IEEE Members: $11.00
Non-members: $15.00Length: 09:22
In this paper, a novel pruning framework is introduced to compress noisy or less discriminant filters in small fractional steps, in deep convolutional networks. The proposed framework utilizes a class-separability criterion that can exploit effectively the labeling information in annotated training sets. Additionally, an asymptotic schedule for the pruning rate and scaling factor is adopted so that the selected filters' weights collapse gradually to zero, providing improved robustness. Experimental results on the CIFAR-10, Google speech commands (GSC) and ImageNet32 (a downsampled version of ILSVRC-2012) show the efficacy of the proposed approach.