arXiv Statistics ML

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.

arXiv Statistics ML
Oct 5

Flexible Nonparametric Inference for Causal Effects under the Front-Door Model

The paper introduces new one‑step and targeted minimum loss‑based estimators for the average treatment effect and the average treatment effect on the treated under the front‑door criterion. These estimators use multiple parameterizations of the observed data distribution, can avoid modeling the mediator density, and are compatible with flexible machine‑learning nuisance estimation. The authors derive conditions for root‑n consistency, asymptotic linearity, and develop doubly robust tests for identification assumptions, demonstrating improved efficiency and favorable performance in simulations and real‑world applications in education and emergency medicine.

By Anna Guo, David Benkeser, Razieh Nabi
arXiv Machine Learning
Jun 15

Attention-Based Estimation of the Individual Treatment Benefit Probability under Dose Variation

arXiv:2606. 13821v1 Announce Type: new Abstract: Estimating the probability that a treatment outperforms a control for an individual patient, called the Individual Probability of Treatment Benefit (IPTB), offers a clinically intuitive alternative to population-average metrics.

By Lev V. Utkin, Andrei V. Konstantinov, Stanislav K. Kogan, Natalya M. Verbova, Maksim I. Goriunov
arXiv Statistics ML
Sep 28

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
Hugging Face Trending Papers
Jun 2

Tailoring Strictly Proper Scoring Rules for Downstream Tasks: An Application to Causal Inference

Probabilistic models are typically trained using task-agnostic objectives like log-loss, which can lead to significant errors in downstream estimation. This disconnect is especially critical in Inverse Probability Weighting (IPW) for causal inference, where propensity score errors near $0$ and $1$ often lead to high bias and variance.

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 AI
Aug 20

CausalProfiler: Generating Synthetic Benchmarks for Rigorous and Transparent Evaluation of Causal Machine Learning

CausalProfiler is a synthetic benchmark generator designed to evaluate causal machine learning (Causal ML) methods more rigorously and transparently. It randomly samples causal models, data, queries, and ground truths based on explicit design choices across observation, intervention, and counterfactual reasoning levels, providing coverage guarantees and transparent assumptions. The authors demonstrate its utility by testing several state‑of‑the‑art methods under diverse conditions, both within and outside the identification regime, highlighting the insights CausalProfiler can reveal.

By Panayiotis Panayiotou, Audrey Poinsot, Alessandro Leite, Nicolas Chesneau, Marc Schoenauer, \"Ozg\"ur \c{S}im\c{s}ek