I Predict Therefore I Am: Is Next Token Prediction Enough to Learn Human-Interpretable Concepts from Data?
Not on the map. Why?
Abstract (arXiv)
Recent empirical evidence shows that LLM representations encode human-interpretable concepts. Nevertheless, the mechanisms by which these representations emerge remain largely unexplored. To shed further light on this, we introduce a novel generative model that generates tokens on the basis of such concepts formulated as latent discrete variables. Under mild conditions, even when the mapping from the latent space to the observed space is non-invertible, we establish rigorous identifiability result: the representations learned by LLMs through next-token prediction can be approximately modeled as the logarithm of the posterior probabilities of these latent discrete concepts given input context, up to an linear transformation. This theoretical finding: 1) provides evidence that LLMs capture essential underlying generative factors, 2) offers a unified and principled perspective for understanding the linear representation hypothesis, and 3) motivates a theoretically grounded approach for evaluating sparse autoencoders. Empirically, we validate our theoretical results through evaluations on both simulation data and the Pythia, Llama, and DeepSeek model families.
Key facts. Cited by 13 papers in the map (data through 7 Oct 2026). Builds on 62 of its 106 references that are in the map. First posted on arXiv 12 Mar 2025.
Builds on
62 of 106 references.
References · 62
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
- 2021On the Opportunities and Risks of Foundation Modelscited by 3,961
- 2023A Survey of Large Language Modelscited by 2,537
- 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,346
- 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,734
- 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,359
- 2023Inference-Time Intervention: Eliciting Truthful Answers from a Language Modelcited by 1,259
- 2022Toy Models of Superpositioncited by 928
- 2022Discovering Latent Knowledge in Language Models Without Supervisioncited by 733
38 more
- 2023The Linear Representation Hypothesis and the Geometry of Large Language Modelscited by 679
- 2023The Geometry of Truth: Emergent Linear Structure in Large Language Model Representations of True/False Datasetscited by 673
- 2024DeepSeek LLM: Scaling Open-Source Language Models with Longtermismcited by 621
- 2024Scaling and evaluating sparse autoencoderscited by 530
- 2019Variational Autoencoders and Nonlinear ICA: A Unifying Frameworkcited by 494
- 2023Explainability for Large Language Models: A Surveycited by 331
- 2023Finding Neurons in a Haystack: Case Studies with Sparse Probingcited by 328
- 2023Emergent Linear Representations in World Models of Self-Supervised Sequence Modelscited by 313
- 2021Self-Supervised Learning with Data Augmentations Provably Isolates Content from Stylecited by 308
- 2023Language Models Represent Space and Timecited by 306
- 2024Jumping Ahead: Improving Reconstruction Fidelity with JumpReLU Sparse Autoencoderscited by 283
- 2018Nonlinear ICA Using Auxiliary Variables and Generalized Contrastive Learningcited by 278
- 2016Unsupervised Feature Extraction by Time-Contrastive Learning and Nonlinear ICAcited by 270
- 2022Relative representations enable zero-shot latent space communicationcited by 175
- 2025Open Problems in Mechanistic Interpretabilitycited by 173
- 2024Improving Dictionary Learning with Gated Sparse Autoencoderscited by 171
- 2016Linear Algebraic Structure of Word Senses, with Applications to Polysemycited by 165
- 2023Linear Representations of Sentiment in Large Language Modelscited by 157
- 2024The Geometry of Categorical and Hierarchical Concepts in Large Language Modelscited by 151
- 2022Weakly supervised causal representation learningcited by 140
- 2022Interventional Causal Representation Learningcited by 140
- 2024BatchTopK Sparse Autoencoderscited by 126
- 2025Are Sparse Autoencoders Useful? A Case Study in Sparse Probingcited by 109
- 2023Nonparametric Identifiability of Causal Representations from Unknown Interventionscited by 105
- 2020On Linear Identifiability of Learned Representationscited by 103
- 2023Learning Linear Causal Representations from Interventions under General Nonlinear Mixingcited by 92
- 2022Linear Causal Disentanglement via Interventionscited by 84
- 2024On the Origins of Linear Representations in Large Language Modelscited by 76
- 2020Weakly Supervised Disentangled Generative Causal Representation Learningcited by 67
- 2024Identifying Functionally Important Features with End-to-End Sparse Dictionary Learningcited by 64
- 2021Learning latent causal graphs via mixture oraclescited by 55
- 2024Measuring Progress in Dictionary Learning for Language Model Interpretability with Board Game Modelscited by 51
- 2023Score-based Causal Representation Learning with Interventionscited by 49
- 2015Simple, Efficient, and Neural Algorithms for Sparse Codingcited by 37
- 2022Analyzing Encoded Concepts in Transformer Language Modelscited by 37
- 2022Identifying Weight-Variant Latent Causal Modelscited by 33
- 2024Learning Interpretable Concepts: Unifying Causal Representation Learning and Foundation Modelscited by 29
- 2022Causal Abstraction with Soft Interventionscited by 23
50 most cited of 62 All 62 (JSON)
Built on by
Citing papers per year: 2025: 3; 2026: 10.
13 papers.
Sort:
By area:
- 2026Truth as a Trajectory: What Internal Representations Reveal About Large Language Model Reasoningcited by 9
- 2026Statistical and structural identifiability in representation learningcited by 4
- 2026What Makes a Representation Good for Single-Cell Perturbation Prediction?cited by 3
- 2025Causal Disentanglement and Cross-Modal Alignment for Enhanced Few-Shot Learningcited by 2
- 2026From Chains to DAGs: Probing the Graph Structure of Reasoning in LLMscited by 2
- 2026Learning Concepts, Not Tokens: Self-Supervised Semantic Alignment for Language Modelscited by 1
- 2026Learning Latent Dynamical Causal Processes for Single-Cell Perturbation Predictioncited by 1
- 2025Beyond the Linear Separability Ceiling: Aligning Representations in VLMscited by 0
- 2025Beyond the Black Box: Identifiable Interpretation and Control in Generative Models via Causal Minimalitycited by 0
- 2026Concept Component Analysis: A Principled Approach for Concept Extraction in LLMscited by 0
- 2026Boundary Embedding Shaping with Adaptive Contrastive Learning for Graph Structural Disentanglementcited by 0
- 2026Reasoning Errors Have a Region and a Direction in the Residual-Stream Trajectory of LLMscited by 0
1 more
- 2026A Survey on the Linear Representation Hypothesiscited by 0
- 2026A Survey on the Linear Representation Hypothesis19 Sep 2026
- 2026Reasoning Errors Have a Region and a Direction in the Residual-Stream Trajectory of LLMs6 Aug 2026
- 2026Boundary Embedding Shaping with Adaptive Contrastive Learning for Graph Structural Disentanglement18 Jun 2026
- 2026Learning Latent Dynamical Causal Processes for Single-Cell Perturbation Prediction25 May 2026
- 2026What Makes a Representation Good for Single-Cell Perturbation Prediction?19 May 2026
- 2026Learning Concepts, Not Tokens: Self-Supervised Semantic Alignment for Language Models31 Mar 2026
- 2026Statistical and structural identifiability in representation learning12 Mar 2026
- 2026Truth as a Trajectory: What Internal Representations Reveal About Large Language Model Reasoning1 Mar 2026
- 2026Concept Component Analysis: A Principled Approach for Concept Extraction in LLMs28 Jan 2026
- 2026From Chains to DAGs: Probing the Graph Structure of Reasoning in LLMs24 Jan 2026
- 2025Beyond the Black Box: Identifiable Interpretation and Control in Generative Models via Causal Minimality11 Dec 2025
- 2025Causal Disentanglement and Cross-Modal Alignment for Enhanced Few-Shot Learning5 Aug 2025
1 more
- 2026Statistical and structural identifiability in representation learningcited by 4
- 2026What Makes a Representation Good for Single-Cell Perturbation Prediction?cited by 3
- 2025Causal Disentanglement and Cross-Modal Alignment for Enhanced Few-Shot Learningcited by 2
- 2026Learning Latent Dynamical Causal Processes for Single-Cell Perturbation Predictioncited by 1
- 2025Beyond the Black Box: Identifiable Interpretation and Control in Generative Models via Causal Minimalitycited by 0
- 2026Concept Component Analysis: A Principled Approach for Concept Extraction in LLMscited by 0
- 2026Boundary Embedding Shaping with Adaptive Contrastive Learning for Graph Structural Disentanglementcited by 0
- 2026A Survey on the Linear Representation Hypothesiscited by 0
- 2026A Survey on the Linear Representation Hypothesis19 Sep 2026
- 2026Boundary Embedding Shaping with Adaptive Contrastive Learning for Graph Structural Disentanglement18 Jun 2026
- 2026Learning Latent Dynamical Causal Processes for Single-Cell Perturbation Prediction25 May 2026
- 2026What Makes a Representation Good for Single-Cell Perturbation Prediction?19 May 2026
- 2026Statistical and structural identifiability in representation learning12 Mar 2026
- 2026Concept Component Analysis: A Principled Approach for Concept Extraction in LLMs28 Jan 2026
- 2025Beyond the Black Box: Identifiable Interpretation and Control in Generative Models via Causal Minimality11 Dec 2025
- 2025Causal Disentanglement and Cross-Modal Alignment for Enhanced Few-Shot Learning5 Aug 2025