Sensitiveness Based Strategy for Network Compression

Yiheng Lu,Kaiyuan Feng,Hao Li,Yue Wu,Maoguo Gong,Yaoting Xu
DOI: https://doi.org/10.1109/IAI50351.2020.9262195
2020-01-01
Abstract:The success of convolutional neural networks has contributed great improvement on various tasks. Typically, larger and deeper networks are designed to increase performance on classification works, such as AlexNet, VGGNets, GoogleNet and ResNets, which are composed by enormous convolutional filters. However, these complicated structures constrain models to be deployed into real application due to limited computational resources. In this paper, we propose a simple method to sort the importance of each convolutional layer, as well as compressing network by removing redundant filters. There are two implications in our work: 1) Downsizing the width of unimportant layers will lead to better performance on classification tasks. 2) The random-pruning in each convolutional layer can present similar results as weight-searching algorithm, which means that structure of the network dominates its ability of representation. As a result, we reveal the property of different convolutional layers for VGG-16, customized VGG-4 and ResNet-18 on different datasets. Consequently, the importance of different convolutional layers is described as sensitiveness to compress these networks greatly.
What problem does this paper attempt to address?