Intrinsically Interpretable Attention via Sparse Post-Training
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
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2021On the Opportunities and Risks of Foundation Modelscited by 4,033
- 2023Lost in the Middle: How Language Models Use Long Contextscited by 3,455
- 2020Longformer: The Long-Document Transformercited by 3,413
- 2016Categorical Reparameterization with Gumbel-Softmaxcited by 2,739
- 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,598
- 2020Big Bird: Transformers for Longer Sequencescited by 1,684
- 2019Generating Long Sequences with Sparse Transformerscited by 1,546
- 2023Progress measures for grokking via mechanistic interpretabilitycited by 839
- 2025DeepSeek-V3.2: Pushing the Frontier of Open Large Language Modelscited by 815
- 2023Towards Automated Circuit Discovery for Mechanistic Interpretabilitycited by 724
- 2024OLMo: Accelerating the Science of Language Modelscited by 613
13 more
- 2024Dolma: an Open Corpus of Three Trillion Tokens for Language Model Pretraining Researchcited by 483
- 2024Mechanistic Interpretability for AI Safety -- A Reviewcited by 422
- 2023Towards Best Practices of Activation Patching in Language Models: Metrics and Methodscited by 359
- 2024ArCHer: Training Language Model Agents via Hierarchical Multi-Turn RLcited by 228
- 2023Landmark Attention: Random-Access Infinite Context Length for Transformerscited by 213
- 2023Attribution Patching Outperforms Automated Circuit Discoverycited by 209
- 2024Transcoders Find Interpretable LLM Feature Circuitscited by 200
- 2021Memory-efficient Transformers via Top-k Attentioncited by 60
- 2025Weight-sparse transformers have interpretable circuitscited by 44
- 2024Unveiling the Secret Recipe: A Guide For Supervised Fine-Tuning Small LLMscited by 40
- 2025Mixture of Experts Made Intrinsically Interpretablecited by 24
- 2024SPARTAN: A Sparse Transformer World Model Attending to What Matterscited by 10
- 2025Tracing Multilingual Representations in LLMs with Cross-Layer Transcoderscited by 8
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026From Observation to Insight: Mechanistic World Models and the Quest for Autonomous Discoverycited by 4
- 2026NeuroGame Transformer: Gibbs-Inspired Attention Driven by Game Theory and Statistical Physicscited by 2
- 2026SOLAR: SVD-Optimized Lifelong Attention for Recommendationcited by 0
- 2026Circuit Condensation: Post-Training that Concentrates a Behavior's Causal Circuitcited by 0
- 2026Circuit Condensation: Post-Training that Concentrates a Behavior's Causal Circuit27 Aug 2026
- 2026From Observation to Insight: Mechanistic World Models and the Quest for Autonomous Discovery14 Jul 2026
- 2026NeuroGame Transformer: Gibbs-Inspired Attention Driven by Game Theory and Statistical Physics19 Mar 2026
- 2026SOLAR: SVD-Optimized Lifelong Attention for Recommendation3 Mar 2026
Similar, not cited
Close in wording, no citation either way
- 2021Predicting Attention Sparsity in Transformers
- 2026SpargeAttention2: Trainable Sparse Attention via Hybrid Top-k+Top-p Masking and Distillation Fine-Tuning
- 2025Towards Understanding the Nature of Attention with Low-Rank Sparse Decomposition
- 2026Sparse but not Simpler: A Multi-Level Interpretability Analysis of Vision Transformers
- 2024Interpreting Attention Layer Outputs with Sparse Autoencoders
- 2026Routing Absorption in Sparse Attention: Why Random Gates Are Hard to Beat
- 2023Rethinking Attention: Exploring Shallow Feed-Forward Neural Networks as an Alternative to Attention Layers in Transformers
- 2026Sparse Transformer: a Controlled Testbed for Mechanistic Interpretability