mlsanityData to 7 Oct 2026

Engineering Verifiable Modularity in Transformers via Per-Layer Supervision

arXiv 2603.18029 ↗First posted 8 Mar 2026cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Transformers resist surgical control. Ablating an attention head identified as critical for capitalization produces minimal behavioral change because distributed redundancy compensates for damage. This Hydra effect renders interpretability illusory: we may identify components through correlation, but cannot predict or control their causal role. We demonstrate that architectural interventions can expose hidden modularity. Our approach combines dual-stream processing separating token and contextual representations, per-layer supervision providing independent gradient signal at each depth, and gated attention regularizing toward discrete activation patterns. When trained with per-layer supervision, models produce ablation effects 5 to 23 times larger than architecturally identical controls trained with standard objectives. This enables 4 times greater control leverage on targeted behaviors: scaling identified attention heads produces smooth, predictable changes in model output. The key finding is architectural. Without per-layer supervision, ablation damage concentrates near zero with low variance (Winograd standard deviation 0.63%). With per-layer supervision, effects spread widely (standard deviation 6.32%), revealing which predictions depend on which circuits. The larger variance is not measurement noise but the signature of unmasked modularity. We validate our approach through three components: engineered features that capture computational dynamics rather than vocabulary structure (validated by near-zero correlation with raw activation clustering), an architecture providing positive control for modularity, and causal experiments demonstrating functional reorganization where different tasks route through different attention heads. This es tablishes a methodology for transforming interpretability from passive observation to active control.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 15 of its 19 references that are in the map. First posted on arXiv 8 Mar 2026.

Builds on

15 of 19 references.

References · 15

  1. 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,413
  2. 2020DeBERTa: Decoding-enhanced BERT with Disentangled Attentioncited by 2,157
  3. 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 smallcited by 1,130
  4. 2019What Does BERT Look At? An Analysis of BERT's Attentioncited by 994
  5. 2022In-context Learning and Induction Headscited by 930
  6. 2019Attention is not Explanationcited by 916
  7. 2019Analyzing Multi-Head Self-Attention: Specialized Heads Do the Heavy Lifting, the Rest Can Be Prunedcited by 844
  8. 2021Probing Classifiers: Promises, Shortcomings, and Advancescited by 831
  9. 2019Are Sixteen Heads Really Better than One?cited by 803
  10. 2021Causal Abstractions of Neural Networkscited by 401
  11. 2025Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink-Freecited by 266
  12. 2020Gradient Starvation: A Learning Proclivity in Neural Networkscited by 262
3 more
  1. 2023The Hydra Effect: Emergent Self-repair in Language Model Computationscited by 145
  2. 2021Inducing Causal Structure for Interpretable Neural Networkscited by 107
  3. 2023Seeing is Believing: Brain-Inspired Modular Training for Mechanistic Interpretabilitycited by 38

Built on by

Not cited yet.