A Unified DNN Weight Compression Framework Using Reweighted Optimization Methods
arXiv:2004.05531
Abstract
To address the large model size and intensive computation requirement of deep neural networks (DNNs), weight pruning techniques have been proposed and generally fall into two categories, i.e., static regularization-based pruning and dynamic regularization-based pruning. However, the former method currently suffers either complex workloads or accuracy degradation, while the latter one takes a long time to tune the parameters to achieve the desired pruning rate without accuracy loss. In this paper, we propose a unified DNN weight pruning framework with dynamically updated regularization terms bounded by the designated constraint, which can generate both non-structured sparsity and different kinds of structured sparsity. We also extend our method to an integrated framework for the combination of different DNN compression tasks.
References in corpus (4)
- Very Deep Convolutional Networks for Large-Scale Image Recognition
- Exploring the Regularity of Sparse Structure in Convolutional Neural Networks
- Extremely Low Bit Neural Network: Squeeze the Last Bit Out with ADMM
- Progressive DNN Compression: A Key to Achieve Ultra-High Weight Pruning and Quantization Rates using ADMM