Pruning Policy for Image Classification Problems Based on Deep Learning

doi:10.3390/informatics11030067

Return

Pruning Policy for Image Classification Problems Based on Deep Learning

2024-09-12

César G. Pachón

Javier O. Pinzón-Arenas

Dora M. Ballesteros *

DOI：10.3390/informatics11030067

Original

Original request for help

Save

Abstract

En 中文

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.

Keywords:

convolutional neural network

compression model

pruning policy

global pruning rate

pruning distribution

pruning method