mlsanityData to 7 Oct 2026

Symmetry Reveals Layerwise Dynamics: How Transformers Perform In-Context Classification

arXiv 2604.11613 ↗First posted 13 Apr 2026cs.LG cs.AIcited by 0

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

  1. 2017Attention Is All You Needcited by 59,932
  2. 2017Prototypical Networks for Few-shot Learningcited by 3,001
  3. 2017Deep Setscited by 1,394
  4. 2016Group Equivariant Convolutional Networkscited by 1,154
  5. 2021E(n) Equivariant Graph Neural Networkscited by 951
  6. 2022In-context Learning and Induction Headscited by 930
  7. 2021An Explanation of In-context Learning as Implicit Bayesian Inferencecited by 915
  8. 2022Transformers learn in-context by gradient descentcited by 714
  9. 2022What Can Transformers Learn In-Context? A Case Study of Simple Function Classescited by 700
  10. 2022What learning algorithm is in-context learning? Investigations with linear modelscited by 671
  11. 2018Invariant and Equivariant Graph Networkscited by 339
  12. 2023Trained Transformers Learn Linear Models In-Contextcited by 331
29 more
  1. 2022Data Distributional Properties Drive Emergent In-Context Learning in Transformerscited by 303
  2. 2023In-Context Learning Creates Task Vectorscited by 302
  3. 2023Function Vectors in Large Language Modelscited by 292
  4. 2023Transformers learn to implement preconditioned gradient descent for in-context learningcited by 274
  5. 2023Transformers as Algorithms: Generalization and Stability in In-context Learningcited by 220
  6. 2021Thinking Like Transformerscited by 182
  7. 2023The emergence of clusters in self-attention dynamicscited by 133
  8. 2023The Learnability of In-Context Learningcited by 126
  9. 2023The mechanistic basis of data dependence and abrupt learning in an in-context classification taskcited by 115
  10. 2023What and How does In-Context Learning Learn? Bayesian Model Averaging, Parameterization, and Generalizationcited by 112
  11. 2023Tracr: Compiled Transformers as a Laboratory for Interpretabilitycited by 99
  12. 2021Graph-based Semi-supervised Learning: A Comprehensive Reviewcited by 79
  13. 2022On the Symmetries of Deep Learning Models and their Internal Representationscited by 63
  14. 2023Pretraining Data Mixtures Enable Narrow Model Selection Capabilities in Transformer Modelscited by 45
  15. 2023The Mystery of In-Context Learning: A Comprehensive Survey on Interpretation and Analysiscited by 34
  16. 2023A Closer Look at In-Context Learning under Distribution Shiftscited by 25
  17. 2025The Mean-Field Dynamics of Transformerscited by 24
  18. 2023Understanding Emergent In-Context Learning from a Kernel Regression Perspectivecited by 23
  19. 2023Permutation Equivariance of Transformers and Its Applicationscited by 20
  20. 2023In-context Learning and Gradient Descent Revisitedcited by 20
  21. 2024Exploring the Robustness of In-Context Learning with Noisy Labelscited by 13
  22. 2025Contextualize-then-Aggregate: Circuits for In-Context Learning in Gemma-2 2Bcited by 10
  23. 2025When and How Unlabeled Data Provably Improve In-Context Learningcited by 5
  24. 2026Dissecting Multimodal In-Context Learning: Modality Asymmetries and Circuit Dynamics in modern Transformerscited by 5
  25. 2026Discovering Interpretable Algorithms by Decompiling Transformers to RASPcited by 4
  26. 2026Unlabeled Data Can Provably Enhance In-Context Learning of Transformerscited by 2
  27. 2025Linear Transformers Implicitly Discover Unified Numerical Algorithmscited by 1
  28. 2025Optimal Attention Temperature Improves the Robustness of In-Context Learning under Distribution Shift in High Dimensionscited by 1
  29. 2025In-Context Semi-Supervised Learningcited by 1

Built on by

Not cited yet.