v9.1.1
Menu
Paper breakdownQuantization & CompressionPRO2025

TurBoQuant

Online Vector Quantization with Near-Optimal Distortion Rate

Amir Zandieh, Majid Daliri, Majid Hadian, Vahab Mirrokni

5

Concepts

Paper Overview

TurBoQuant is a training-free, data-oblivious vector quantization method that achieves near-optimal compression for high-dimensional vectors — the first method to deliver zero quantization overhead at any target bit width. It tackles a fundamental problem in deploying large language models: the KV cache, which stores all previous key-value pairs during autoregressive generation, grows linearly with sequence length and dominates GPU memory at long contexts.

Consider a 70B parameter model (Llama-2-70B) with L=80L = 80 layers, H=64H = 64 attention heads, head dimension dh=128d_h = 128, at a 128K context window. The KV cache alone requires 2×80×64×128×131,072×22 \times 80 \times 64 \times 128 \times 131{,}072 \times 2 bytes (FP16) = ~43 GB — exceeding the model weights themselves on a single GPU. This memory bottleneck limits context length, batch size, and serving throughput.

Traditional quantization methods reduce this by storing values in fewer bits (INT8 = 1 byte, INT4 = 0.5 bytes), but they require quantization constants — scale factors ss and zero-points zz — stored in full precision (FP32 = 4 bytes) for every block of BB values. This overhead adds 32/B32/B extra bits per value: at 2-bit quantization with block size 32, the overhead is 1 full bit per value — a 50% penalty that undermines the compression goal. Methods like GPTQ, AWQ, and SmoothQuant all suffer from this overhead because they compute data-dependent statistics that must be stored alongside the compressed data.

TurBoQuant eliminates this overhead entirely through a two-stage approach: PolarQuant converts vectors from Cartesian to polar coordinates after random rotation, exploiting the fact that randomly rotated high-dimensional vectors have angle distributions that are known analytically (Beta distributions) — no per-block constants needed. QJL (Quantized Johnson-Lindenstrauss) then corrects the residual quantization error using only sign bits (+1/-1), the most extreme form of quantization.

The result: 3-bit quantization with zero accuracy loss on LongBench, Needle In A Haystack, ZeroSCROLLS, and RULER benchmarks, achieving 5.3x memory reduction (6x practical) and 8x speedup in attention computation on H100 GPUs using custom JAX kernels. Presented at ICLR 2026, TurBoQuant operates within a constant factor of approximately 2.7x of the Shannon rate-distortion lower bound — provably near-optimal for Gaussian sources at any bit width and any dimension.

Chapter Roadmap

Click any topic to jump in

1
Quantization Overhead

Uniform bit allocation wastes capacity on anisotropic neural activations; a distribution-aware definition is needed.

addressed by Stage 1
2
PolarQuant (Stage 1)

Recursive random rotation + polar coordinate transform uniformizes the angle distribution for efficient quantization.

3
QJL (Stage 2)

1-bit sign quantization of the angular residual yields an unbiased inner-product estimator.

followed by Stage 2
4
KV Cache Compression

Applies TurBoQuant to LLM inference, cutting K/V memory by 5x while preserving attention scores in expectation.

5
Optimality Bound

TurBoQuant's distortion matches the Shannon rate-distortion lower bound — provably optimal compression.

Concept by concept

5 sections, in reading order.

PRO

Unlock the full breakdown

All 5 concepts — figures, key points, formulas and interactive demos — are part of PixelBank Premium.

  1. 01The Quantization Overhead Problem
  2. 02Stage 1: PolarQuant — Polar Coordinate Transform
  3. 03Stage 2: QJL — 1-Bit Error Correction
  4. 04KV Cache Compression in LLMs
  5. 05Theoretical Optimality & Results