Loss Constrains Added Squeeze and Excitation Blocks for Pruning Deep Neural Networks

Yiqin Wang,Ming Li,Hongye Su,Lei Xie,Xiaochen Li,Weihua Xu,Xiaozhou Xu
DOI: https://doi.org/10.1109/icarcv50220.2020.9305335
2020-01-01
Abstract:Deep neural networks are proved to be very effective to solve problems on image classification, object detection and segmentation. However, in cases where only limited hardware is acquired, it may be a problem to deploy big models with excellent performance as they are sometimes calculation consuming. To overcome the limits on power, memory and calculation, channel pruning is proposed to compress the model in channel wise and soon become a common approach to have big models compressed. Generally, pruning is a three-stage pipeline containing training, pruning and finetuning. In this work, we come up with a new pruning approach that needs no finetuning. The major idea is extracting channel saliences by squeeze and excitation block and pushing the salience to either 0 or 1 by a sin-based function. Then take the salience as criteria for pruning. As the criteria of our approach is activation rather than trainable parameter, finetuning is not necessary in our pruning strategy which make the pruning process more stable and time saving. Experiment on flowers demonstrates our new designed pruning method is effective on reducing the model scale while maintaining the overall accuracy.
What problem does this paper attempt to address?