atlas

Research map / Architectures and efficiency / Quantization

QBB: Quantization with Binary Bases for LLMs

NeurIPS 2024 · Architectures and efficiency › Quantization · approach: Post-training weight quantization · 2 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 QBB, a post-training weight-only quantization that decomposes LLM weights into N=4 binary matrices with fp16 scales (analytical residual init, iterative gradient refinement) and calibrates scales with data-free student-teacher distillation.

Problem

Low-bit post-training quantization of LLMs, especially sub-7B models, degrades sharply below 4 bits, and existing methods still rely on mixed-precision multiplications that have limited hardware support.

Approach

Approximate each weight matrix as a sum of N binary matrices with fp16 scaling vectors, initialized by cascaded analytical residual binarization and refined layer-wise by block-coordinate gradient descent; then calibrate only the scales with data-free logit and feature distillation on self-generated text, so inference needs almost only summations.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
add data-free holistic distillation to input-agnostic binarization (ablation)QBB · WikiText2, Phi-2 2.7Bperplexity: 12.32 -> 10.48Helped
add data-free holistic distillation to input-agnostic binarization (ablation)QBB · WikiText2, LLaMA-2 7Bperplexity: 5.37 -> 5.21Helped
use KD-MSE with gradual teacher/student block swapping instead of CE, KD, or KD-MSE (ablation)QBB calibration on Phi-2 2.7B · WikiText2perplexity: 11.82 -> 10.40; 10.61 -> 10.40; 10.48 -> 10.40Helped
initialize binary bases from FP16 weights rather than GPTQ or OmniQuant int4 weights (ablation)QBB on LLaMA-2 7B · WikiText2perplexity: 5.49 -> 7.10; 5.21 -> 7.10Hurt
initialize binary bases from OmniQuant rather than GPTQ int4 weights (ablation)QBB on LLaMA-2 7B · WikiText2perplexity: 5.49 -> 5.21Helped
use filtered generated samples with the highest teacher-student discrepancy (ablation)QBB calibration on Phi-2 2.7B · WikiText2perplexityHelped
use cascaded residual initialization rather than direct global-error initialization (ablation)QBB binary bases · Phi-2 2.7B layer reconstructionL2 reconstruction error: 2× smallerHelped
use iterative one-binary-weight-at-a-time optimization rather than joint optimization (ablation)QBB binary bases · Phi-2 2.7B layer reconstructionaverage reconstruction error: one order of magnitude higherHelped

Surprising findings

Limitations

Experimental setup

Datasets: self-generated synthetic text from the teacher model

Benchmarks: WikiText2

Metrics: perplexity

Baselines: RTN GPTQ AWQ OmniQuant SpQR QuIP#

Builds on: BinaryConnect XNOR-Net GPTQ OmniQuant Data-free quantization (Liu et al. 2023)

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