VACL: Variance-Aware Cross-Layer Regularization for Pruning Deep Residual Networks

09/10/2019
by   Shuang Gao, et al.
0

Improving weight sparsity is a common strategy for producing light-weight deep neural networks. However, pruning models with residual learning is more challenging. In this paper, we introduce Variance-Aware Cross-Layer (VACL), a novel approach to address this problem. VACL consists of two parts, a Cross-Layer grouping and a Variance Aware regularization. In Cross-Layer grouping the i^th filters of layers connected by skip-connections are grouped into one regularization group. Then, the Variance-Aware regularization term takes into account both the first and second-order statistics of the connected layers to constrain the variance within a group. Our approach can effectively improve the structural sparsity of residual models. For CIFAR10, the proposed method reduces a ResNet model by up to 79.5 and reduces a ResNeXt model by up to 82 ImageNet, it yields a pruned ratio of up to 63.3 accuracy drop. Our experimental results show that the proposed approach significantly outperforms other state-of-the-art methods in terms of overall model size and accuracy.

READ FULL TEXT

Please sign up or login with your details

Forgot password? Click here to reset