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 layers, attention heads, head dimension , at a 128K context window. The KV cache alone requires 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 and zero-points — stored in full precision (FP32 = 4 bytes) for every block of values. This overhead adds 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
Quantization Overhead
Uniform bit allocation wastes capacity on anisotropic neural activations; a distribution-aware definition is needed.
PolarQuant (Stage 1)
Recursive random rotation + polar coordinate transform uniformizes the angle distribution for efficient quantization.
QJL (Stage 2)
1-bit sign quantization of the angular residual yields an unbiased inner-product estimator.
KV Cache Compression
Applies TurBoQuant to LLM inference, cutting K/V memory by 5x while preserving attention scores in expectation.
Optimality Bound
TurBoQuant's distortion matches the Shannon rate-distortion lower bound — provably optimal compression.
Concept by concept
5 sections, in reading order.
Unlock the full breakdown
All 5 concepts — figures, key points, formulas and interactive demos — are part of PixelBank Premium.
- 01The Quantization Overhead Problem
- 02Stage 1: PolarQuant — Polar Coordinate Transform
- 03Stage 2: QJL — 1-Bit Error Correction
- 04KV Cache Compression in LLMs
- 05Theoretical Optimality & Results
Keep reading
Related Papers
Go deeper