atlas

Research map / Architectures and efficiency / Pruning and efficient tuning

SlimGPT: Layer-wise Structured Pruning for Large Language Models

NeurIPS 2024 · Architectures and efficiency › Pruning and efficient tuning · approach: Layer-wise pruning of LLMs · 12 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.

Proposes SlimGPT, a layer-wise structured pruning method for LLMs that extends Optimal Brain Surgeon with batched greedy pruning (grouped Cholesky) and a logarithmic Incremental Pruning Ratio schedule, evaluated on LLaMA-family models.

Problem

Structured pruning of LLMs under limited compute is hard: performance degrades strongly at high pruning ratios and post-training recovery is expensive; OBS-style layer-wise methods were not previously applied to structured pruning of LLMs.

Approach

Extends OBS to column/head-level structured pruning with grouped Cholesky decomposition for fast head-wise error estimation, dynamic group size for FFN channel pruning, and a logarithmically increasing per-layer pruning ratio to counter error accumulation in layer-wise pruning.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add logarithmically increasing layer-wise pruning ratios instead of uniform ratios (ablation)SlimGPT · LLaMA-7B pruned 50%, without post-trainingPPL: 123.05 -> 38.83Helped
Add logarithmically increasing layer-wise pruning ratios instead of uniform ratios (ablation)SlimGPT · LLaMA-7B pruned 50%, without post-trainingzero-shot average: 44.34 -> 52.23Helped
Use logarithmic rather than linear increasing pruning ratios (ablation)SlimGPT · LLaMA-7B pruned 50%, without post-trainingPPL: 46.57 -> 38.83Helped
Use logarithmic rather than linear increasing pruning ratios (ablation)SlimGPT · LLaMA-7B pruned 50%, without post-trainingzero-shot average: 53.45 -> 52.23Hurt
Add Dynamic Group Size instead of fixing FFN group size at 128 (ablation)SlimGPT · LLaMA-7B pruned 50%, without post-trainingzero-shot average: 51.63 -> 52.23Helped
Add grouped Cholesky decomposition for attention head error estimation (ablation)SlimGPT · LLaMA-7B pruned 50%, without post-trainingPPL: 54.94 -> 38.83Helped
Add LoRA post-training (ablation)SlimGPT · LLaMA-7B pruned 50%zero-shot average: 52.23 -> 53.76Helped
Add LoRA post-training (ablation)SlimGPT · LLaMA-7B pruned 50%BoolQHurt
Use C4 rather than Alpaca calibration data (ablation)SlimGPT · LLaMA-7B pruned 50%, without post-trainingPPL: 48.26 -> 42.06Helped
Use Alpaca rather than C4 calibration data (ablation)SlimGPT · LLaMA-7B pruned 50%, without post-trainingzero-shot average: 54.44 -> 52.70Hurt
Prune with SlimGPT rather than LoRAPruneLLaMA-7B · WikiText2 validation, 20% pruningPPL: 16.80 -> 16.68Helped
Prune with SlimGPT rather than LLM-PrunerLLaMA-7B · Commonsense Reasoning zero-shot average, 20% pruningaverage score: 61.50 -> 65.07Helped

Surprising findings

Limitations

Experimental setup

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

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

Metrics: perplexity zero-shot accuracy average accuracy latency memory usage

Baselines: LLM-Pruner Compresso LoraPrune

Builds on: Optimal Brain Surgeon OBC GPTQ SparseGPT LLM-Pruner

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