Pruning Policy for Image Classification Problems Based on Deep Learning

Abstract

In recent years, several methods have emerged for compressing image classification models using CNNs, for example, by applying pruning to the convolutional layers of the network. Typically, each pruning method uses a type of pruning distribution that is not necessarily the most appropriate for a given classification problem. Therefore, this paper proposes a methodology to select the best pruning policy (method + pruning distribution) for a specific classification problem and global pruning rate to obtain the best performance of the compressed model. This methodology was applied to several image datasets to show the influence not only of the method but also of the pruning distribution on the quality of the pruned model. It was shown that the selected pruning policy affects the performance of the pruned model to different extents, and that it depends on the classification problem to be addressed. For example, while for the Date Fruit Dataset, variations of more than 10% were obtained, for CIFAR10, variations were less than 5% for the same cases evaluated.