atlas

Research map / Architectures and efficiency / Quantization

OneBit: Towards Extremely Low-bit Large Language Models

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

Quantizes LLM linear-layer weights to 1 bit (sign matrix plus two FP16 value vectors) and initializes the 1-bit model via sign-value-independent decomposition, then recovers capability with quantization-aware knowledge distillation.

Problem

Post-training and existing QAT quantization degrade severely below 2-3 bits, so LLMs have not been usable at extremely low (1-bit) weight precision when quantizing pretrained models.

Approach

Represent each weight matrix as a ±1 sign matrix times two FP16 value vectors inside a new Linear layer (with Post-LayerNorm), initialize these from a rank-1 SVID (NMF/SVD of |W|), and train with logit cross-entropy plus hidden-state MSE distillation from the FP16 teacher.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use OneBit W1A16 instead of OmniQuant W2A16OPT-1.3B · WikiText2perplexity: 42.43 -> 25.42Helped
Use OneBit W1A16 instead of OmniQuant W2A16LLaMA-7B · Zero-shot averageaccuracy: 48.17 -> 51.33Helped
Add hidden-state MSE alignment to KD loss (ablation)OneBit, LLaMA-7B · WikiText2perplexity: 13.48 -> 10.19Helped
Set hidden-state MSE weight α=10 rather than α=1 (ablation)OneBit, LLaMA-7B · Zero-shot averageaccuracy: 51.33 -> 52.05Helped
Add attention-score alignment to KD + MSE (ablation)OneBit, LLaMA-7B · WikiText2perplexityHurt
Use NMF rather than SVD for SVID initialization (ablation)OneBit · Training convergenceconvergence speedHelped
Use Post-LayerNorm rather than Pre-LayerNorm (ablation)OneBit · Quantization-aware trainingtraining stabilityHelped
Include two FP16 value vectors rather than the BitNet-style architecture without them (ablation)OneBit · Knowledge transfer trainingtraining stabilityHelped

Surprising findings

Limitations

Experimental setup

Datasets: self-generated corpus from LLaMA-7B (132k entries) C4 WikiText2 alpaca_en_52k MMLU

Benchmarks: WikiText2 C4 Winogrande HellaSwag PIQA BoolQ ARC-easy ARC-challenge MMLU

Metrics: perplexity zero-shot accuracy average accuracy memory footprint

Baselines: GPTQ LLM-QAT OmniQuant BitNet Pythia-1.0B TinyLLaMA-1.1B LowRank LLaMA

Builds on: BitNet LLM-QAT OmniQuant GPTQ

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