Jeczmionek, Ernest ; Kowalski, Piotr A.
Contributor:Korbicz, Józef (1951- ) - red. ; Uciński, Dariusz - red.
Title: Group publication title: Subject and Keywords:convolutional neural network (CNN) ; pruning ; sensitivity analysis ; transfer learning ; ImageNet
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.
Publisher:Zielona Góra: Uniwersytet Zielonogórski
Date: Resource Type: DOI: Pages: Source:AMCS, volume 33, number 4 (2023) ; click here to follow the link
Language: License CC BY 4.0: Rights: