Back to Search
Start Over
Providing clear pruning threshold: A novel CNN pruning method via L 0 regularisation
- 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