Tracking States or Tracking Cosets? An Algebraic Account of Learned State Tracking
Not on the map. Why?
Abstract (arXiv)
State tracking requires composing a sequence of updates, but accuracy alone does not reveal what a model has learned. We study neural networks trained to predict the running product of group elements. We identify quotient solutions in Transformers, where models recover the quotient class while predicting nearly uniformly among its members. The reciprocal of class size predicts partial accuracy without a fitted parameter, extending parity-based accounts to non-parity quotients. Our baseline Transformers' predictions change little under prefix reordering beyond the exact-tracking frontier. We prove that, for finite groups under uniform i.i.d. full-group inputs, optimal order-blind exact accuracy converges to the reciprocal of abelianization class size as prefix length grows, consistent with the observed abelianization plateaus. Sequential updates permit more: any partition into right cosets of a subgroup, normal or not, survives sequential updates. In our census of standard Transformers, every recovered coset partition comes from a normal subgroup, whereas parameter-matched recurrent networks pass through both normal and non-normal right-coset stages during training. On A5, we identify low-dimensional subspaces of the recurrent state that encode non-normal cosets. In the three-dimensional cases, coset mean vectors form approximate dodecahedra, and swapping the state components in these subspaces transfers the donor's coset state through a shared input suffix. Our results connect partial accuracy, learning stages, and internal computation through the subgroup cosets that models learn to track.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 21 of its 33 references that are in the map. First posted on arXiv 24 Sep 2026.
Builds on
21 of 33 references.
References · 21
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
- 2017Decoupled Weight Decay Regularizationcited by 1,831
- 2021Show Your Work: Scratchpads for Intermediate Computation with Language Modelscited by 886
- 2023Progress measures for grokking via mechanistic interpretabilitycited by 839
- 2022GPT-NeoX-20B: An Open-Source Autoregressive Language Modelcited by 713
- 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 629
- 2022Transformers Learn Shortcuts to Automatacited by 268
- 2022Neural Networks and the Chomsky Hierarchycited by 217
- 2024The Illusion of State in State-Space Modelscited by 187
- 2023A Toy Model of Universality: Reverse Engineering How Networks Learn Group Operationscited by 144
- 2025A Little Depth Goes a Long Way: The Expressive Power of Log-Depth Transformerscited by 53
- 2025(How) Do Language Models Track State?cited by 34
9 more
- 2023Grokking Group Multiplication with Cosetscited by 24
- 2025Alternating Gradient Flows: A Theory of Feature Learning in Two-layer Neural Networkscited by 14
- 2024Towards a unified and verified understanding of group-operation networkscited by 10
- 2026Sequential Group Composition: A Window into the Mechanics of Deep Learningcited by 10
- 2025Learning words in groups: fusion algebras, tensor ranks and grokkingcited by 4
- 2026Rethinking State Tracking in Recurrent Models Through Error Control Dynamicscited by 4
- 2026A Held-Out Transition-Pair Falsifier for Long-Horizon Non-Abelian State Trackingcited by 3
- 2026Recirculationcited by 2
- 2026Marginals Before Conditionalscited by 1
Built on by
Not cited yet.