atlas

Research map / Architectures and efficiency / Quantization

QTIP: Quantization with Trellises and Incoherence Processing

NeurIPS 2024 · Architectures and efficiency › Quantization · approach: Post-training quantization of LLMs · 4 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.

Proposes QTIP, a weight-only LLM post-training quantizer that uses ultra-high-dimensional trellis-coded quantization with a hardware-efficient bitshift trellis and compute-based pseudorandom Gaussian codes for fast inference.

Problem

Vector quantization for LLM weight PTQ is limited to low dimensions (≤8) because its codebook size grows exponentially with dimension, both in quantization cost and in inference-time cache footprint.

Approach

Replace VQ with trellis-coded quantization over very long sequences (dimension >100, e.g. 256) using a bitshift trellis so that decoding is parallel and stateless, and generate the codebook values on the fly with 1-3 instruction pseudorandom Gaussian functions (1MAD, 3INST) or a small 2D LUT (HYB), so no large codebook must be stored.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use QTIP TCQ instead of QuIP# E8P vector quantization2-bit Gaussian source quantization · i.i.d. GaussianMSE: 0.089 -> 0.069Helped
Use QTIP HYB instead of QuIP#Llama 2 7B, 2-bit quantization · Wikitext2perplexity: 6.19 -> 5.86Helped
Use QTIP HYB instead of QuIP#Llama 2 7B, 2-bit quantization · C4perplexity: 8.16 -> 7.73Helped
Use QTIP 2-bit instead of QuIP# 2-bitLlama 2 7B inference on RTX6000 Ada · batch size 1 decodingtokens/s: 186 -> 188Helped
Increase trellis size L from 8 to 16 (ablation)QTIP quantization of Llama 2 7B · Wikitext2, 2-bitperplexity: 7.83 -> 6.83Helped
Increase trellis size L from 10 to 16 (ablation)QTIP quantization of Llama 2 7B · Wikitext2, 2-bitperplexity: 7.49 -> 6.83Helped
Increase trellis size L from 12 to 16 (ablation)QTIP quantization of Llama 2 7B · Wikitext2, 2-bitperplexity: 6.97 -> 6.83Helped
Increase vector dimension V from 1 to 4 at L=12 (ablation)QTIP lookup codebook for Llama 2 7B · Wikitext2, 2-bitperplexity: 6.97 -> 7.55Hurt
Increase vector dimension V from 1 to 4 at L=16 (ablation)QTIP lookup codebook for Llama 2 7B · Wikitext2, 2-bitperplexity: 6.83 -> 6.92Hurt
Increase vector dimension V from 1 to 2 at L=16 (ablation)QTIP lookup codebook for Llama 2 7B · Wikitext2, 2-bitperplexity: 6.83 -> 6.79Helped

Surprising findings

Limitations

Experimental setup

Datasets: RedPajama WikiText2 C4

Benchmarks: WikiText2 C4 ArcC ArcE PiQA Winogrande BoolQ HellaSwag

Metrics: perplexity zero-shot accuracy decoding throughput (tokens/s) MSE on i.i.d. Gaussian

Baselines: QuIP# AQLM GPTVQ-2D PV-Tuning GPTQ AWQ OmniQuant

Builds on: QuIP# QuIP LDLQ / GPTQ-style BlockLDLQ random permutation trellis coder

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