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.

LBS: Loss-aware Bit Sharing for Automatic Model Compression

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

ABS: Automatic Bit Sharing for Model Compression

Lossy and Lossless (L$^2$) Post-training Model Size Compression

Compression for Better: A General and Stable Lossless Compression Framework

Context-Based Lossless Compression of Mosaic Image with Bayer Pattern

Adaptive Layerwise Quantization for Deep Neural Network Compression

Learning Scalable ℓ∞-Constrained Near-lossless Image Compression Via Joint Lossy Image and Residual Compression

Learned Block-based Hybrid Image Compression

Learning Content-Weighted Deep Image Compression

Learned Image Compression With Gaussian-Laplacian-Logistic Mixture Model and Concatenated Residual Modules

Hybrid Network Compression Via Meta-Learning

Lossless Compression with Probabilistic Circuits

Weightless: Lossy Weight Encoding For Deep Neural Network Compression

Modeling Lost Information in Lossy Image Compression.

Downscaling and Overflow-aware Model Compression for Efficient Vision Processors

LCS: Learning Compressible Subspaces for Adaptive Network Compression at Inference Time

Make Lossy Compression Meaningful for Low-Light Images

Exploring Structural Sparsity in Neural Image Compression

BitStack: Fine-Grained Size Control for Compressed Large Language Models in Variable Memory Environments