atlas

Research map / Large language models / Language, safety and interpretability

Compressing LLMs: The Truth is Rarely Pure and Never Simple

ICLR 2024 · Large language models › Language, safety and interpretability · approach: LLM pruning and scaling data · 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.

Introduces LLM-KICK, a knowledge-intensive benchmark suite that re-evaluates training-free pruning and quantization of Vicuna models beyond perplexity, finding that pruning degrades knowledge-intensive performance at modest sparsity while retrieval-augmented and summarization abilities hold up better.

Problem

Perplexity is an insufficient metric for judging whether training-free, data-free compression (pruning and quantization) preserves the capabilities of LLMs, since compressed models stay close to perplexity of dense models while losing knowledge-intensive abilities.

Approach

Assemble a curated multi-task benchmark (factoid QA, MMLU, in-context retrieval-augmented QA, in-context summarization, MT-Bench-style instruction following) and define "matching" compressed LLMs as those within 5% of dense performance, to evaluate SoTA pruning (Magnitude, SparseGPT, Wanda) and quantization (GPTQ) methods on Vicuna models.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Apply unstructured pruning, including magnitude pruning, SparseGPT, or WandaVicuna-7B · FreebaseQAaccuracyHurt
Use structured N:M pruningVicuna models · LLM-KICK knowledge-intensive taskstask performanceHurt
Use quantization instead of pruningVicuna models · LLM-KICK taskstask performanceHelped
Provide relevant external knowledge in the promptCompressed Vicuna models · TriviaQA, in-context retrieval augmented QAtask performanceHelped
Apply compression to summarize increasingly long storiesCompressed Vicuna-7B · CNN/DailyMail in-context summarizationsummary qualityHurt
Apply 50% unstructured magnitude pruning (ablation)Vicuna-7B · MMLUaccuracy: 0.471 -> 0.050Hurt
Apply 50% unstructured SparseGPT pruning (ablation)Vicuna-7B · MMLUaccuracy: 0.471 -> 0.308Hurt
Apply 50% unstructured Wanda pruning (ablation)Vicuna-7B · MMLUaccuracy: 0.471 -> 0.386Hurt
Increase Wanda calibration sample count (ablation)Vicuna-7B · MMLU at 70% sparsityzero-shot performanceNo clear effect
Increase SparseGPT calibration sample count (ablation)Vicuna-7B · MMLU at 70% sparsityzero-shot performanceHelped
Add 2-3 in-context examples (ablation)Wanda-pruned Vicuna-7B · MMLUperformance at high sparsityHelped
Use LLM-int8() rather than AWQ or GPTQ 8-bitVicuna-7B · Factoid-QA, MMLU, Retrieval Augmented QA, Instruction Followingtask performanceHelped

Surprising findings

Limitations

Experimental setup

Datasets: FreebaseQA TriviaQA MMLU CNN/DailyMail MT-Bench prompts WikiText

Benchmarks: FreebaseQA MMLU TriviaQA CNN/DailyMail MT-Bench

Metrics: accuracy exact match perplexity GPT-4 score (0-10) unique token count

Baselines: Magnitude pruning SparseGPT Wanda GPTQ AWQ LLM.int8()

Builds on: SparseGPT Wanda GPTQ MT-Bench

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