mlsanityData to 14 Oct 2026

Large language models reorganize representational geometry during in-context learning

arXiv 2605.28854 ↗First posted 16 May 2026cs.CL cs.LG q-bio.NCcited by 2

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

  1. 2020Language Models are Few-Shot Learnerscited by 32,419
  2. 2025Qwen3 Technical Reportcited by 8,820
  3. 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,899
  4. 2022Rethinking the Role of Demonstrations: What Makes In-Context Learning Work?cited by 1,434
  5. 2021Calibrate Before Use: Improving Few-Shot Performance of Language Modelscited by 1,392
  6. 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 smallcited by 1,165
  7. 2022In-context Learning and Induction Headscited by 964
  8. 2021An Explanation of In-context Learning as Implicit Bayesian Inferencecited by 938
  9. 2022Transformers learn in-context by gradient descentcited by 732
  10. 2020Aligning AI With Shared Human Valuescited by 723
  11. 2022What Can Transformers Learn In-Context? A Case Study of Simple Function Classescited by 715
  12. 2023Larger language models do in-context learning differentlycited by 394
25 more
  1. 2017"Liar, Liar Pants on Fire": A New Benchmark Dataset for Fake News Detectioncited by 363
  2. 2021Transformers Can Do Bayesian Inferencecited by 360
  3. 2023In-Context Learning Creates Task Vectorscited by 312
  4. 2022Data Distributional Properties Drive Emergent In-Context Learning in Transformerscited by 309
  5. 2023Function Vectors in Large Language Modelscited by 308
  6. 2023Transformers as Statisticians: Provable In-Context Learning with In-Context Algorithm Selectioncited by 298
  7. 2023Transformers learn to implement preconditioned gradient descent for in-context learningcited by 279
  8. 2023Uncovering mesa-optimization algorithms in Transformerscited by 100
  9. 2024What needs to go right for an induction head? A mechanistic study of in-context learning circuits and their formationcited by 89
  10. 2022Transformers as Meta-Learners for Implicit Neural Representationscited by 79
  11. 2023Transformers Implement Functional Gradient Descent to Learn Non-Linear Functions In Contextcited by 63
  12. 2017Classification and Geometry of General Perceptual Manifoldscited by 62
  13. 2024ICLR: In-Context Learning of Representationscited by 50
  14. 2025Learning without training: The implicit dynamics of in-context learningcited by 43
  15. 2025Language Models Are Capable of Metacognitive Monitoring and Control of Their Internal Activationscited by 41
  16. 2023Towards Understanding In-Context Learning with Contrastive Demonstrations and Saliency Mapscited by 35
  17. 2025Can Transformers Learn Full Bayesian Inference in Context?cited by 31
  18. 2025Belief Dynamics Reveal the Dual Nature of In-Context Learning and Activation Steeringcited by 29
  19. 2024Can In-context Learning Really Generalize to Out-of-distribution Tasks?cited by 23
  20. 2025The Geometry of Prompting: Unveiling Distinct Mechanisms of Task Adaptation in Language Modelscited by 18
  21. 2024Dual Process Learning: Controlling Use of In-Context vs. In-Weights Strategies with Weight Forgettingcited by 15
  22. 2023A Spectral Theory of Neural Prediction and Alignmentcited by 14
  23. 2026Linear representations in language models can change dramatically over a conversationcited by 10
  24. 2025In-Context Representation Hijackingcited by 5
  25. 2025Provable Low-Frequency Bias of In-Context Learning of Representationscited by 3

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 2.

2 papers.