English

Pushing the Limits of Large Language Model Quantization via the Linearity Theorem

Machine Learning 2024-11-27 v1

Abstract

Quantizing large language models has become a standard way to reduce their memory and computational costs. Typically, existing methods focus on breaking down the problem into individual layer-wise sub-problems, and minimizing per-layer error, measured via various metrics. Yet, this approach currently lacks theoretical justification and the metrics employed may be sub-optimal. In this paper, we present a "linearity theorem" establishing a direct relationship between the layer-wise 2\ell_2 reconstruction error and the model perplexity increase due to quantization. This insight enables two novel applications: (1) a simple data-free LLM quantization method using Hadamard rotations and MSE-optimal grids, dubbed HIGGS, which outperforms all prior data-free approaches such as the extremely popular NF4 quantized format, and (2) an optimal solution to the problem of finding non-uniform per-layer quantization levels which match a given compression constraint in the medium-bitwidth regime, obtained by reduction to dynamic programming. On the practical side, we demonstrate improved accuracy-compression trade-offs on Llama-3.1 and 3.2-family models, as well as on Qwen-family models. Further, we show that our method can be efficiently supported in terms of GPU kernels at various batch sizes, advancing both data-free and non-uniform quantization for LLMs.

Keywords

Cite

@article{arxiv.2411.17525,
  title  = {Pushing the Limits of Large Language Model Quantization via the Linearity Theorem},
  author = {Vladimir Malinovskii and Andrei Panferov and Ivan Ilin and Han Guo and Peter Richtárik and Dan Alistarh},
  journal= {arXiv preprint arXiv:2411.17525},
  year   = {2024}
}
R2 v1 2026-06-28T20:13:18.062Z