Papers · AWQ
AI PAPER

AWQ: Activation-aware Weight Quantization for LLM Compression and Acceleration

Developed an activation-aware quantization method that identifies and preserves salient weight channels, achieving better quality than GPTQ at the same compression level.

Authors
Ji Lin, Jiaming Tang, Haotian Tang +4
Institution
MIT
Published
June NaN, 2023
Citations
900
Category
Efficiency
Impact
notable
PAPER EXPLAINED

Plain-English Summary

Quantizing a language model (reducing the precision of its weights from 16-bit to 4-bit) saves memory and speeds up inference, but naive quantization degrades quality. AWQ observed that a small fraction of weights are far more important than others — specifically, the weights connected to channels that produce large activations. By identifying these critical weights and keeping them at higher precision, AWQ achieves better compression quality than methods that treat all weights equally.

The insight is simple: not all weights matter equally. About 1% of weights are responsible for most of the model’s performance, and these correspond to channels with large activation magnitudes.

Key Innovation

AWQ identifies “salient” weight channels by measuring the magnitude of corresponding activations on calibration data. Rather than keeping these weights at higher precision (which would complicate the hardware implementation), AWQ applies per-channel scaling that effectively protects important weights from quantization error. This achieves the efficiency of uniform quantization with the quality benefits of mixed-precision.

The method requires no backpropagation or retraining — just a forward pass on calibration data to identify important channels. This makes it extremely fast to apply.

Impact on the Field

AWQ became the preferred quantization method for production inference systems. Its superior quality-compression tradeoff compared to GPTQ made it the default in vLLM and TensorRT-LLM, the two most popular inference engines for serving language models at scale.

The activation-awareness principle also influenced subsequent quantization research, establishing that understanding which parts of a model are most important enables better compression across the board.

Models That Built on This

AWQ is integrated into vLLM (the dominant open-source serving framework), NVIDIA’s TensorRT-LLM, and Hugging Face’s quantization tools. Most production deployments of open-weight models use AWQ quantization. The technique has become standard for deploying models efficiently in both cloud and edge environments.