arXiv Machine Learning

TLRNet: Estimating Individual Treatment Effect based on Local Information and Single Learner Structure

arXiv:2607. 22762v1 Announce Type: cross Abstract: Causal inference has become a central issue across various fields, including computer science, statistics, economics, education, healthcare, and medicine.

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 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
Aug 5

Causal Inference with Unstructured Outcomes

arXiv:2608. 03085v1 Announce Type: cross Abstract: Causal inference has traditionally centered on scalar outcomes: whether a patient recovers, how much a worker earns, or how many visits a website receives.

By Kevin Christian Wibisono, Yixin Wang
arXiv Statistics ML
6d 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 Machine Learning
Sep 24

Artificial intelligence surrogates for treatment effect estimation with before-and-after data

The paper proposes a method for estimating treatment effects using AI-generated predictions as surrogates, applied to paired before-and-after measurements for each treated individual. By comparing AI predictions before and after treatment, the approach can identify the average treatment effect on the treated under certain technical assumptions, even when clinical outcomes are never observed for treated subjects. When assumptions are questionable, the authors introduce prediction‑powered inference that corrects bias with a small set of observed outcomes, and validate the method with synthetic and cardio‑oncology data.

By Frances Dean, Anna Neufeld, Joshua Barrios, Geoffrey H Tison, Ahmed Alaa