atlas

Research map / Architectures and efficiency / Pruning and efficient tuning

Junk DNA Hypothesis: Pruning Small Pre-Trained Weights $\textit{Irreversibly}$ and $\textit{Monotonically}$ Impairs ``Difficult" Downstream Tasks in LLMs

ICML 2024 · Architectures and efficiency › Pruning and efficient tuning · approach: Layer-wise pruning of LLMs

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 the Junk DNA Hypothesis: magnitude-pruning small pre-trained LLM weights monotonically and irreversibly impairs harder downstream tasks, unlike quantization.

Problem

Whether small-magnitude pre-trained weights in LLMs are truly redundant, and how their removal affects downstream tasks of varying difficulty.

Approach

Defines task difficulty along controllable axes (data adequacy, option count, context length, k-shot, human-LLM gap, factoid vs MCQ) and shows that magnitude pruning degrades harder tasks monotonically (quantized models do not), with Theil-Sen slopes and Spearman correlation quantifying the trend; fine-tuning cannot recover the lost knowledge.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Prune increasing fractions of small-magnitude pre-trained weightsVicuna-7B · TriviaQA and MMLU difficulty spectraDownstream performance degradation across task difficulty: 0.996 vs 0.799; 0.979 vs 0.872Mixed
Replace pruning with GPTQ quantizationVicuna-7B · TriviaQA and MMLU difficulty spectraSpearman rank correlation: 0.996 vs 0.799; 0.979 vs 0.872Hurt
Use uniform SVD low-rank compression instead of pruningVicuna-7B · TriviaQA and MMLU difficulty spectraSpearman rank correlation: 0.899; 0.942; 0.905; 0.912; 0.933; 0.956; 0.956Helped
Prune small-magnitude weights before downstream fine-tuningSparse Transfer · Difficult downstream tasksTask performanceHurt
Allow pruned weights to regain non-zero values during fine-tuning (ablation)Sparse to Dense Transfer · Difficult downstream tasksTask performanceHurt
Freeze pre-trained small-magnitude weights during fine-tuning (ablation)Dense Transfer · SST-2, MNLI, and QNLITask performance: 90%No clear effect
Freeze pre-trained small-magnitude weights during fine-tuning (ablation)Dense Transfer · COLA, CSQA, and WinoGrandeTask performance: 70%No clear effect
Use SparseGPT or Wanda weight-importance pruning (ablation)Vicuna-7B · Downstream tasks across difficulty levelsPerformance gap between easy and challenging tasksMixed
Use structured N:M pruning instead of unstructured pruning (ablation)RoBERTa-Base and Vicuna models · Downstream tasks across difficulty levelsTask performance degradation with difficultyNo clear effect
Increase the amount of magnitude pruningPre-trained LLMs · Downstream tasks across difficulty levelsTask performanceHurt
Prune up to 14% of small weightsVicuna-7B · MMLU with 5 ICL demonstrationsAccuracy: 14%No clear effect
Prune around 10-15% of small weightsModels in the four within-task settings · Easier downstream tasksTask performance: 10-15%No clear effect

Surprising findings

Limitations

Experimental setup

Datasets: GLUE MMLU TriviaQA PIQA HellaSwag OpenBookQA FreebaseQA CSQA WinoGrande

Benchmarks: GLUE MMLU TriviaQA PIQA HellaSwag OpenBookQA FreebaseQA CSQA WinoGrande

Metrics: normalized performance drop Spearman rank correlation Theil-Sen slope accuracy

Baselines: SparseGPT Wanda GPTQ Magnitude pruning Sparse Transfer

Builds on: Lottery Ticket Hypothesis SparseGPT Wanda

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