mlsanityData to 14 Oct 2026

Directional Routing in Transformers

arXiv 2603.14923 ↗First posted 16 Mar 2026cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

We introduce directional routing, a lightweight mechanism that gives each transformer attention head learned suppression directions controlled by a shared router, at 3.9% parameter cost. We train a 433M-parameter model alongside an identical baseline in a single run, then trace the resulting circuits through mechanistic interpretability. Routing becomes the model's dominant computational pathway. Disabling it collapses factual recall to near-zero probability across all 8 test prompts and drops induction accuracy from 93.4% to 0.0%. Knocking out individual attention heads has negligible effect: the primary mover head's removal actually increases target probability, and induction heads retain 98.6% accuracy without their strongest member. The coordination mechanism is irreplaceable; the components it coordinates are not. The model also self-organizes, without explicit pressure, into two regimes: domain-adaptive routing in early layers and fixed syntactic pruning in late layers, where the least-varying layer is the most critical (+42.6 PPL when disabled). Routing reduces perplexity 31-56% relative to the baseline, though downstream multiple-choice benchmarks do not yet reflect these gains.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 24 of its 30 references that are in the map. First posted on arXiv 16 Mar 2026.

Builds on

24 of 30 references.

References · 24

  1. 2017Attention Is All You Needcited by 60,361
  2. 2020Scaling Laws for Neural Language Modelscited by 6,586
  3. 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,886
  4. 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,509
  5. 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,073
  6. 2022Training Compute-Optimal Large Language Modelscited by 3,019
  7. 2022Locating and Editing Factual Associations in GPTcited by 2,691
  8. 2019BoolQ: Exploring the Surprising Difficulty of Natural Yes/No Questionscited by 2,183
  9. 2024Mixtral of Expertscited by 1,823
  10. 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,786
  11. 2019Similarity of Neural Network Representations Revisitedcited by 1,755
  12. 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Shardingcited by 1,631
12 more
  1. 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,420
  2. 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 smallcited by 1,165
  3. 2024The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scalecited by 1,160
  4. 2019What Does BERT Look At? An Analysis of BERT's Attentioncited by 1,003
  5. 2016The LAMBADA dataset: Word prediction requiring a broad discourse contextcited by 992
  6. 2022Toy Models of Superpositioncited by 971
  7. 2022In-context Learning and Induction Headscited by 964
  8. 2021Probing Classifiers: Promises, Shortcomings, and Advancescited by 876
  9. 2024DeepSeekMoE: Towards Ultimate Expert Specialization in Mixture-of-Experts Language Modelscited by 867
  10. 2019Analyzing Multi-Head Self-Attention: Specialized Heads Do the Heavy Lifting, the Rest Can Be Prunedcited by 856
  11. 2019Are Sixteen Heads Really Better than One?cited by 816
  12. 2023Dissecting Recall of Factual Associations in Auto-Regressive Language Modelscited by 528

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Adaptive Computation Depth via Learned Token Routing in Transformers
  2. 2026Sparsity Moves Computation: How FFN Architecture Reshapes Attention in Small Transformers
  3. 2026Grounding latent algorithm routing in transformer reasoning
  4. 2026Align, Unify, Suppress, Route: A Coherentist View of Transformer Computation
  5. 2026Adaptive Loops and Memory in Transformers: Think Harder or Know More?
  6. 2025From Condensation to Rank Collapse: A Two-Stage Analysis of Transformer Training Dynamics
  7. 2026Symmetry Breaking in Transformers for Efficient and Interpretable Training
  8. 2026Sparse Token Routing in Efficient Transformers