English

Data-Free Pruning of Self-Attention Layers in LLMs

Machine Learning 2025-12-25 v1 Artificial Intelligence

Abstract

Many self-attention sublayers in large language models (LLMs) can be removed with little to no loss. We attribute this to the Attention Suppression Hypothesis: during pre-training, some deep attention layers learn to mute their own contribution, leaving the residual stream and the MLP to carry the representation. We propose Gate-Norm, a one-shot, weight-only criterion that ranks attention sublayers by query--key coupling and removes the least coupled ones, requiring no calibration data, no forward passes, no fine-tuning, and no specialized kernels. On 40-layer, 13B-parameter LLaMA models, Gate-Norm prunes the model in under a second. Pruning 88--1616 attention sublayers yields up to 1.30×1.30\times higher inference throughput while keeping average zero-shot accuracy within 2%2\% of the unpruned baseline across BoolQ, RTE, HellaSwag, WinoGrande, ARC-Easy/Challenge, and OpenBookQA. Across these settings, Gate-Norm matches data-driven pruning methods in accuracy while being 1000×\sim 1000\times faster to score layers, enabling practical, data-free compression of LLMs.

Keywords

Cite

@article{arxiv.2512.20636,
  title  = {Data-Free Pruning of Self-Attention Layers in LLMs},
  author = {Dhananjay Saikumar and Blesson Varghese},
  journal= {arXiv preprint arXiv:2512.20636},
  year   = {2025}
}
R2 v1 2026-07-01T08:39:02.781Z