Don't Go Breaking My LLM: The Impact of Pruning Attention Layers on Explanation Faithfulness and Confidence Calibration

14d ago · Global · primary source: export.arxiv.org

Pruning attention layers in large language models can erode their interpretability and reliability even when accuracy holds steady, according to a new study that examined five models across eight datasets [1]. The research, posted to arXiv on 23 June 2026, investigates how removing attention layers — the most computationally expensive components of transformer-based models — affects explanation faithfulness and confidence calibration [1]. Prior work established that up to 33% of attention layers can be pruned with minimal accuracy loss, making the technique a cost-effective compression strategy [1][2]. However, the authors note that the downstream effects on interpretability had not been systematically studied [2]. Across the five large language models and eight datasets tested, pruned models frequently maintained high accuracy scores [1]. Yet the study found that faithfulness and calibration often degraded, and the two metrics could fluctuate significantly even when accuracy remained stable [2]. This misalignment suggests that standard accuracy benchmarks alone may not capture the full picture of a compressed model's reliability [1]. The paper recommends that evaluation protocols for pruned models incorporate explainability and calibration metrics alongside traditional accuracy and efficiency measures [1][2]. The findings arrive as the machine-learning community grapples with broader questions about model compression and transfer learning. Recent work in other domains, such as catalyst informatics, has explored how models trained on one dataset can aid performance on another through transfer learning or joint training, underscoring the importance of understanding what information is preserved or lost when models are modified [4]. While the arXiv study focuses on language models, its implications extend to any field where compressed neural networks are deployed in high-stakes settings. The authors caution that pruning attention layers can affect interpretability and reliability in ways that accuracy and efficiency measures alone do not reveal [2].

safety-researchresearch-paperinfrastructure

Background sources we checked (6)
  • arxiv.org ↗ Pruning Large Language Models (LLMs) reduces memory and inference costs by removing parts of the network, producing smaller models that retain most of their accuracy. As attention layers are the most resource-intensive parts of LLMs, pruning them is a promising compression strate…
  • arxiv.org ↗ CatalyzeX Code Finder for Papers (What is CatalyzeX?) ... DagsHub Toggle ... DagsHub (What is DagsHub?)…
  • arxiv.org ↗ With the creation of new datasets, the question arises of whether the data in them is complementary to other datasets for training ML models (see recent reviews for a perspective of catalysts informatics22, 23, 24). This is especially important when consolidating data with a vari…
  • arxiv.org ↗ CatalyzeX Code Finder for Papers (What is CatalyzeX?) ... DagsHub Toggle ... DagsHub (What is DagsHub?)…
  • en.wikipedia.org ↗ Sustainable Development Goals (abbr. SDGs) were adopted in 2015 by all United Nations (UN) members for the 2030 Agenda for Sustainable Development. The aim of the 17 global goals is "peace and prosperity for people and the planet", tackling climate change, and working to preserv…
  • en.wikipedia.org ↗ In molecular biology, a transcription factor (TF) (or sequence-specific DNA-binding factor) is a protein that controls the rate of transcription of genetic information from DNA to messenger RNA, by binding to DNA sequences. Specificity can be due to sequence motifs, or epigenetic…

Sources

Spot something wrong? Report an issue