ThiNet: Pruning CNN Filters for a Thinner Net

• Published in: IEEE transactions on pattern analysis and machine intelligence Vol. 41; no. 10; pp. 2525 - 2538
• Main Authors: Luo, Jian-Hao, Zhang, Hao, Zhou, Hong-Yu, Xie, Chen-Wei, Wu, Jianxin, Lin, Weiyao
• Format: Journal Article
• Language: English
• Published: United States IEEE 01.10.2019; The Institute of Electrical and Electronics Engineers, Inc. (IEEE)
• Subjects: Acceleration; Artificial neural networks; Computational modeling; Convolution; Convolutional neural networks; deep learning; Electronic devices; filter pruning; Image coding; Model accuracy; model compression; Neural networks; Optimization; Pruning; Segmentation; Task analysis; Training
• DOI: 10.1109/TPAMI.2018.2858232
• ISSN: 0162-8828; 1939-3539

This paper aims at accelerating and compressing deep neural networks to deploy CNN models into small devices like mobile phones or embedded gadgets. We focus on filter level pruning, i.e., the whole filter will be discarded if it is less important. An effective and unified framework, ThiNet (stands for “Thin Net”), is proposed in this paper. We formally establish filter pruning as an optimization problem, and reveal that we need to prune filters based on statistics computed from its next layer, not the current layer, which differentiates ThiNet from existing methods. We also propose “gcos” (Group COnvolution with Shuffling), a more accurate group convolution scheme, to further reduce the pruned model size. Experimental results demonstrate the effectiveness of our method, which has advanced the state-of-the-art. Moreover, we show that the original VGG-16 model can be compressed into a very small model (ThiNet-Tiny) with only 2.66 MB model size, but still preserve AlexNet level accuracy. This small model is evaluated on several benchmarks with different vision tasks (e.g., classification, detection, segmentation), and shows excellent generalization ability.