Title
Entropy-based pruning method for convolutional neural networks
Abstract
Various compression approaches including pruning techniques have been developed to lighten the computational complexity of neural networks. Most pruning techniques determine the threshold of pruning weights or input features based on statistical analysis of the value of weights after completing their training. Their compression performance is limited because they do not take into account the contribution of weights to output during training. To solve this problem, we propose an entropy-based pruning technique that determines the threshold by considering the average amount of information from the weights to output while training. In the experiment section, we demonstrate and analyze our method for a convolutional neural network image classifier modeled by using Mixed National Institute of Standards and Technology image data. From the experimental results, our technique shows that compression performance has improved by more than 28% overall, compared to the well-known pruning technique. Also, the pruning speed has improved by 14%.
Year
DOI
Venue
2019
10.1007/s11227-018-2684-z
The Journal of Supercomputing
Keywords
Field
DocType
Convolutional neural network, Gaussian, Entropy, Pruning, Threshold, Weight
Image classifier,Pattern recognition,Convolutional neural network,Computer science,Parallel computing,Gaussian,Artificial intelligence,Artificial neural network,Pruning,Statistical analysis,Computational complexity theory
Journal
Volume
Issue
ISSN
75
6
1573-0484
Citations 
PageRank 
References 
0
0.34
15
Authors
2
Name
Order
Citations
PageRank
Cheonghwan Hur111.71
Sanggil Kang214327.14