atlas

Research map / Architectures and efficiency / Quantization

FrameQuant: Flexible Low-Bit Quantization for Transformers

ICML 2024 · Architectures and efficiency › Quantization · approach: Post-training quantization of ViTs · 0 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 FrameQuant, a post-training quantization method that quantizes transformer weights in a fusion frame (redundant tight frame) representation, enabling ~2-bit (and 2.x-bit via redundancy) weight quantization with small accuracy loss.

Problem

Post-training quantization of transformers degrades sharply at very low bit widths (around 2 bits), and existing methods only jump between integer bit widths, offering no fine-grained trade-off.

Approach

Compute a redundant fusion-frame representation of each layer's weight matrix (projected onto tight fusion frame subspaces with the previous layer's frame), clip at 2 sigma, and quantize the transformed weights with a modified GPTQ; redundancy factor r gives fractional bit widths (e.g., 2.2 bits at r=1.1) and frame robustness reduces quantization noise impact.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
increase redundancy from r=1.0 to r=1.1 (ablation)FrameQuant · ImageNet-1K val, ViT-STop-1 accuracy: 48.10 -> 61.51Helped
increase redundancy from r=1.0 to r=1.1 (ablation)FrameQuant · ImageNet-1K val, ViT-BTop-1 accuracy: 79.53 -> 80.93Helped
add TFF representation to GPTQ (ablation)GPTQ · ImageNet-1K val, ViT-STop-1 accuracy: 0.4 -> 0.88Helped
add 2σ clipping to GPTQ + TFF (ablation)FrameQuant · ImageNet-1K val, ViT-STop-1 accuracy: 0.88 -> 48.10Helped
apply 2σ clipping directly to GPTQ weights rather than to Fusion Frame representations (ablation)GPTQ · WikiText2, Llama2-7BPerplexity: 6.40e3 -> 9.45e3Hurt
increase redundancy from r=1.0 to r=1.1 (ablation)FrameQuant · C4, Llama2-7BPerplexity: 19.62 -> 11.23Helped
quantize activations to 8 bits rather than leave them at 32 bits (ablation)FrameQuant W2, ViT-S · ImageNet-1K valTop-1 accuracy: 48.17 -> 48.02No clear effect
quantize activations to 6 bits rather than leave them at 32 bits (ablation)FrameQuant W2.2, ViT-S · ImageNet-1K valTop-1 accuracy: 61.51 -> 58.35Hurt
use FrameQuant instead of GPTQLlama2-7B · inferenceInference speed (t/s): 1425.07 -> 974.20Hurt

Surprising findings

Limitations

Experimental setup

Datasets: ImageNet-1K WikiText2 C4 MNLI ARC BoolQ HellaSwag PIQA WinoGrande

Benchmarks: ImageNet-1K WikiText2 C4 MNLI ARC BoolQ HellaSwag PIQA WinoGrande COCO object detection and segmentation

Metrics: top-1 accuracy perplexity AP box/mask inference tokens/sec model size

Baselines: PTQ4ViT GPTQ QuIP ZeroQuant RepQ-ViT

Builds on: GPTQ QuIP PTQ4ViT fusion frame theory

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