Back to Search Start Over

Providing clear pruning threshold: A novel CNN pruning method via L 0 regularisation

Authors :
Gang Xu
Guo Li
Source :
IET Image Processing, Vol 15, Iss 2, Pp 405-418 (2021)
Publication Year :
2020
Publisher :
Institution of Engineering and Technology (IET), 2020.

Abstract

Network pruning is a significant way to improve the practicability of convolution neural networks (CNNs) by removing the redundant structure of the network model. However, in most of the existing network pruning methods l1 or l2 regularisation is applied to parameter matrices and the manual selection of pruning threshold is difficult and labor‐intensive. A novel CNNs network pruning method via l0 regularisation is proposed, which adopts l0 regularisation to expand the saliency gap between neurons. A half‐quadratic splitting (HQS) based iterative algorithm is put forward to calculate the approximation solution of l0 regularisation, which makes the joint optimisation problem of regularisation term and training loss function can be solved by various gradient‐based algorithms. Meanwhile, a hyperparameters selection method is designed to make most of the hyperparameters in the algorithm can be determined by examining the pre‐trained model. The results of experiments on MNIST, Fashion‐MNIST and CIFAR100 show that the proposed method can provide a much clearer pruning threshold by widening the saliency gap, and achieve a similar or even better compression performance, compared with the state‐of‐the‐art studies.

Details

ISSN :
17519667 and 17519659
Volume :
15
Database :
OpenAIRE
Journal :
IET Image Processing
Accession number :
edsair.doi.dedup.....7eb1a75bb2a6dda708acd1f316517dfb