Discovering Hierarchical Latent Capabilities of Language Models via Causal Representation Learning
Not on the map. Why?
Abstract (arXiv)
Faithful evaluation of language model capabilities is crucial for deriving actionable insights that can inform model development. However, rigorous causal evaluations in this domain face significant methodological challenges, including complex confounding effects and prohibitive computational costs associated with extensive retraining. To tackle these challenges, we propose a causal representation learning framework wherein observed benchmark performance is modeled as a linear transformation of a few latent capability factors. Crucially, these latent factors are identified as causally interrelated after appropriately controlling for the base model as a common confounder. Applying this approach to a comprehensive dataset encompassing over 1500 models evaluated across six benchmarks from the Open LLM Leaderboard, we identify a concise three-node linear causal structure that reliably explains the observed performance variations. Further interpretation of this causal structure provides substantial scientific insights beyond simple numerical rankings: specifically, we reveal a clear causal direction starting from general problem-solving capabilities, advancing through instruction-following proficiency, and culminating in mathematical reasoning ability. Our results underscore the essential role of carefully controlling base model variations during evaluation, a step critical to accurately uncovering the underlying causal relationships among latent model capabilities.
Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Builds on 52 of its 73 references that are in the map. First posted on arXiv 12 Jun 2025.
Builds on
52 of 73 references.
References · 52
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2023GPT-4 Technical Reportcited by 18,396
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2020Measuring Massive Multitask Language Understandingcited by 7,963
- 2020Scaling Laws for Neural Language Modelscited by 6,586
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,330
- 2024Qwen2.5 Technical Reportcited by 4,942
- 2022Training a Helpful and Harmless Assistant with Reinforcement Learning from Human Feedbackcited by 3,720
- 2022Scaling Instruction-Finetuned Language Modelscited by 3,024
- 2022Training Compute-Optimal Large Language Modelscited by 3,019
- 2022Emergent Abilities of Large Language Modelscited by 2,512
38 more
- 2024Gemma 2: Improving Open Language Models at a Practical Sizecited by 1,998
- 2024Phi-3 Technical Report: A Highly Capable Language Model Locally on Your Phonecited by 1,917
- 2019Invariant Risk Minimizationcited by 1,854
- 2022Challenging BIG-Bench Tasks and Whether Chain-of-Thought Can Solve Themcited by 1,807
- 2022Rethinking the Role of Demonstrations: What Makes In-Context Learning Work?cited by 1,434
- 2022Solving Quantitative Reasoning Problems with Language Modelscited by 1,392
- 2023Are Emergent Abilities of Large Language Models a Mirage?cited by 523
- 2025Cognitive Behaviors that Enable Self-Improving Reasoners, or, Four Habits of Highly Effective STaRscited by 377
- 2023The False Promise of Imitating Proprietary LLMscited by 263
- 2020Weakly-Supervised Disentanglement Without Compromisescited by 239
- 2023Inverse Scaling: When Bigger Isn't Bettercited by 207
- 2024When Scaling Meets LLM Finetuning: The Effect of Data, Model and Finetuning Methodcited by 204
- 2020Invariant Risk Minimization Gamescited by 188
- 2023Time Travel in LLMs: Tracing Data Contamination in Large Language Modelscited by 177
- 2019Causal Discovery from Heterogeneous/Nonstationary Data with Independent Changescited by 166
- 2022Weakly supervised causal representation learningcited by 140
- 2024Observational Scaling Laws and the Predictability of Language Model Performancecited by 130
- 2025Echo Chamber: RL Post-training Amplifies Behaviors Learned in Pretrainingcited by 129
- 2023Investigating the Catastrophic Forgetting in Multimodal Large Language Modelscited by 119
- 2023A Theory for Emergence of Complex Skills in Language Modelscited by 113
- 2023Nonparametric Identifiability of Causal Representations from Unknown Interventionscited by 106
- 2025A Sober Look at Progress in Language Model Reasoning: Pitfalls and Paths to Reproducibilitycited by 93
- 2023Learning Linear Causal Representations from Interventions under General Nonlinear Mixingcited by 92
- 2023Identifiability Guarantees for Causal Disentanglement from Soft Interventionscited by 84
- 2024A Closer Look at the Limitations of Instruction Tuningcited by 77
- 2021Desiderata for Representation Learning: A Causal Perspectivecited by 76
- 2024Causal Representation Learning from Multiple Distributions: A General Settingcited by 68
- 2023Skill-Mix: a Flexible and Expandable Family of Evaluations for AI modelscited by 64
- 2024Safetywashing: Do AI Safety Benchmarks Actually Measure Safety Progress?cited by 61
- 2024Training on the Test Task Confounds Evaluation and Emergencecited by 34
- 2023Skills-in-Context Prompting: Unlocking Compositionality in Large Language Modelscited by 33
- 2024Unveiling the Impact of Coding Data Instruction Fine-Tuning on Large Language Models Reasoningcited by 31
- 2025Reliable and Efficient Amortized Model-based Evaluationcited by 31
- 2024Sloth: scaling laws for LLM skills to predict multi-benchmark performance across familiescited by 24
- 2024Does RLHF Scale? Exploring the Impacts From Data, Model, and Methodcited by 23
- 2015Structured Matrix Completion with Applications to Genomic Data Integrationcited by 17
- 2025Fundamental Safety-Capability Trade-offs in Fine-tuning Large Language Modelscited by 17
- 2025Not-Just-Scaling Laws: Towards a Better Understanding of the Downstream Impact of Language Model Design Decisionscited by 12
50 most cited of 52 All 52 (JSON)
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026Domain-Aware Scaling Laws Uncover Data Synergycited by 2
- 2026Prescriptive Scaling Reveals the Evolution of Language Model Capabilitiescited by 1
- 2026Beyond identifiability: Learning causal representations with few environments and finite samplescited by 1
- 2026Peer-Predictive Self-Training for Language Model Reasoningcited by 1
- 2026Domain-Aware Scaling Laws Uncover Data Synergy13 Jul 2026
- 2026Peer-Predictive Self-Training for Language Model Reasoning14 Apr 2026
- 2026Beyond identifiability: Learning causal representations with few environments and finite samples26 Mar 2026
- 2026Prescriptive Scaling Reveals the Evolution of Language Model Capabilities17 Feb 2026
Similar, not cited
Close in wording, no citation either way
- 2026Capacity, Responsiveness and Alignment: What Makes a Latent Structure Actionable
- 2024CausalBN-Bench: A Comprehensive Benchmark for Causal Learning Capability of LLMs
- 2026Causality is Key for Interpretability Claims to Generalise
- 2025A Survey on Enhancing Causal Reasoning Ability of Large Language Models
- 2024Causal Evaluation of Language Models
- 2025Unveiling Causal Reasoning in Large Language Models: Reality or Mirage?
- 2024Can Large Language Models Learn Independent Causal Mechanisms?
- 2024Large Language Models and Causal Inference in Collaboration: A Survey