atlas

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

Compressing Large Language Models by Joint Sparsification and Quantization

ICML 2024 · Large language models › Language, safety and interpretability · approach: LLM pruning and scaling data

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 JSQ, a joint sparsification and quantization framework for LLMs that uses a salience-with-activation-range sparsity metric and a simulated-annealing-searched activation editor to compress models at high compression ratios.

Problem

Single-technique LLM compression (sparsification or quantization alone) degrades sharply at high compression ratios, and outliers hinder both sparsification and quantization when combined naively.

Approach

Combine unstructured sparsity and quantization in one pipeline, with a Salience with Activation Range (SAR) metric that penalizes weights whose removal enlarges the output activation range, and a search-based activation editor that clamps activation outliers with a per-layer strength found by simulated annealing.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use JSQ instead of SparseGPT+SmoothQuantLLaMA-7B · WikiText2perplexity: 11.98 -> 11.04Helped
Use JSQ instead of SparseGPT+SmoothQuantLLaMA-7B · MMLUaccuracy: 30.2 -> 30.0Hurt
Add auxiliary salience A to the sparsity metric (ablation)JSQ on LLaMA-7B · WikiText2perplexity: 11.75 -> 11.04Helped
Add activation editing (ablation)JSQ on LLaMA-7B · WikiText2perplexity: 11.62 -> 11.04Helped
Use simulated annealing search rather than fixed editing strength (ablation)JSQ on LLaMA-7B · WikiText2perplexity: 11.29 -> 11.04Helped
Use GPTQ as the subsequent quantization approach (ablation)JSQ on LLaMA-7B · WikiText2perplexity: 11.04 -> 11.06No clear effect
Finetune the compressed model (ablation)JSQ on LLaMA-7B · WikiText2perplexity: 11.04 -> 10.30Helped
Vary λ in the SAR metric (ablation)JSQ on LLaMA-7B · WikiText2perplexity: 11.15; 11.04; 14.22; 147.83; 136.64Mixed
Increase calibration data from 8 to 128 samples (ablation)JSQ on LLaMA-7B · WikiText2perplexity: 11.14 -> 11.04Helped

Surprising findings

Limitations

Experimental setup

Datasets: C4 WikiText2 PIQA BoolQ MMLU HellaSwag ARC-easy ARC-challenge WinoGrande

Benchmarks: WikiText2 PIQA BoolQ MMLU HellaSwag ARC-easy ARC-challenge WinoGrande

Metrics: perplexity zero-shot accuracy MACs model size

Baselines: Wanda LLM-Pruner SparseGPT SmoothQuant OmniQuant Wanda+SmoothQuant SparseGPT+SmoothQuant

Builds on: Wanda SmoothQuant SparseGPT 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