arXiv Statistics ML

Identification and Estimation under Multiple Versions of Treatment: Mixture-of-Experts Approach

arXiv Statistics ML
3d ago

Identifying Causal Effects Using a Single Proxy Variable

The paper tackles the problem of estimating causal effects when an unobserved confounder is present. It assumes a single, possibly multi‑dimensional proxy variable for the confounder and knowledge of the mechanism that generates this proxy. Under the Single Proxy Identifiability of Causal Effects (SPICE) assumption, the authors prove that the error mechanism is complete and causal effects are identifiable, extending prior proxy‑based results to continuous, multi‑dimensional settings and more flexible functional forms. They also introduce SPICE‑Net, a neural‑network‑based framework for estimating causal effects applicable to both discrete and continuous treatments.

By Silvan Vollmer, Niklas Pfister, Sebastian Weichwald
arXiv Statistics ML
Aug 24

Double Machine Learning of Continuous Treatment Effects with Additive Instrumental Variables

The paper introduces a new framework for identifying average dose-response functions in the presence of unmeasured confounding by using instrumental variables. It defines a uniform regular weighting function and partitions the treatment space into open sets where local identification is possible. For estimation, the authors propose an augmented inverse probability weighted score within a debiased machine learning setting, along with practical guidance for constructing weighting functions, falsification tests for the additive IV condition, and asymptotic theory for kernel regression or empirical risk minimization estimators.

By Shuyuan Chen, Peng Zhang, Yifan Cui
arXiv Machine Learning
Sep 4

Causal Foundation Models

Causal Foundation Models (CFMs) are pretrained neural networks designed to estimate causal quantities—such as the average treatment effect—across new datasets using in‑context learning, eliminating the need for bespoke pipelines or model updates. The paper introduces CFMs, reviews foundational concepts in causal inference and machine learning, and provides practical code examples and Jupyter notebooks to illustrate their application.

By Christopher Stith, Hossein Rahmani, Jesse C. Cresswell
arXiv Machine Learning
Sep 22

Causal Inference with Unobserved Confounding: A Mixture Learning Perspective

The paper introduces a mixture‑learning framework for causal inference with unobserved confounding, treating latent confounders as sources of heterogeneity that create mixture structures in observed data. By assuming suitable structural and identifiability conditions, it shows that recovering the mixing distribution and component mechanisms allows estimation of interventional distributions and causal estimands. The authors illustrate the approach with Bernoulli mixture examples, extend it to high‑dimensional exponential‑family mixtures with dependent outcomes, and relate it to panel‑data settings, latent factor models, and synthetic interventions.

By Mansi Sood, Devavrat Shah
arXiv AI
Aug 24

Foundation Models for Partial Causal Identification

The paper introduces causal foundation models that can bound the effects of interventions and counterfactuals using only observational data. It defines a canonical prior with full support over structural causal models with discrete observables, enabling the translation of counterfactual bounding into learning distributions over functions that map data and structural assumptions to causal queries. This approach extends causal foundational modelling to partially-identifiable causal effects, where unobserved confounding leads to multiple compatible values for the effect.

By Alexis Bellot, Anish Dhir