The Trick to Make LLaMa Fit into Your Pocket: Meet OmniQuant, an AI Method that Bridges the Efficiency and Performance of LLMs
quantizationllmmodel-compressionpost-training-quantization
Abstraction: OmniQuant learnable post-training quantization achieves low-bit LLM compression efficiently
Key points:
- OmniQuant freezes original full-precision weights and learns only a small set of quantization parameters, avoiding the cost of quantization-aware training (QAT)
- Two components: Learnable Weight Clipping (LWC) optimizes clipping thresholds; Learnable Equivalent Transformation (LET) handles activation outliers in transformer layers
- Quantization is done layer-by-layer sequentially, enabling optimization with simple SGD on a single GPU in ~16 hours
- No extra parameters are added to the final quantized model — quantization parameters fold into the weights
- Outperforms prior PTQ methods, especially in low-bit settings where handcrafted quantization parameters degrade performance
- Applicable to both weight-only and weight-activation quantization scenarios
Connections: Llama · Meta · Model Quantization · Large Language Models · Post Training Quantization