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.

Deep Semantic Image Compression Via Cooperative Network Pruning

Pruning by Training: A Novel Deep Neural Network Compression Framework for Image Processing.

Semantic Communication Approach for Multi-Task Image Transmission

Hybrid Network Compression Via Meta-Learning

Multi-Modality Deep Network for Extreme Learned Image Compression

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

Efficient Network Compression Through Smooth-Lasso Constraint

MTP: Multi-Task Pruning for Efficient Semantic Segmentation Networks

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

Neural Network Compression by Joint Sparsity Promotion and Redundancy Reduction

An End-to-End Deep Learning Image Compression Framework Based on Semantic Analysis

Towards Efficient Network Compression Via Few-Shot Slimming.

Semantic Scalable Image Compression with Cross-Layer Priors.

DeepSIC: Deep Semantic Image Compression

Semantic-assisted image compression

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

Unified Architecture Adaptation for Compressed Domain Semantic Inference

Towards Compact CNNs via Collaborative Compression

Semantic Perceptual Image Compression with a Laplacian Pyramid of Convolutional Networks

Students and teachers learning together: a robust training strategy for neural network pruning