International Journal of applied mathematics and computer science

online read us now

Paper details

Number 4 - December 2023
Volume 33 - 2023

Choice of the p-norm for high level classification features pruning in modern convolutional neural networks with local sensitivity analysis

Ernest Jeczmionek, Piotr A. Kowalski

Abstract
Transfer learning has surfaced as a compelling technique in machine learning, enabling the transfer of knowledge across networks. This study evaluates the efficacy of ImageNet pretrained state-of-the-art networks, including DenseNet, ResNet, and VGG, in implementing transfer learning for prepruned models on compact datasets, such as FashionMNIST, CIFAR10, and CIFAR100. The primary objective is to reduce the number of neurons while preserving high-level features. To this end, local sensitivity analysis is employed alongside p-norms and various reduction levels. This investigation discovers that VGG16, a network rich in parameters, displays resilience to high-level feature pruning. Conversely, the ResNet architectures reveal an interesting pattern of increased volatility. These observations assist in identifying an optimal combination of the norm and the reduction level for each network architecture, thus offering valuable directions for model-specific optimization. This study marks a significant advance in understanding and implementing effective pruning strategies across diverse network architectures, paving the way for future research and applications.

Keywords
convolutional neural network, pruning, sensitivity analysis, transfer learning, ImageNet

DOI
10.34768/amcs-2023-0047