Intrinsically Interpretable Attention via Sparse Post-Training
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 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.
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}
}