Abstract:Network pruning and quantization are proven to be effective ways for deep model compression. To obtain a highly compact model, most methods first perform network pruning and then conduct quantization based on the pruned model. However, this strategy may ignore that the pruning and quantization would affect each other and thus performing them separately may lead to sub-optimal performance. To address this, performing pruning and quantization jointly is essential. Nevertheless, how to make a trade-off between pruning and quantization is non-trivial. Moreover, existing compression methods often rely on some pre-defined compression configurations (i.e., pruning rates or bitwidths). Some attempts have been made to search for optimal configurations, which however may take unbearable optimization cost. To address these issues, we devise a simple yet effective method named Single-path Bit Sharing (SBS) for automatic loss-aware model compression. To this end, we consider the network pruning as a special case of quantization and provide a unified view for model pruning and quantization. We then introduce a single-path model to encode all candidate compression configurations, where a high bitwidth value will be decomposed into the sum of a lowest bitwidth value and a series of re-assignment offsets. Relying on the single-path model, we introduce learnable binary gates to encode the choice of configurations and learn the binary gates and model parameters jointly. More importantly, the configuration search problem can be transformed into a subset selection problem, which helps to significantly reduce the optimization difficulty and computation cost. In this way, the compression configurations of each layer and the trade-off between pruning and quantization can be automatically determined. Extensive experiments on CIFAR-100 and ImageNet show that SBS significantly reduces computation cost while achieving promising performance. For example, our SBS compressed MobileNetV2 achieves 22.6× Bit-Operation (BOP) reduction with only 0.1% drop in the Top-1 accuracy.

Hybrid Network Compression Via Meta-Learning

MCMC: Multi-Constrained Model Compression Via One-Stage Envelope Reinforcement Learning.

Efficient Network Compression Through Smooth-Lasso Constraint

CLIP-Q: Deep Network Compression Learning by In-parallel Pruning-Quantization

Single-path Bit Sharing for Automatic Loss-aware Model Compression

Towards Optimal Compression: Joint Pruning and Quantization

OPQ: Compressing Deep Neural Networks with One-shot Pruning-Quantization

Automated Model Compression by Jointly Applied Pruning and Quantization

Deep Compression: Compressing Deep Neural Networks with Pruning, Trained Quantization and Huffman Coding

A Highly Efficient Training-Aware Convolutional Neural Network Compression Paradigm

Deep Neural Network Compression With Single and Multiple Level Quantization

Neural Network Compression by Joint Sparsity Promotion and Redundancy Reduction

Optimal Brain Compression: A Framework for Accurate Post-Training Quantization and Pruning

Learning a Virtual Codec Based on Deep Convolutional Neural Network to Compress Image

Multi-Modality Deep Network for Extreme Learned Image Compression

Deep Semantic Image Compression Via Cooperative Network Pruning

Kernel-wise difference minimization for convolutional neural network compression in metaverse

Differentiable Joint Pruning and Quantization for Hardware Efficiency

Adaptive Layerwise Quantization for Deep Neural Network Compression

Region-of-interest and channel attention-based joint optimization of image compression and computer vision