atlas

Research map / Large language models / Instruction and fine-tuning

Cherry on Top: Parameter Heterogeneity and Quantization in Large Language Models

NeurIPS 2024 · Large language models › Instruction and fine-tuning · approach: Efficient LLM fine-tuning systems · 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.

Reveals parameter heterogeneity in LLMs (a ~1% subset of "cherry" parameters has outsized loss impact) and proposes CherryQ, a QAT-based mixed-precision quantizer that keeps cherry parameters in FP16 while quantizing the rest end-to-end.

Problem

Post-training quantization of LLMs to low bit-widths degrades performance, and existing mixed-precision methods struggle to identify important parameters and to jointly optimize high-precision and low-precision parameters.

Approach

Quantify parameter impact via the diagonal Hessian (approximated by the Fisher information), show that a tiny fraction of parameters dominate it, keep those "cherry" parameters in FP16, and train the whole mixed-precision model end-to-end with gradient descent for cherry parameters and STE for normal parameters.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use CherryQ instead of OmniQuant for 3-bit quantizationLLaMA2-7B-3bit-g128 · C4 validationperplexity: 7.75 -> 7.39Helped
Use CherryQ instead of OmniQuant for 3-bit quantizationLLaMA2-7B-3bit-g128 · WikiText-2perplexity: 6.03 -> 5.93Helped
Use CherryQ instead of SqueezeLLM for 3-bit quantizationLLaMA2-7B-3bit-g64 · C4 validationperplexity: 7.51 -> 7.34Helped
Use CherryQ instead of SqueezeLLM for 3-bit quantizationLLaMA2-13B-3bit-g128 · WikiText-2perplexity: 5.23 -> 5.26Hurt
Use CherryQ instead of OmniQuant for 4-bit quantizationLLaMA2-13B-4bit-g128 · WikiText-2perplexity: 4.95 -> 4.99Hurt
Use CherryQ instead of OmniQuant for 2-bit quantizationLLaMA2-7B-2bit-g128 · C4 validationperplexity: 15.02 -> 9.55Helped
Use CherryQ instead of QATLLaMA2-7B-3bit-g64 · OpenLLM averageaverage score: 46.9 -> 48.7Helped
Use CherryQ instead of QATVicuna-1.5-13B-3bit-g128 · zero-shot MMLUaccuracy: 50.9 -> 51.9Helped
Select cherry parameters by impact rather than weight (ablation)LLaMA2-7B-3bit-g64 · C4perplexity: 7.93 -> 7.34Helped
Select cherry parameters by impact rather than activation (ablation)LLaMA2-7B-3bit-g64 · C4perplexity: 7.37 -> 7.34Helped
Select cherry parameters by impact rather than activation (ablation)LLaMA2-7B-3bit-g64 · WikiText-2perplexity: 5.89 -> 5.87Helped

Surprising findings

Limitations

Experimental setup

Datasets: C4 WikiText-2 ShareGPT Vicuna-bench MMLU HuggingFace OpenLLM Leaderboard tasks

Benchmarks: C4 WikiText-2 HuggingFace OpenLLM Leaderboard Vicuna-bench MMLU

Metrics: perplexity accuracy win-tie-lose ratio

Baselines: QAT GPTQ SqueezeLLM OmniQuant AWQ

Builds on: GPTQ OmniQuant SqueezeLLM AWQ LLM-QAT

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