mlsanityData to 7 Oct 2026

Convergent World Representations and Divergent Tasks

arXiv 2602.00533 ↗First posted 31 Jan 2026cs.LG cs.AIcited by 1

Not on the map. Why?

Abstract (arXiv)

While neural representations are central to modern deep learning, the conditions governing their geometry and their roles in downstream adaptability remain poorly understood. We develop a framework clearly separating the underlying world, the data generation process and the resulting model representations to study these questions in a controlled setup. 5,075 city coordinates define the world and 7 geometric tasks generate the training data for autoregressive training. We find that different tasks give rise to qualitatively and quantitatively distinct world representation geometries. However, multi-task training drives convergence of world representations: models trained on non-overlapping tasks develop aligned geometric representations, providing controlled evidence for the Multitask Scaling Hypothesis of the Platonic Representation Hypothesis. To study adaptation, we pretrain models on all tasks, then test whether new entities (cities) can be consistently integrated into the representation space via fine-tuning. Surprisingly, we find that despite multi-task pretraining, some tasks, which we call divergent, actively harm the representational integration of new entities and harm generalization. Our results show that training on multiple relational tasks reliably produces convergent world representations, but lurking divergent tasks can catastrophically harm new entity integration via fine-tuning.

Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 64 of its 87 references that are in the map. First posted on arXiv 31 Jan 2026.

Builds on

64 of 87 references.

References · 64

  1. 2015Deep Residual Learning for Image Recognitioncited by 51,809
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
  3. 2020Language Models are Few-Shot Learnerscited by 32,199
  4. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
  5. 2024Qwen2.5 Technical Reportcited by 4,713
  6. 2021The Power of Scale for Parameter-Efficient Prompt Tuningcited by 3,769
  7. 2019Similarity of Neural Network Representations Revisitedcited by 1,695
  8. 2021Geometric Deep Learning: Grids, Groups, Graphs, Geodesics, and Gaugescited by 1,131
  9. 2022Editing Models with Task Arithmeticcited by 1,076
  10. 2024Refusal in Language Models Is Mediated by a Single Directioncited by 960
  11. 2025Does Reinforcement Learning Really Incentivize Reasoning Capacity in LLMs Beyond the Base Model?cited by 948
  12. 2021An Explanation of In-context Learning as Implicit Bayesian Inferencecited by 915
38 more
  1. 2023Progress measures for grokking via mechanistic interpretabilitycited by 822
  2. 2022Fine-Tuning can Distort Pretrained Features and Underperform Out-of-Distributioncited by 722
  3. 2023The Geometry of Truth: Emergent Linear Structure in Large Language Model Representations of True/False Datasetscited by 673
  4. 2019General E(2)-Equivariant Steerable CNNscited by 441
  5. 2022Emergent World Representations: Exploring a Sequence Model Trained on a Synthetic Taskcited by 417
  6. 2023The Reversal Curse: LLMs trained on "A is B" fail to learn "B is A"cited by 399
  7. 2024The Platonic Representation Hypothesiscited by 362
  8. 2020The Pitfalls of Simplicity Bias in Neural Networkscited by 349
  9. 2023Fine-Tuning Language Models with Just Forward Passescited by 337
  10. 2023Emergent Linear Representations in World Models of Self-Supervised Sequence Modelscited by 313
  11. 2023Language Models Represent Space and Timecited by 306
  12. 2022Data Distributional Properties Drive Emergent In-Context Learning in Transformerscited by 303
  13. 2023Physics of Language Models: Part 3.1, Knowledge Storage and Extractioncited by 264
  14. 2020Gradient Starvation: A Learning Proclivity in Neural Networkscited by 262
  15. 2021Muppet: Massive Multi-task Representations with Pre-Finetuningcited by 192
  16. 2024A Mechanistic Understanding of Alignment Algorithms: A Case Study on DPO and Toxicitycited by 179
  17. 2024The pitfalls of next-token predictioncited by 156
  18. 2023Pretraining task diversity and the emergence of non-Bayesian in-context learning for regressioncited by 138
  19. 2023The Quantization Model of Neural Scalingcited by 137
  20. 2025Echo Chamber: RL Post-training Amplifies Behaviors Learned in Pretrainingcited by 123
  21. 2023The mechanistic basis of data dependence and abrupt learning in an in-context classification taskcited by 115
  22. 2023Mechanistically analyzing the effects of fine-tuning on procedurally defined taskscited by 110
  23. 2023Compositional Abilities Emerge Multiplicatively: Exploring Diffusion Models on a Synthetic Taskcited by 91
  24. 2024What needs to go right for an induction head? A mechanistic study of in-context learning circuits and their formationcited by 86
  25. 2023Physics of Language Models: Part 1, Learning Hierarchical Language Structurescited by 83
  26. 2017Critical Learning Periods in Deep Neural Networkscited by 78
  27. 2024Transformers represent belief state geometry in their residual streamcited by 73
  28. 2025Self-Adapting Language Modelscited by 71
  29. 2025Projecting Assumptions: The Duality Between Sparse Autoencoders and Concept Geometrycited by 58
  30. 2024Competition Dynamics Shape Algorithmic Phases of In-Context Learningcited by 52
  31. 2025What Has a Foundation Model Found? Using Inductive Bias to Probe for World Modelscited by 46
  32. 2024Recurrent Neural Networks Learn to Store and Generate Sequences using Non-Linear Representationscited by 45
  33. 2023Maintaining Plasticity in Deep Continual Learningcited by 42
  34. 2025On the generalization of language models from in-context learning and finetuning: a controlled studycited by 39
  35. 2025Steering Out-of-Distribution Generalization with Concept Ablation Fine-Tuningcited by 34
  36. 2024Emergence of Hidden Capabilities: Exploring Learning Dynamics in Concept Spacecited by 33
  37. 2025Tracing the Representation Geometry of Language Models from Pretraining to Post-trainingcited by 33
  38. 2025Overcoming Sparsity Artifacts in Crosscoders to Interpret Chat-Tuningcited by 31

50 most cited of 64 All 64 (JSON)

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.