atlas

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

Any-Precision LLM: Low-Cost Deployment of Multiple, Different-Sized LLMs

ICML 2024 · Large language models › Reasoning and chain of thought · approach: Efficient LLM serving · 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 any-precision LLMs, obtained by incrementally upscaling a 3-bit SqueezeLLM seed model into a single parent model whose bit-prefixes serve 3–8-bit models, plus a bitplane-based GPU kernel that reduces memory traffic at lower bit-widths.

Problem

Deploying multiple LLMs of different sizes multiplies memory cost and training cost, and prior any-precision DNN methods require QAT from scratch and do not reduce memory bandwidth for LLM inference.

Approach

Incremental upscaling: starting from a low-bit non-uniform (clustering-based) PTQ seed model, each cluster is split into two sub-clusters per added bit, so lower-bit models are bit-prefixes of the parent; paired with a bitplane weight layout and GPU kernel that loads only the required bitplanes.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
incremental upscaling of SqueezeLLM from 3-bit seed (vs independently quantized)SqueezeLLM · Wikitext2/PTB/C4 perplexity, Llama-2-7B, OPT, Mistralperplexity: <0.1 in most casesNo clear effect
incremental upscaling of SqueezeLLM from 3-bit seedSqueezeLLM · zero-shot average accuracy, five tasksaccuracy: within 0.2%No clear effect
incremental upscaling applied to uniform GPTQ (ablation)GPTQ · Wikitext2 perplexity, Llama-2-7B 4-bitperplexity: 5.68 -> 36.82Hurt
incremental upscaling applied to uniform AWQ (ablation)AWQ · Wikitext2 perplexity, Llama-2-7B 4-bitperplexity: 5.59 -> INFHurt
bitplane-based kernel with weight bitplane layout optimization, bit-transpose and table merging vs SqueezeLLM kernelany-precision GEMV kernel · matvec speedup vs cuBLAS FP16, RTX 4090 and Jetson AGX Orinspeedup: on par on RTX 4090/4070 Laptop; substantially faster on Jetson (e.g. 3.02x vs 1.94x at 4-bit)Mixed
table lookup merging for 3-bit (merging two 3-bit indices into one 6-bit lookup) (ablation)any-precision kernel 3-bit · kernel microbenchmark ablationspeedupHelped
weight bitplane layout optimization, improved bit-transpose (ablation)any-precision kernel · kernel microbenchmark ablationspeedupHelped
end-to-end integration of the any-precision engine into TensorRT-LLMLlama-2-7B · end-to-end throughput, RTX 4090 and RTX 4070 LaptopthroughputHelped

Surprising findings

Limitations

Experimental setup

Datasets: WikiText2 C4 Penn Treebank

Benchmarks: WikiText2 PTB C4 ARC-easy ARC-challenge HellaSwag PIQA WinoGrande

Metrics: perplexity zero-shot accuracy matrix-vector latency end-to-end throughput memory footprint

Baselines: SqueezeLLM AWQ GPTQ LUT-GEMM ExLlamaV2 TensorRT-LLM cuBLAS FP16

Builds on: SqueezeLLM any-precision DNN AWQ GPTQ LUT-GEMM

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