Large language models reorganize representational geometry during in-context learning
Not on the map. Why?
Abstract (arXiv)
Large language models (LLMs) show remarkable flexibility in adapting to novel tasks without parameter updates, a capacity known as in-context learning (ICL). Prior work has sought to understand ICL by studying the circuits, algorithms, and representations that support it. Yet why some ICL tasks are easy to solve while others are difficult remains unresolved. In this paper, we ask whether LLMs can adapt their representations arbitrarily to solve a simple linear classification task. Specifically, we construct a family of binary classification tasks in which labels are defined by projecting LLMs' own representations onto different axes. Surprisingly, although all tasks are linearly separable by construction, their in-context learnability varies systematically across axes. We find that successful ICL is accompanied by a geometric reorganization of internal representations that increases task-relevant separability. Causal interventions that amplify neural activity along the axis defining the task are insufficient to improve behavioral performance or induce this representational reorganization. We also show that LLM behavior is best described by a prototype-like algorithm operating on representations that are themselves reorganized in context to adapt to the task. Together, these findings offer a geometric account of ICL in LLMs, showing that representations acquired through training constrain what can be exploited through in-context learning.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 37 of its 51 references that are in the map. First posted on arXiv 16 May 2026.
Builds on
37 of 51 references.
References · 37
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2025Qwen3 Technical Reportcited by 8,820
- 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,899
- 2022Rethinking the Role of Demonstrations: What Makes In-Context Learning Work?cited by 1,434
- 2021Calibrate Before Use: Improving Few-Shot Performance of Language Modelscited by 1,392
- 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 smallcited by 1,165
- 2022In-context Learning and Induction Headscited by 964
- 2021An Explanation of In-context Learning as Implicit Bayesian Inferencecited by 938
- 2022Transformers learn in-context by gradient descentcited by 732
- 2020Aligning AI With Shared Human Valuescited by 723
- 2022What Can Transformers Learn In-Context? A Case Study of Simple Function Classescited by 715
- 2023Larger language models do in-context learning differentlycited by 394
25 more
- 2017"Liar, Liar Pants on Fire": A New Benchmark Dataset for Fake News Detectioncited by 363
- 2021Transformers Can Do Bayesian Inferencecited by 360
- 2023In-Context Learning Creates Task Vectorscited by 312
- 2022Data Distributional Properties Drive Emergent In-Context Learning in Transformerscited by 309
- 2023Function Vectors in Large Language Modelscited by 308
- 2023Transformers as Statisticians: Provable In-Context Learning with In-Context Algorithm Selectioncited by 298
- 2023Transformers learn to implement preconditioned gradient descent for in-context learningcited by 279
- 2023Uncovering mesa-optimization algorithms in Transformerscited by 100
- 2024What needs to go right for an induction head? A mechanistic study of in-context learning circuits and their formationcited by 89
- 2022Transformers as Meta-Learners for Implicit Neural Representationscited by 79
- 2023Transformers Implement Functional Gradient Descent to Learn Non-Linear Functions In Contextcited by 63
- 2017Classification and Geometry of General Perceptual Manifoldscited by 62
- 2024ICLR: In-Context Learning of Representationscited by 50
- 2025Learning without training: The implicit dynamics of in-context learningcited by 43
- 2025Language Models Are Capable of Metacognitive Monitoring and Control of Their Internal Activationscited by 41
- 2023Towards Understanding In-Context Learning with Contrastive Demonstrations and Saliency Mapscited by 35
- 2025Can Transformers Learn Full Bayesian Inference in Context?cited by 31
- 2025Belief Dynamics Reveal the Dual Nature of In-Context Learning and Activation Steeringcited by 29
- 2024Can In-context Learning Really Generalize to Out-of-distribution Tasks?cited by 23
- 2025The Geometry of Prompting: Unveiling Distinct Mechanisms of Task Adaptation in Language Modelscited by 18
- 2024Dual Process Learning: Controlling Use of In-Context vs. In-Weights Strategies with Weight Forgettingcited by 15
- 2023A Spectral Theory of Neural Prediction and Alignmentcited by 14
- 2026Linear representations in language models can change dramatically over a conversationcited by 10
- 2025In-Context Representation Hijackingcited by 5
- 2025Provable Low-Frequency Bias of In-Context Learning of Representationscited by 3
Built on by
Citing papers per year: 2026: 2.
2 papers.