Symmetry Reveals Layerwise Dynamics: How Transformers Perform In-Context Classification
Not on the map. Why?
Abstract (arXiv)
Transformers can perform in-context classification from a few labeled examples, yet the inference-time algorithm remains opaque. We study multi-class linear classification in the hard no-margin regime and make the computation identifiable by enforcing feature- and label-permutation equivariance at every layer. This enables interpretability while maintaining functional equivalence and yields highly structured weights. From these models we extract an explicit depth-indexed recursion: an end-to-end identified, emergent update rule inside a softmax transformer, to our knowledge the first of its kind. Attention matrices formed from mixed feature-label Gram structure drive coupled updates of training points, labels, and the test probe. The resulting dynamics implement a geometry-driven algorithmic motif, which can provably amplify class separation and yields robust expected class alignment.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 41 of its 55 references that are in the map. First posted on arXiv 13 Apr 2026.
Builds on
41 of 55 references.
References · 41
- 2017Attention Is All You Needcited by 59,932
- 2017Prototypical Networks for Few-shot Learningcited by 3,001
- 2017Deep Setscited by 1,394
- 2016Group Equivariant Convolutional Networkscited by 1,154
- 2021E(n) Equivariant Graph Neural Networkscited by 951
- 2022In-context Learning and Induction Headscited by 930
- 2021An Explanation of In-context Learning as Implicit Bayesian Inferencecited by 915
- 2022Transformers learn in-context by gradient descentcited by 714
- 2022What Can Transformers Learn In-Context? A Case Study of Simple Function Classescited by 700
- 2022What learning algorithm is in-context learning? Investigations with linear modelscited by 671
- 2018Invariant and Equivariant Graph Networkscited by 339
- 2023Trained Transformers Learn Linear Models In-Contextcited by 331
29 more
- 2022Data Distributional Properties Drive Emergent In-Context Learning in Transformerscited by 303
- 2023In-Context Learning Creates Task Vectorscited by 302
- 2023Function Vectors in Large Language Modelscited by 292
- 2023Transformers learn to implement preconditioned gradient descent for in-context learningcited by 274
- 2023Transformers as Algorithms: Generalization and Stability in In-context Learningcited by 220
- 2021Thinking Like Transformerscited by 182
- 2023The emergence of clusters in self-attention dynamicscited by 133
- 2023The Learnability of In-Context Learningcited by 126
- 2023The mechanistic basis of data dependence and abrupt learning in an in-context classification taskcited by 115
- 2023What and How does In-Context Learning Learn? Bayesian Model Averaging, Parameterization, and Generalizationcited by 112
- 2023Tracr: Compiled Transformers as a Laboratory for Interpretabilitycited by 99
- 2021Graph-based Semi-supervised Learning: A Comprehensive Reviewcited by 79
- 2022On the Symmetries of Deep Learning Models and their Internal Representationscited by 63
- 2023Pretraining Data Mixtures Enable Narrow Model Selection Capabilities in Transformer Modelscited by 45
- 2023The Mystery of In-Context Learning: A Comprehensive Survey on Interpretation and Analysiscited by 34
- 2023A Closer Look at In-Context Learning under Distribution Shiftscited by 25
- 2025The Mean-Field Dynamics of Transformerscited by 24
- 2023Understanding Emergent In-Context Learning from a Kernel Regression Perspectivecited by 23
- 2023Permutation Equivariance of Transformers and Its Applicationscited by 20
- 2023In-context Learning and Gradient Descent Revisitedcited by 20
- 2024Exploring the Robustness of In-Context Learning with Noisy Labelscited by 13
- 2025Contextualize-then-Aggregate: Circuits for In-Context Learning in Gemma-2 2Bcited by 10
- 2025When and How Unlabeled Data Provably Improve In-Context Learningcited by 5
- 2026Dissecting Multimodal In-Context Learning: Modality Asymmetries and Circuit Dynamics in modern Transformerscited by 5
- 2026Discovering Interpretable Algorithms by Decompiling Transformers to RASPcited by 4
- 2026Unlabeled Data Can Provably Enhance In-Context Learning of Transformerscited by 2
- 2025Linear Transformers Implicitly Discover Unified Numerical Algorithmscited by 1
- 2025Optimal Attention Temperature Improves the Robustness of In-Context Learning under Distribution Shift in High Dimensionscited by 1
- 2025In-Context Semi-Supervised Learningcited by 1
Built on by
Not cited yet.