mlsanityData to 7 Oct 2026

Black Box Causal Inference: Effect Estimation via Meta Prediction

arXiv 2503.05985 ↗First posted 7 Mar 2025cs.LG cs.AI stat.COcited by 13

Not on the map. Why?

Abstract (arXiv)

Causal inference and the estimation of causal effects plays a central role in decision-making across many areas, including healthcare and economics. Estimating causal effects typically requires an estimator that is tailored to each problem of interest. But developing estimators can take significant effort for even a single causal inference setting. For example, algorithms for regression-based estimators, propensity score methods, and doubly robust methods were designed across several decades to handle causal estimation with observed confounders. Similarly, several estimators have been developed to exploit instrumental variables (IVs), including two-stage least-squares (TSLS), control functions, and the method-of-moments. In this work, we instead frame causal inference as a dataset-level prediction problem, offloading algorithm design to the learning process. The approach we introduce, called black box causal inference (BBCI), builds estimators in a black-box manner by learning to predict causal effects from sampled dataset-effect pairs. We demonstrate accurate estimation of average treatment effects (ATEs) and conditional average treatment effects (CATEs) with BBCI across several causal inference problems with known identification, including problems with less developed estimators.

Key facts. Cited by 13 papers in the map (data through 7 Oct 2026). Builds on 17 of its 38 references that are in the map. First posted on arXiv 7 Mar 2025.

Builds on

17 of 38 references.

References · 17

  1. 2022TabPFN: A Transformer That Solves Small Tabular Classification Problems in a Secondcited by 643
  2. 2021Transformers Can Do Bayesian Inferencecited by 338
  3. 2013Black Box Variational Inferencecited by 317
  4. 2019A Meta-Transfer Objective for Learning to Disentangle Causal Mechanismscited by 158
  5. 2016Identifying Causal Effects With Proxy Variables of an Unmeasured Confoundercited by 138
  6. 2018The Blessings of Multiple Causescited by 103
  7. 2021The Causal-Neural Connection: Expressiveness, Learnability, and Inferencecited by 97
  8. 2022Deep End-to-end Causal Inferencecited by 93
  9. 2020An Introduction to Proximal Causal Learningcited by 91
  10. 2016Generalized Optimal Matching Methods for Causal Inferencecited by 39
  11. 2024Sample, estimate, aggregate: A recipe for causal discovery foundation modelscited by 19
  12. 2022Set Norm and Equivariant Skip Connections: Putting the Deep in Deep Setscited by 17
5 more
  1. 2021Causal Estimation with Functional Confounderscited by 9
  2. 2024Deep Autoregressive Models as Causal Inference Enginescited by 6
  3. 2024Language Models as Causal Effect Generatorscited by 4
  4. 2019General Control Functions for Causal Effect Estimation from Instrumental Variablescited by 2
  5. 2024Regression-Based Proximal Causal Inferencecited by 2

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2025: 4; 2026: 9.

13 papers.

Sort: