Litcius/Paper detail

Rethinking the Pruning Criteria for Convolutional Neural Network

Zhongzhan Huang, Xinjiang Wang, Ping Luo

2021Neural Information Processing Systems27 citations

Abstract

Channel pruning is a popular technique for compressing convolutional neural networks (CNNs), and various pruning criteria have been proposed to remove the redundant filters of CNNs. From our comprehensive experiments, we find some blind spots on pruning criteria: (1) Similarity: There are some strong similarities among several primary pruning criteria that are widely cited and compared. According to these criteria, the ranks of filters’ importance in a convolutional layer are almost the same, resulting in similar pruned structures. (2) Applicability: For a large network (each convolutional layer has a large number of filters), some criteria can not distinguish the network redundancy well from their measured filters' importance. In this paper, we theoretically validate these two findings with our assumption that the well-trained convolutional filters in each layer approximately follow a Gaussian-alike distribution. This assumption is verified through systematic and extensive statistical tests.

Topics & Concepts

Convolutional neural networkRedundancy (engineering)PruningComputer sciencePattern recognition (psychology)Artificial intelligenceGaussianLayer (electronics)AlgorithmMachine learningOrganic chemistryPhysicsBiologyAgronomyChemistryOperating systemQuantum mechanicsAdversarial Robustness in Machine LearningAdvanced Neural Network ApplicationsAnomaly Detection Techniques and Applications