Doubly robust identification of treatment effects from multiple environments

Piersilvio De Bartolomeis, Julia Kostin, Javier Abad, Yixin Wang, Fanny Yang
3/18/2025
stat.MLcs.LGstat.ME

No official code found. Best implementation: jaabmar/RAMEN (2 stars, MIT, last commit 2026-01-19).

Abstract

Practical and ethical constraints often require the use of observational data for causal inference, particularly in medicine and social sciences. Yet, observational datasets are prone to confounding, potentially compromising the validity of causal conclusions. While it is possible to correct for biases if the underlying causal graph is known, this is rarely a feasible ask in practical scenarios. A common strategy is to adjust for all available covariates, yet this approach can yield biased treatment effect estimates, especially when post-treatment or unobserved variables are present. We propose RAMEN, an algorithm that produces unbiased treatment effect estimates by leveraging the heterogeneity of multiple data sources without the need to know or learn the underlying causal graph. Notably, RAMEN achieves doubly robust identification: it can identify the treatment effect whenever the causal parents of the treatment or those of the outcome are observed, and the node whose parents are observed satisfies an invariance assumption. Empirical evaluations on synthetic and real-world datasets show that our approach outperforms existing methods.

View on arXivView PDF

Code Implementations(1)

jaabmar/RAMENCommunity100%
MIT

Implementation for the paper "Doubly robust identification of treatment effects from multiple environments"

20PythonFeb 11, 20259 months agoMIT
causal-inferencedouble-robustidentifiabilitymachine-learningobservational-study+1 more

Similar Papers

Cite this paper

@article{bartolomeis2025doubly,
  title  = {Doubly robust identification of treatment effects from multiple environments},
  author = {Piersilvio De Bartolomeis and Julia Kostin and Javier Abad and Yixin Wang and Fanny Yang},
  year   = {2025},
  eprint = {2503.14459},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2503.14459v1},
  journal = {ICLR 2025 2025}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2503.14459.svg)](https://codeofpaper.com/paper/2503.14459)

Discussion