vix.ing · top · new · best · stats · spec

SparseGPT: Massive Language Models Can Be Accurately Pruned in One-Shot

2023/01/02 by Elias Frantar, Dan Alistarh, Frantar, Elias +1 · 3 voices · 170 citations
Computer Science · Engineering · #Advanced Data Storage Technologies #Ferroelectric and Negative Capacitance Devices #Topic Modeling #cs.LG

paper · pdf · doi:10.48550/arxiv.2301.00774

openalex publication_date 2023/01/02 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/30

Abstract

We show for the first time that large-scale generative pretrained transformer (GPT) family models can be pruned to at least 50% sparsity in one-shot, without any retraining, at minimal loss of accuracy. This is achieved via a new pruning method called SparseGPT, specifically designed to work efficiently and accurately on massive GPT-family models. We can execute SparseGPT on the largest available open-source models, OPT-175B and BLOOM-176B, in under 4.5 hours, and can reach 60% unstructured sparsity with negligible increase in perplexity: remarkably, more than 100 billion weights from these models can be ignored at inference time. SparseGPT generalizes to semi-structured (2:4 and 4:8) patterns, and is compatible with weight quantization approaches. The code is available at: https://github.com/IST-DASLab/sparsegpt.

Cited by

Discussions

Related