mlsanityData to 14 Oct 2026

Intrinsically Interpretable Attention via Sparse Post-Training

arXiv 2512.05865 ↗First posted 5 Dec 2025cs.LG cs.AIcited by 4

Not on the map. Why?

Abstract (arXiv)

We introduce a simple post-training method that makes transformer attention sparse without sacrificing performance. Applying a flexible sparsity regularisation under a constrained-loss objective, we show on models up to 7B parameters that it is possible to retain the original pretraining loss while reducing attention connectivity to ≈ 0.4 % of its edges. Unlike sparse-attention methods designed for computational efficiency, our approach leverages sparsity as a structural prior: it preserves capability while exposing a more organized and interpretable connectivity pattern. We find that this local sparsity cascades into global circuit simplification: task-specific circuits involve far fewer components (attention heads and MLPs) with up to 100x fewer edges connecting them. Additionally, using cross-layer transcoders, we show that sparse attention substantially simplifies attention attribution, enabling a unified view of feature-based and circuit-based perspectives. These results demonstrate that transformer attention can be made orders of magnitude sparser, suggesting that much of its computation is redundant and that sparsity may serve as a guiding principle for more structured and interpretable models.

Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Builds on 25 of its 37 references that are in the map. First posted on arXiv 5 Dec 2025.

Builds on

25 of 37 references.

References · 25

  1. 2022Training language models to follow instructions with human feedbackcited by 15,971
  2. 2021On the Opportunities and Risks of Foundation Modelscited by 4,033
  3. 2023Lost in the Middle: How Language Models Use Long Contextscited by 3,455
  4. 2020Longformer: The Long-Document Transformercited by 3,413
  5. 2016Categorical Reparameterization with Gumbel-Softmaxcited by 2,739
  6. 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,598
  7. 2020Big Bird: Transformers for Longer Sequencescited by 1,684
  8. 2019Generating Long Sequences with Sparse Transformerscited by 1,546
  9. 2023Progress measures for grokking via mechanistic interpretabilitycited by 839
  10. 2025DeepSeek-V3.2: Pushing the Frontier of Open Large Language Modelscited by 815
  11. 2023Towards Automated Circuit Discovery for Mechanistic Interpretabilitycited by 724
  12. 2024OLMo: Accelerating the Science of Language Modelscited by 613
13 more
  1. 2024Dolma: an Open Corpus of Three Trillion Tokens for Language Model Pretraining Researchcited by 483
  2. 2024Mechanistic Interpretability for AI Safety -- A Reviewcited by 422
  3. 2023Towards Best Practices of Activation Patching in Language Models: Metrics and Methodscited by 359
  4. 2024ArCHer: Training Language Model Agents via Hierarchical Multi-Turn RLcited by 228
  5. 2023Landmark Attention: Random-Access Infinite Context Length for Transformerscited by 213
  6. 2023Attribution Patching Outperforms Automated Circuit Discoverycited by 209
  7. 2024Transcoders Find Interpretable LLM Feature Circuitscited by 200
  8. 2021Memory-efficient Transformers via Top-k Attentioncited by 60
  9. 2025Weight-sparse transformers have interpretable circuitscited by 44
  10. 2024Unveiling the Secret Recipe: A Guide For Supervised Fine-Tuning Small LLMscited by 40
  11. 2025Mixture of Experts Made Intrinsically Interpretablecited by 24
  12. 2024SPARTAN: A Sparse Transformer World Model Attending to What Matterscited by 10
  13. 2025Tracing Multilingual Representations in LLMs with Cross-Layer Transcoderscited by 8

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 4.

4 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2021Predicting Attention Sparsity in Transformers
  2. 2026SpargeAttention2: Trainable Sparse Attention via Hybrid Top-k+Top-p Masking and Distillation Fine-Tuning
  3. 2025Towards Understanding the Nature of Attention with Low-Rank Sparse Decomposition
  4. 2026Sparse but not Simpler: A Multi-Level Interpretability Analysis of Vision Transformers
  5. 2024Interpreting Attention Layer Outputs with Sparse Autoencoders
  6. 2026Routing Absorption in Sparse Attention: Why Random Gates Are Hard to Beat
  7. 2023Rethinking Attention: Exploring Shallow Feed-Forward Neural Networks as an Alternative to Attention Layers in Transformers
  8. 2026Sparse Transformer: a Controlled Testbed for Mechanistic Interpretability