atlas

Research map / Large language models / Reasoning and chain of thought

SLEB: Streamlining LLMs through Redundancy Verification and Elimination of Transformer Blocks

ICML 2024 · Large language models › Reasoning and chain of thought · approach: Efficient LLM serving · 1 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 SLEB, a training-free method that iteratively removes redundant transformer blocks in LLMs using a calibration-based perplexity metric, yielding real end-to-end inference speedups.

Problem

Pruning LLMs with unstructured, 2:4, or channel-wise methods rarely gives end-to-end speedup, and early-exit or layer-skipping approaches need dynamic decisions or training. The paper targets training-free, static, hardware-agnostic compression that yields speedup proportional to the pruning ratio.

Approach

Treats whole transformer blocks as the pruning unit, exploits high cosine similarity between neighboring block outputs, and greedily removes the block whose removal least increases calibration perplexity, recomputing scores after each removal.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Remove 10% of transformer blocks with SLEBLLaMA-2-70B · C4 validationperplexity: 5.71 -> 6.32Hurt
Remove 20% of transformer blocks with SLEBLLaMA-2-70B · C4 validationperplexity: 5.71 -> 7.31Hurt
Remove 10% of transformer blocks with SLEBLLaMA-2-70B · Zero-shot task averageaccuracy: 76.57 -> 73.14Hurt
Remove 20% of transformer blocks with SLEBLLaMA-2-70B · Zero-shot task averageaccuracy: 76.57 -> 70.81Hurt
Remove 10% of transformer blocks with SLEBLLaMA-2-70B · Prompt processing, A100latency: 1718.4 -> 1529.1Helped
Remove 20% of transformer blocks with SLEBLLaMA-2-70B · Prompt processing, A100latency: 1718.4 -> 1364.1Helped
Remove 10% of transformer blocks with SLEBLLaMA-2-70B · Token generation, A100throughput: 299 -> 336Helped
Remove 20% of transformer blocks with SLEBLLaMA-2-70B · Token generation, A100throughput: 299 -> 381Helped
Use iterative network-level block assessment (Metric³) instead of Metric¹ or Metric² to select blocks (ablation)SLEB · OPT-6.7B and OPT-13B, WikiText-2perplexityHelped
Select blocks with Metric¹ based on input-output distance (ablation)SLEB · OPT-6.7B and OPT-13B, WikiText-2perplexityHurt
Use Metric² without recalculating block significance after each removal (ablation)SLEB · OPT-6.7B, WikiText-2token prediction qualityHurt
Increase SLEB block-removal sparsity from 10% to 20% (ablation)OPT and LLaMA-2 models · C4 and zero-shot tasksperformanceHurt

Surprising findings

Limitations

Experimental setup

Datasets: WikiText-2 C4 PIQA WinoGrande HellaSwag ARC-easy ARC-challenge Alpaca

Benchmarks: WikiText-2 C4 PIQA WinoGrande HellaSwag ARC-easy ARC-challenge

Metrics: perplexity zero-shot accuracy throughput (tokens/s) latency (ms) speedup

Baselines: SparseGPT Wanda DSnoT LLM-Pruner SliceGPT

Builds on: SparseGPT Wanda SliceGPT LLM-Pruner early exit

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