Paper summary: The Unreasonable Ineffectiveness of the Deeper Layers
Removing up to about half of the layers of some open LLMs, chosen by layer similarity and followed by a little QLoRA "healing", barely hurts question-answering performance.
Problem
How is knowledge stored across an LLM’s layers? Pruning can test this: if removing a layer doesn’t change performance on knowledge-heavy QA benchmarks, that layer isn’t needed to store the knowledge those questions use.
Method
- Measure similarity between the representations at different depths, and find the contiguous block of layers whose removal changes the representation least.
- Remove that block.
- “Heal” the model with a small amount of parameter-efficient fine-tuning (QLoRA). Each experiment fits on a single 40GB A100.
Key results
For some common open-weight models, performance on QA benchmarks shows minimal degradation until a large fraction, up to half, of the layers is removed. The authors give two possible readings: current pre-training doesn’t make good use of deeper layers, or shallow layers play a critical role in storing knowledge.
Limitations
There’s no dedicated limitations section in the extracted text. The finding is measured on question-answering benchmarks. Other abilities, such as reasoning or long-form generation, can depend on different layers.
Sources
- The Unreasonable Ineffectiveness of the Deeper Layers (arXiv:2403.17887) — arXiv
- Hugging Face Daily Papers, 2024-03-27 — Hugging Face
- The Unreasonable Ineffectiveness of the Deeper Layers (arXiv HTML full text) — arXiv
This is a summary written from the linked public sources. Figures and numbers are as reported there.