atlas

Research map / Large language models / Reasoning and chain of thought

Learning from Students: Applying t-Distributions to Explore Accurate and Efficient Formats for LLMs

ICML 2024 · Large language models › Reasoning and chain of thought · approach: Routing, pruning and distillation for reasoning · 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.

Profiles 30 DNN weight/activation distributions as Student's t, derives the Student Float (SF4) lookup datatype, and proposes supernormal E2M1 and APoT4 variants, mapping an accuracy-area Pareto frontier across 11 four-bit-or-so datatypes.

Problem

Choosing 4-bit quantization datatypes that match LLM weight and activation distributions while keeping MAC unit hardware cost low; existing options trade accuracy (lookup formats like NF4) against hardware efficiency (INT4).

Approach

Most DNN weights and activations fit Student's t-distributions with small degrees of freedom; deriving a quantile-based datatype for t(nu=5) gives SF4, and adding "supernormal" values to E2M1 (reusing the negative zero code) yields hardware-cheap formats with higher accuracy.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Replace NF4 with SF4 for weight-only quantizationLLaMA2-7B · Weight-only zero-shot tasksaverage relative accuracy change from FP32: -1.10 -> -0.22Helped
Add super-precision support to E2M1 (ablation)Phi-2 · Weight-only LAMBADAaccuracy: 59.95 -> 61.73Helped
Add super-precision support to E2M1 (ablation)Phi-2 · Weight-only WikiText-2perplexity: 11.97 -> 11.92Helped
Add super-range support to E2M1 (ablation)Phi-2 · Weight-only WikiText-2perplexity: 11.97 -> 12.58Hurt
Add super-precision support to E2M1 (ablation)LLaMA2-7B · Weight-only zero-shot tasksaverage relative accuracy change from FP32: -0.85 -> -0.80Helped
Add super-precision support to APoT4 (ablation)LLaMA2-7B · Weight-only zero-shot tasksaverage relative accuracy change from FP32: -0.86 -> -0.39Helped
Set SF4 degrees of freedom to 5 rather than 3 (ablation)OPT-125M · LAMBADAaccuracy: 37.18 -> 38.56Helped
Set SF4 degrees of freedom to 5 rather than 6 (ablation)Phi-2 · LAMBADAaccuracy: 60.92 -> 61.56Helped
Quantize vision model weights and activations with SF4 instead of NF4ResNet18 · ImageNetaccuracy: 58.04 -> 63.12Helped
Use INT4 rather than E2M1 for the MAC datatypeMAC unit · Hardware synthesisarea: 170.4 -> 160.7Helped

Surprising findings

Limitations

Experimental setup

Datasets: LAMBADA WikiText-2 HellaSwag WinoGrande PIQA BoolQ ARC-c ImageNet

Benchmarks: LAMBADA WikiText-2 HellaSwag WinoGrande PIQA BoolQ ARC-c ImageNet

Metrics: accuracy perplexity relative accuracy change MAC area power

Baselines: NF4 INT4 E2M1-I E2M1-B E3M0 APoT4 SmoothQuant GPTQ

Builds on: NF4 APoT SmoothQuant GPTQ bitsandbytes FP4

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