Activation Sensitivity as a Unifying Principle for Post-Training Quantization
Abstract
Post-training quantization (PTQ) methods for large language models compress pretrained weights to low bit-widths without retraining by optimizing layer-local proxy objectives. Activation-aware scaling methods such as AWQ protect input channels with large activation norms, whereas second-order error-compensation methods such as GPTQ redistribute rounding error using the inverse of the unweighted input covariance matrix. We show that both paradigms -- along with diagonal Fisher quantization, block reconstruction, and Hadamard rotation methods -- are special cases of minimizing a single loss-derived quadratic form governed by gradient-weighted activation sensitivity. From a Gauss-Newton expansion of the task loss at a linear layer , we prove that under isotropic output curvature the expected excess loss from weight perturbation reduces to , where weights each calibration token's outer product by its squared output-gradient norm . Within this framework, AWQ corresponds to a diagonal approximation of with uniform token weights () optimized via channel scaling, while GPTQ corresponds to Cholesky error compensation on under . We derive closed-form optimal scaling exponents for diagonal sensitivity protection, formulate sensitivity-weighted Cholesky quantization (SAQ-GPTQ) on , and prove bounds on token-gradient heterogeneity bias, cross-layer error amplification across residual blocks, and finite-sample calibration concentration. Controlled numerical simulations confirm that under token-gradient heterogeneity, replacing with preserves top-channel saliency recall and reduces INT3/INT4 excess loss by roughly over unweighted GPTQ.
Figures & tables
| Method Family | Sensitivity Metric | Token | Off-Diag | Mechanism |
|---|---|---|---|---|
| RTN (Baseline) | (Uniform) | None ( ) | Ignored | Per-group rounding |
| AWQ [ 3 ] | Uniform ( ) | Ignored | Scaling | |
| SmoothQuant [ 8 ] | Uniform ( ) | Ignored | W/X scale migration | |
| OBC / GPTQ [ 14 , 2 ] | Uniform ( ) | Cholesky | Sequential OBS update | |
| QuIP / QuIP# [ 16 , 17 ] | Uniform ( ) | Cholesky | Rotation + lattice/OBS | |
| BRECQ [ 24 , 26 ] | Block-local | Intra-block | Block reconstruction |
| INT4 ( ) | INT3 ( ) | ||||
|---|---|---|---|---|---|
| Method | Sensitivity Proxy | Excess Loss | Recon. MSE | Excess Loss | Recon. MSE |
| RTN | None ( ) | ||||
| AWQ [ 3 ] | |||||
| SAQ-Scale (Ours) | |||||
| GPTQ [ 2 ] | |||||
| SAQ-GPTQ (Ours) | |||||
LLM Quantization
Large language model quantization reduces numerical precision to lower memory use and accelerate training or inference.
See all 381 papers