AI RESEARCH

Layerwise Dynamics for In-Context Classification in Transformers

arXiv CS.AI

ArXi:2604.11613v1 Announce Type: cross 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.