# Foundation Models for Partial Causal Identification

arXiv: https://arxiv.org/abs/2608.20841 · HTML: https://mlsanity.com/paper/2608.20841/

## Key facts

- Cited by 2 papers in the map (data through 14 Oct 2026).
- Builds on 20 of its 40 references that are in the map.
- First posted on arXiv 21 Aug 2026.

## Builds on (most cited first, 20 of 20)

- [Transformers Can Do Bayesian Inference](https://mlsanity.com/paper/2112.10510/) (2021)
- [Neural Processes](https://mlsanity.com/paper/1807.01622/) (2018)
- [Probabilities of Causation: Bounds and Identification](https://mlsanity.com/paper/1301.3898/) (2013)
- [Counterfactual Probabilities: Computational Methods, Bounds and Applications](https://mlsanity.com/paper/1302.6784/) (2013)
- [Partial Counterfactual Identification from Observational and Experimental Data](https://mlsanity.com/paper/2110.05690/) (2021)
- [Do-PFN: In-Context Learning for Causal Effect Estimation](https://mlsanity.com/paper/2506.06039/) (2025)
- [CausalPFN: Amortized Causal Effect Estimation via In-Context Learning](https://mlsanity.com/paper/2506.07918/) (2025)
- [An Automated Approach to Causal Inference in Discrete Settings](https://mlsanity.com/paper/2109.13471/) (2021)
- [Foundation Models for Causal Inference via Prior-Data Fitted Networks](https://mlsanity.com/paper/2506.10914/) (2025)
- [Stochastic Causal Programming for Bounding Treatment Effects](https://mlsanity.com/paper/2202.10806/) (2022)
- [A General Method for Deriving Tight Symbolic Bounds on Causal Effects](https://mlsanity.com/paper/2003.10702/) (2020)
- [Partial Identification of Treatment Effects with Implicit Generative Models](https://mlsanity.com/paper/2210.08139/) (2022)
- [Black Box Causal Inference: Effect Estimation via Meta Prediction](https://mlsanity.com/paper/2503.05985/) (2025)
- [Estimating Interventional Distributions with Uncertain Causal Graphs through Meta-Learning](https://mlsanity.com/paper/2507.05526/) (2025)
- [Deriving Bounds and Inequality Constraints Using LogicalRelations Among Counterfactuals](https://mlsanity.com/paper/2007.00628/) (2020)
- [Towards Bounding Causal Effects under Markov Equivalence](https://mlsanity.com/paper/2311.07259/) (2023)
- [Confounding Robust Deep Reinforcement Learning: A Causal Approach](https://mlsanity.com/paper/2510.21110/) (2025)
- [Use What You Know: Causal Foundation Models with Partial Graphs](https://mlsanity.com/paper/2602.14972/) (2026)
- [Deriving Complete Constraints in Hidden Variable Models](https://mlsanity.com/paper/2601.11242/) (2026)
- [Information-Theoretic Causal Bounds under Unmeasured Confounding](https://mlsanity.com/paper/2601.17160/) (2026)

## Built on by (most cited first, 2 of 2)

- [Causal Foundation Models](https://mlsanity.com/paper/2609.03003/) (2026)
- [What You Observe Determines How You Identify Causal Effects: Evaluating Causal Models across Observational Views](https://mlsanity.com/paper/2609.36881/) (2026)

## Abstract (arXiv)

This paper investigates the development of causal foundation models for bounding the effect of interventions and counterfactuals from observational data. We show that a canonical prior can be defined with full support over the space of structural causal models with discrete observables. With this canonical prior, we translate the problem of bounding counterfactuals into that of learning distributions over functions that map data (and possibly structural assumptions) to a causal query of interest. This extends the promising causal foundational modelling paradigm to the estimation of partially-identifiable causal effects, i.e., under unobserved confounding, where multiple values are equally compatible with the observed data and prior structural assumptions.
