Abstract:Quantization is of significance for compressing the over-parameterized deep neural models and deploying them on resource-limited devices. Fixed-precision quantization suffers from performance drop due to the limited numerical representation ability. Conversely, mixed-precision quantization (MPQ) is advocated to compress the model effectively by allocating heterogeneous bit-width for layers. MPQ is typically organized into a searching-retraining two-stage process. In this paper, we devise a one-shot training-searching paradigm for mixed-precision model compression. Specifically, in the first stage, all potential bit-width configurations are coupled and thus optimized simultaneously within a set of shared weights. However, our observations reveal a previously unseen and severe bit-width interference phenomenon among highly coupled weights during optimization, leading to considerable performance degradation under a high compression ratio. To tackle this problem, we first design a bit-width scheduler to dynamically freeze the most turbulent bit-width of layers during training, to ensure the rest bit-widths converged properly. Then, taking inspiration from information theory, we present an information distortion mitigation technique to align the behavior of the bad-performing bit-widths to the well-performing ones. In the second stage, an inference-only greedy search scheme is devised to evaluate the goodness of configurations without introducing any additional training costs. Extensive experiments on three representative models and three datasets demonstrate the effectiveness of the proposed method. Code can be available on \href{<a class="link-external link-https" href="https://www.github.com/1hunters/retraining-free-quantization" rel="external noopener nofollow">this https URL</a>}{<a class="link-external link-https" href="https://github.com/1hunters/retraining-free-quantization" rel="external noopener nofollow">this https URL</a>}.

Post-training Quantization or Quantization-aware Training? That is the Question

Hessian-based Mixed-Precision Quantization with Transition Aware Training for Neural Networks

Solving Oscillation Problem in Post-Training Quantization Through a Theoretical Perspective

Attention Round for post-training quantization

COMQ: A Backpropagation-Free Algorithm for Post-Training Quantization

Bit-shrinking: Limiting Instantaneous Sharpness for Improving Post-training Quantization

decoupleQ: Towards 2-bit Post-Training Uniform Quantization via decoupling Parameters into Integer and Floating Points

Error-aware Quantization through Noise Tempering

Attention-aware Post-training Quantization without Backpropagation

Optimizing Large Language Models through Quantization: A Comparative Analysis of PTQ and QAT Techniques

One Model for All Quantization: A Quantized Network Supporting Hot-Swap Bit-Width Adjustment

PTQ-SL: Exploring the Sub-layerwise Post-training Quantization

Retraining-free Model Quantization via One-Shot Weight-Coupling Learning

Optimization-based Post-training Quantization with Bit-split and Stitching

BRECQ: Pushing the Limit of Post-Training Quantization by Block Reconstruction

RAPQ: Rescuing Accuracy for Power-of-Two Low-bit Post-training Quantization

EasyQuant: Post-training Quantization via Scale Optimization

Understanding the difficulty of low-precision post-training quantization of large language models

Loss Aware Post-training Quantization

Automated Backend-Aware Post-Training Quantization

Mixed-Precision Neural Network Quantization Via Learned Layer-Wise Importance