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