mlsanityData to 7 Oct 2026

One Model, Two Roles: Emergent Specialization in a Shared Recurrent Transformer

arXiv 2605.17811 ↗First posted 18 May 2026cs.LG cs.AI math.OCcited by 0

Not on the map. Why?

Abstract (arXiv)

Can a shared-weight recurrent Transformer develop distinct internal roles without being partitioned into separate modules? We study this in Asymmetric Input Recurrence (AIR), a minimal two-state reasoning architecture in which the same Transformer model is reused for both updates (per literature, L and H) and the only built-in difference in the update rule is that the encoded input is injected during L-updates but not H-updates. Across Sudoku-Extreme and Maze, decoded rollouts reveal a consistent split: zH behaves like a fully committed proposal state, whereas zL retains local uncertainty and shifting intermediate structure. Freeze experiments show that this split is, in practice, related to the model's state dynamics: in Sudoku, freezing zH reduces zL's content changes whereas freezing zL increases zH's, while in Maze, freezing either state increases content changes in the other state. Ablations show that to induce specialization, the shared model needs to be able to tell the two update types apart, either from input injection asymmetry or from a separate level token. Mechanistically, attention analysis shows that L-updates are consistently more local than H-updates in both Sudoku and Maze. Together, these results show that, in a two-state recurrent setting, a clear state-identity signal can induce stable, related functional roles inside a shared-parameter recurrent Transformer. Code is available at hrefhttps://github.com/juchengshen/airtextcolorbluehttps://github.com/juchengshen/air.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 20 of its 25 references that are in the map. First posted on arXiv 18 May 2026.

Builds on

20 of 25 references.

References · 20

  1. 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
  2. 2022In-context Learning and Induction Headscited by 930
  3. 2021Show Your Work: Scratchpads for Intermediate Computation with Language Modelscited by 871
  4. 2024Training Large Language Models to Reason in a Continuous Latent Spacecited by 763
  5. 2018Universal Transformerscited by 613
  6. 2016Adaptive Computation Time for Recurrent Neural Networkscited by 479
  7. 2015Neural Module Networkscited by 375
  8. 2023Looped Transformers as Programmable Computerscited by 224
  9. 2019Recurrent Independent Mechanismscited by 219
  10. 2025Hierarchical Reasoning Modelcited by 167
  11. 2021PonderNet: Learning to Pondercited by 156
  12. 2021Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networkscited by 108
8 more
  1. 2020Are Neural Nets Modular? Inspecting Functional Modularity Through Differentiable Weight Maskscited by 97
  2. 2024Scaling Exponents Across Parameterizations and Optimizerscited by 71
  3. 2024A Mechanistic Analysis of a Transformer Trained on a Symbolic Multi-Step Reasoning Taskcited by 49
  4. 2025Grokking at the Edge of Numerical Stabilitycited by 41
  5. 2026A Mechanistic Analysis of Looped Reasoning Language Modelscited by 27
  6. 2024A Implies B: Circuit Analysis in LLMs for Propositional Logical Reasoningcited by 15
  7. 2025Hierarchical Reasoning Models: Perspectives and Misconceptionscited by 13
  8. 2025Emergence of Functionally Differentiated Structures via Mutual Information Minimization in Recurrent Neural Networkscited by 1

Built on by

Not cited yet.