atlas

Research map / Architectures and efficiency / Quantization

QuaRot: Outlier-Free 4-Bit Inference in Rotated LLMs

NeurIPS 2024 · Architectures and efficiency › Quantization · approach: KV cache quantization · 47 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

QuaRot applies randomized Hadamard rotations, fused into weights via computational invariance plus online Hadamards, to quantize LLaMA weights, activations, and KV cache to 4 bits end-to-end with small perplexity loss.

Problem

Activation and KV-cache quantization in LLMs is hard to do at 4 bits because of outlier features in hidden states and keys/values, and prior 4-bit methods keep outliers in higher precision or require retraining.

Approach

Apply randomized Hadamard rotations to the residual stream and attention/FFN internals; fuse them into the weight matrices using computational invariance (RMSNorm commutes with orthogonal rotations), add only 1.5 online Hadamard transforms per layer, and then quantize everything to INT4 without keeping any outlier channels in higher precision.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Apply QuaRot 4-bit quantizationLLaMA 7B · WikiText-2perplexity: 5.47 -> 6.10Hurt
Apply QuaRot 4-bit quantizationLLaMA 70B · WikiText-2perplexity: 3.32 -> 3.79Hurt
Apply QuaRot 4-bit quantizationLLaMA Big · Zero-shot tasks, averageaccuracy: 77.07 -> 75.98Hurt
Use RTN rather than GPTQ for weight quantization (ablation)QuaRot on LLaMA Small · WikiText-2, 4-bitperplexity: 6.10 -> 8.37Hurt
Use RTN rather than GPTQ for weight quantization (ablation)QuaRot on LLaMA Big · WikiText-2, 4-bitperplexity: 3.79 -> 4.14Hurt
Use group size 64 instead of per-tensor quantization (ablation)QuaRot on LLaMA 7B · WikiText-2, 4-bitperplexity: 6.10 -> 5.88Helped
Use group size 256 instead of per-tensor quantization (ablation)QuaRot on LLaMA 7B · WikiText-2, 4-bitperplexity: 6.10 -> 5.98Helped
Quantize values to 3 bits while keeping keys at 4 bits (ablation)QuaRot KV cache on LLaMA 7B · WikiText-2perplexity: 5.51 -> 5.54Hurt
Quantize keys and values to 3 bits (ablation)QuaRot KV cache on LLaMA 7B · WikiText-2perplexity: 5.51 -> 5.68Hurt
Apply QuaRot rotations to weight-only GPTQ quantization (ablation)LLaMA 7B · WikiText-2, W4perplexity: 8.25 -> 5.60Helped
Replace Hadamard matrices with random orthogonal matrices (ablation)QuaRot on LLaMA 7B · WikiText-2, 4-bitperplexity: 6.10 -> 7.45Hurt
Use FP16 rather than FP32 online Hadamard transformations (ablation)QuaRot on LLaMA 7B · WikiText-2perplexity: 6.10 -> 6.08No clear effect

Surprising findings

Limitations

Experimental setup

Datasets: WikiText-2

Benchmarks: WikiText-2 PIQA WinoGrande HellaSwag ARC-Easy ARC-Challenge LAMBADA

Metrics: perplexity zero-shot accuracy prefill speedup peak memory saving

Baselines: SmoothQuant OmniQuant QUIK Atom LLM.int8() KVQuant KIVI

Builds on: computational invariance (SliceGPT) QuIP QuIP# GPTQ

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy