mlsanityData to 14 Oct 2026

Discovering Hierarchical Latent Capabilities of Language Models via Causal Representation Learning

arXiv 2506.10378 ↗First posted 12 Jun 2025cs.LG cs.AI cs.CLcited by 4

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

  1. 2020Language Models are Few-Shot Learnerscited by 32,419
  2. 2023GPT-4 Technical Reportcited by 18,396
  3. 2022Training language models to follow instructions with human feedbackcited by 15,971
  4. 2024The Llama 3 Herd of Modelscited by 14,715
  5. 2020Measuring Massive Multitask Language Understandingcited by 7,963
  6. 2020Scaling Laws for Neural Language Modelscited by 6,586
  7. 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,330
  8. 2024Qwen2.5 Technical Reportcited by 4,942
  9. 2022Training a Helpful and Harmless Assistant with Reinforcement Learning from Human Feedbackcited by 3,720
  10. 2022Scaling Instruction-Finetuned Language Modelscited by 3,024
  11. 2022Training Compute-Optimal Large Language Modelscited by 3,019
  12. 2022Emergent Abilities of Large Language Modelscited by 2,512
38 more
  1. 2024Gemma 2: Improving Open Language Models at a Practical Sizecited by 1,998
  2. 2024Phi-3 Technical Report: A Highly Capable Language Model Locally on Your Phonecited by 1,917
  3. 2019Invariant Risk Minimizationcited by 1,854
  4. 2022Challenging BIG-Bench Tasks and Whether Chain-of-Thought Can Solve Themcited by 1,807
  5. 2022Rethinking the Role of Demonstrations: What Makes In-Context Learning Work?cited by 1,434
  6. 2022Solving Quantitative Reasoning Problems with Language Modelscited by 1,392
  7. 2023Are Emergent Abilities of Large Language Models a Mirage?cited by 523
  8. 2025Cognitive Behaviors that Enable Self-Improving Reasoners, or, Four Habits of Highly Effective STaRscited by 377
  9. 2023The False Promise of Imitating Proprietary LLMscited by 263
  10. 2020Weakly-Supervised Disentanglement Without Compromisescited by 239
  11. 2023Inverse Scaling: When Bigger Isn't Bettercited by 207
  12. 2024When Scaling Meets LLM Finetuning: The Effect of Data, Model and Finetuning Methodcited by 204
  13. 2020Invariant Risk Minimization Gamescited by 188
  14. 2023Time Travel in LLMs: Tracing Data Contamination in Large Language Modelscited by 177
  15. 2019Causal Discovery from Heterogeneous/Nonstationary Data with Independent Changescited by 166
  16. 2022Weakly supervised causal representation learningcited by 140
  17. 2024Observational Scaling Laws and the Predictability of Language Model Performancecited by 130
  18. 2025Echo Chamber: RL Post-training Amplifies Behaviors Learned in Pretrainingcited by 129
  19. 2023Investigating the Catastrophic Forgetting in Multimodal Large Language Modelscited by 119
  20. 2023A Theory for Emergence of Complex Skills in Language Modelscited by 113
  21. 2023Nonparametric Identifiability of Causal Representations from Unknown Interventionscited by 106
  22. 2025A Sober Look at Progress in Language Model Reasoning: Pitfalls and Paths to Reproducibilitycited by 93
  23. 2023Learning Linear Causal Representations from Interventions under General Nonlinear Mixingcited by 92
  24. 2023Identifiability Guarantees for Causal Disentanglement from Soft Interventionscited by 84
  25. 2024A Closer Look at the Limitations of Instruction Tuningcited by 77
  26. 2021Desiderata for Representation Learning: A Causal Perspectivecited by 76
  27. 2024Causal Representation Learning from Multiple Distributions: A General Settingcited by 68
  28. 2023Skill-Mix: a Flexible and Expandable Family of Evaluations for AI modelscited by 64
  29. 2024Safetywashing: Do AI Safety Benchmarks Actually Measure Safety Progress?cited by 61
  30. 2024Training on the Test Task Confounds Evaluation and Emergencecited by 34
  31. 2023Skills-in-Context Prompting: Unlocking Compositionality in Large Language Modelscited by 33
  32. 2024Unveiling the Impact of Coding Data Instruction Fine-Tuning on Large Language Models Reasoningcited by 31
  33. 2025Reliable and Efficient Amortized Model-based Evaluationcited by 31
  34. 2024Sloth: scaling laws for LLM skills to predict multi-benchmark performance across familiescited by 24
  35. 2024Does RLHF Scale? Exploring the Impacts From Data, Model, and Methodcited by 23
  36. 2015Structured Matrix Completion with Applications to Genomic Data Integrationcited by 17
  37. 2025Fundamental Safety-Capability Trade-offs in Fine-tuning Large Language Modelscited by 17
  38. 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

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 4.

4 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Capacity, Responsiveness and Alignment: What Makes a Latent Structure Actionable
  2. 2024CausalBN-Bench: A Comprehensive Benchmark for Causal Learning Capability of LLMs
  3. 2026Causality is Key for Interpretability Claims to Generalise
  4. 2025A Survey on Enhancing Causal Reasoning Ability of Large Language Models
  5. 2024Causal Evaluation of Language Models
  6. 2025Unveiling Causal Reasoning in Large Language Models: Reality or Mirage?
  7. 2024Can Large Language Models Learn Independent Causal Mechanisms?
  8. 2024Large Language Models and Causal Inference in Collaboration: A Survey