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