Optimal Post-Training Quantization Scales and Where to Find Them
About
Post-training quantization (PTQ) compresses large language models by mapping weights to low-bit representations. The scaling factor that defines the quantization grid is typically chosen using simple, data-free heuristics. In this work, we present PiSO (Piecewise Scale Optimization), an algorithm that leverages calibration data to compute the optimal channel-wise weight scales exactly and efficiently under round-to-nearest quantization. PiSO partitions the scale search space into finitely many intervals on which the objective admits a closed-form minimizer. We extend PiSO to group-wise quantization via principled heuristics and propose effective strategies for interleaving scale optimization with error correction. Experiments on Llama and Qwen models across multiple model sizes and target weight bit-widths demonstrate consistent improvements in perplexity and downstream zero-shot accuracy, both standalone and combined with error correction. In particular, we observe increased benefits as the target bit-width narrows and quantization becomes more challenging.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Language Modeling | WikiText-2 | Perplexity (PPL)6.18 | 2862 | |
| Language Modeling | WikiText-2 (test) | -- | 2416 | |
| Language Modeling | WikiText-2 (val) | Perplexity (BVS)15.6 | 179 | |
| Zero-shot Evaluation | Zero-shot Tasks Average | -- | 95 | |
| Zero-shot Evaluation | Zero-shot Benchmarks Average | Average Accuracy68.3 | 60 | |
| Commonsense Reasoning | ARC-Easy, ARC-Challenge, and HellaSwag Average | Zero-Shot Accuracy44 | 60 |