Joint Multi-Dimension Pruning

05/18/2020

∙

by   Zechun Liu, et al.

∙

6

∙

share

We present joint multi-dimension pruning (named as JointPruning), a new perspective of pruning a network on three crucial aspects: spatial, depth and channel simultaneously. The joint strategy enables to search a better status than previous studies that focused on individual dimension solely, as our method is optimized collaboratively across the three dimensions in a single end-to-end training. Moreover, each dimension that we consider can promote to get better performance through colluding with the other two. Our method is realized by the adapted stochastic gradient estimation. Extensive experiments on large-scale ImageNet dataset across a variety of network architectures MobileNet V1 V2 and ResNet demonstrate the effectiveness of our proposed method. For instance, we achieve significant margins of 2.5 improvement over the state-of-the-art approach on the already compact MobileNet V1 V2 under an extremely large compression ratio.