返回
CCPrune: Collaborative channel pruning for learning compact convolutional networks
DOI:10.1016/j.neucom.2021.04.063.png)
摘要
En 中文
Deep convolutional neural networks (CNNs) is difficult to deploy on resource-constrained devices due to its huge amount of computation. Channel pruning is an effective method to reduce the amount of computation and accelerate network inference. Most of channels pruning methods use statistics from a single structure (convolutional layer or batch normalization layer) of the sparse network to evaluate the importance of channels. The limitation of these methods is that it may often mistakenly delete the important channels. In view of this, we propose a novel method, namely Collaborative Channel Pruning (CCPrune), to evaluate the importance of channels, which combines the convolution layer weights and the BN layer scaling factors. The proposed method first introduces the regularization on the convolution layer weights and the BN layer scaling factors respectively. Then combine the weight of the convolutional layer and the scaling factor of the BN layer to evaluate the importance of the channel. Finally, it can delete the unimportant channels without reduces the performance of the model. The experimental results well demonstrate the effectiveness of our method. On CIFAR-10, it can reduce the FLOPs of VGG-19 by 85.50% while only slightly reducing the accuracy of the model, and it can reduce the FLOPs of Resnet-50 by 78.31% without reducing the accuracy of the model, respectively. (c) 2021 Elsevier B.V. All rights reserved.
Keyword:
Deep convolutional neural network
Model compression
Channel pruning
期刊
IF:
6.5
论文数:
2.5W
被引数:
6.5W
机构
引用论文
MAPD: An improved multi-attribute pedestrian detection in a crowdMAPD: 一种改进的人群多属性行人检测
NEUROCOMPUTING
IF6.5
Structural study of lanthanides(III) in aqueous nitrate and chloride solutions by EXAFS通过EXAFS对硝酸盐和氯化物水溶液中镧系元素 (III) 的结构研究

