Prune 50–60% of GPT-scale weights in one pass, no retraining, with minor accuracy loss

January 2, 20237 min

Overview

Decision SnapshotReady For Pilot

Strong empirical evidence on multiple GPT-family models and datasets; code released. Some deployment caveats remain (hardware support for sparse kernels, semi-structured accuracy trade-offs).

Citations69

Evidence Strength0.90

Confidence0.88

Risk Signals10

Trust Signals

Findings with numeric evidence: 4/4

Findings with evidence refs: 4/4

Results with explicit delta: 3/5

Reproducibility

Status: Code + data available

Open source: Partial

At A Glance

Cost impact: 90%

Production readiness: 80%

Novelty: 70%

Authors

Elias Frantar, Dan Alistarh

Links

Abstract / PDF / Code / Data

Why It Matters For Business

SparseGPT can cut model memory and inference compute roughly in half for massive GPT models, enabling cheaper hosting and faster inference without retraining. Joint sparsity+quantization can match lower-bit storage with better accuracy than pure quantization.

Who Should Care

Summary TLDR

SparseGPT is a fast post-training pruning method that can make very large GPT-family models (e.g., OPT-175B, BLOOM-176B) 50–60% sparse in one shot, without finetuning, while keeping perplexity and zero-shot accuracy nearly intact. The method reuses layer Hessian information to perform efficient weight reconstruction, supports hardware-friendly 2:4 and 4:8 patterns, and can be combined with weight quantization (e.g., joint 50% sparsity + 4-bit) for further memory savings. The authors provide code and report runs on a single A100 GPU (≈4 hours for 175B).

Problem Statement

Large GPT-family models are expensive to store and serve. Existing accurate pruning methods need expensive retraining or do not scale to 10–100B+ parameters. We need a fast, accurate post-training pruning method that works at GPT scale without retraining.

Main Contribution

SparseGPT: a one-shot, post-training pruning algorithm that scales to 10–100+B parameter GPT models without finetuning.

An efficient approximate reconstruction solver that reuses a sequence of inverse Hessians across columns to reduce computation and memory.

Key Findings

Large GPT models can be pruned to 50–60% unstructured sparsity in one shot with little accuracy loss.

Numbers5060% sparsity; removes ≈100B weights from OPT-175B/BLOOM-176B

Practical UseYou can drop half the weights of GPT-scale models for inference without retraining; expect big memory savings and minor accuracy changes on evaluated benchmarks.

Evidence RefAbstract; Section 4; Discussion

SparseGPT runs quickly on a single GPU for the largest open models.

NumbersOPT-175B / BLOOM-176B sparsification ≲4.5 hours on one A100 (80GB)

Practical UseCompressing a production-scale model can be done in hours on one high-memory GPU — feasible for deployment pipelines.

Evidence RefAbstract; Section 4 (Setup)

Results

MetricValueBaselineDeltaSplit / DatasetEvidenceEvidence Ref
Perplexity (OPT-175B dense)8.35raw-WikiText2Dense OPT-175B perplexity reported as 8.35Table in Section 4 / Figure 1
Perplexity (OPT-175B, 50% sparsity + 4-bit joint)8.29Dense (8.35)-0.06raw-WikiText250% + 4-bit PPL 8.29 reported and compared to 3-bit baselineSection 4 (Joint Sparsification & Quantization); Figure 6

What To Try In 7 Days

Run SparseGPT on a large model you already use and profile memory and latency (use 128 calibration segments).

Try joint 50% sparsity + 4-bit quantization and compare to your current quantized model for accuracy vs storage.

If targeting GPU speedups, test semi-structured 2:4/4:8 patterns and measure real end-to-end latency with your inference stack.

Optimization Features

Infra Optimization
Single-GPU (A100 80GB) runnable for 175B models in hours
Model Optimization
Post-training pruning (one-shot)Unstructured pruningSemi-structured n:m pruning (2:4, 4:8)
System Optimization
Column-block lazy updates to improve compute-to-memory ratio
Training Optimization
No retraining / no finetuning (post-training)
Inference Optimization
Reduced weight memory footprintPotential inference speedups on CPU (DeepSparse) and GPUs supporting n:m kernels

Reproducibility

Code AvailableYes
Data AvailableYes
Open Source StatusPartial
LicenseUnknown

Data URLs

C4 (first shard) used for calibration; public C4 dataset referenced in paper

Risks & Boundaries

Limitations

Semi-structured patterns (2:4,4:8) reduce accuracy more on smaller models than on the largest ones.

Real end-to-end speedups depend on hardware and inference software; reported layer speedups may not equal full-system gains.

When Not To Use

If you need strictly lossless accuracy for your task (no tolerance for any PPL or task drop).

If your deployment stack lacks optimized sparse kernels or does not support 2:4-style acceleration.

Failure Modes

Simple magnitude pruning collapses accuracy at moderate sparsities for GPT-scale models (observed >30% collapse).

Joint semi-structured sparsity may harm smaller models more than large ones; poor layer selection for partial n:m leads to accuracy loss.

Core Entities

Models

OPT-175BBLOOM-176BOPT family (2.7B,6.7B,13B,30B,66B,175B)

Metrics

PerplexityAccuracyEnd-to-end inference speedup

Datasets

raw-WikiText2PTBC4 subsetLambadaARC (easy/challenge)PIQAStoryCloze

Benchmarks

Perplexity (language modeling)Accuracy

Context Entities

Models

AdaPrune (baseline)Magnitude pruning (baseline)GPTQ (joint quantization baseline)

Metrics

Perplexity (HuggingFace procedure)Accuracy

Datasets

C4 (used for calibration: 128 segments × 2048 tokens)

Benchmarks

raw-WikiText2, PTB, C4 subset, Lambada, ARC, PIQA, StoryCloze