English

Intrinsically Interpretable Attention via Sparse Post-Training

Machine Learning 2026-05-26 v5 Artificial Intelligence

Abstract

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%\approx 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.

Keywords

Cite

@article{arxiv.2512.05865,
  title  = {Intrinsically Interpretable Attention via Sparse Post-Training},
  author = {Florent Draye and Anson Lei and Hsiao-Ru Pan and Ingmar Posner and Bernhard Schölkopf},
  journal= {arXiv preprint arXiv:2512.05865},
  year   = {2026}
}
R2 v1 2026-07-01T08:11:52.965Z