English
Related papers

Related papers: Causal Representation Learning with Optimal Compre…

200 papers

Estimating individual-level treatment effect from observational data is a fundamental problem in causal inference and has attracted increasing attention in the fields of education, healthcare, and public policy.In this work, we concentrate…

Machine Learning · Computer Science 2025-07-10 Hui Meng , Keping Yang , Xuyu Peng , Bo Zheng

Learning individualized treatment rules (ITRs) is an important topic in precision medicine. Current literature mainly focuses on deriving ITRs from a single source population. We consider the observational data setting when the source…

Machine Learning · Statistics 2023-07-04 Rui Chen , Jared D. Huling , Guanhua Chen , Menggang Yu

Estimation and inference of treatment effects under unconfounded treatment assignments often suffer from bias and the `curse of dimensionality' due to the nonparametric estimation of nuisance parameters for high-dimensional confounders.…

Methodology · Statistics 2025-07-08 Zeqi Wu , Meilin Wang , Wei Huang , Zheng Zhang

Personalized medicine aims to tailor treatments to individual patients, especially when people respond heterogeneously to therapies. A key objective is to learn individualized treatment rules that recommend optimal treatments from patient…

Methodology · Statistics 2026-03-12 Zhu Wang

Conditional average treatment effects (CATEs) allow us to understand the effect heterogeneity across a large population of individuals. However, typical CATE learners assume all confounding variables are measured in order for the CATE to be…

Machine Learning · Computer Science 2022-02-01 Yao Zhang , Jeroen Berrevoets , Mihaela van der Schaar

Causal inference methods for treatment effect estimation usually assume independent units. However, this assumption is often questionable because units may interact, resulting in spillover effects between them. We develop augmented inverse…

Methodology · Statistics 2025-04-08 Corinne Emmenegger , Meta-Lina Spohn , Timon Elmer , Peter Bühlmann

We consider Targeted Maximum Likelihood Estimation (TMLE) of weighted average treatment effects (WATEs), a class of causal estimands that reweight the covariate distribution using a specified function of the propensity score. This class…

Statistics Theory · Mathematics 2026-04-02 Yang Liu , Patrick Lopatto , Ivana Malenica

In this paper, we develop a multiply robust inference procedure of the average treatment effect (ATE) for data with high-dimensional covariates. We consider the case where it is difficult to correctly specify a single parametric model for…

Methodology · Statistics 2025-09-03 Xintao Xia , Yumou Qiu

We study the problem of observational causal inference with continuous treatments in the framework of inverse propensity-score weighting. To obtain stable weights, we design a new algorithm based on entropy balancing that learns weights to…

Machine Learning · Computer Science 2022-07-12 Mohammad Taha Bahadori , Eric Tchetgen Tchetgen , David E. Heckerman

Causal estimands can vary significantly depending on the relationship between outcomes in treatment and control groups, potentially leading to wide partial identification (PI) intervals that impede decision making. Incorporating covariates…

Methodology · Statistics 2025-06-02 Sirui Lin , Zijun Gao , Jose Blanchet , Peter Glynn

This paper introduces entropy balancing for continuous treatments (EBCT) by extending the original entropy balancing methodology of Hainm\"uller (2012). In order to estimate balancing weights, the proposed approach solves a globally convex…

Econometrics · Economics 2020-05-29 Stefan Tübbicke

End-to-end representation learning has become a powerful tool for estimating causal quantities from high-dimensional observational data, but its efficiency remained unclear. Here, we face a central tension: End-to-end representation…

Machine Learning · Computer Science 2026-04-28 Valentyn Melnychuk , Dennis Frauen , Jonas Schweisthal , Stefan Feuerriegel

The preponderance of large-scale healthcare databases provide abundant opportunities for comparative effectiveness research. Evidence necessary to making informed treatment decisions often relies on comparing effectiveness of multiple…

Methodology · Statistics 2020-10-06 Liangyuan Hu , Chenyang Gu

Generating probabilistic forecasts of potential outcomes and individual treatment effects (ITE) is essential for risk-aware decision-making in domains such as healthcare, policy, marketing, and finance. We propose two novel methods: the…

Machine Learning · Computer Science 2025-09-24 Jef Jonkers , Jarne Verhaeghe , Glenn Van Wallendael , Luc Duchateau , Sofie Van Hoecke

In this paper, we focus on estimating the average treatment effect (ATE) of a target population when individual-level data from a source population and summary-level data (e.g., first or second moments of certain covariates) from the target…

Methodology · Statistics 2023-01-18 Rui Chen , Guanhua Chen , Menggang Yu

Automated model selection is often proposed to users to choose which machine learning model (or method) to apply to a given regression task. In this paper, we show that combining different regression models can yield better results than…

Machine Learning · Computer Science 2022-06-24 Patrick Echtenbruck , Martina Echtenbruck , Joost Batenburg , Thomas Bäck , Boris Naujoks , Michael Emmerich

Inferring causal individual treatment effect (ITE) from observational data is a challenging problem whose difficulty is exacerbated by the presence of treatment assignment bias. In this work, we propose a new way to estimate the ITE using…

Machine Learning · Computer Science 2021-03-16 Abhin Shah , Kartik Ahuja , Karthikeyan Shanmugam , Dennis Wei , Kush Varshney , Amit Dhurandhar

Existing weighting methods for treatment effect estimation are often built upon the idea of propensity scores or covariate balance. They usually impose strong assumptions on treatment assignment or outcome model to obtain unbiased…

Machine Learning · Computer Science 2023-05-09 Dongcheng Zhang , Kunpeng Zhang

Data aggregation, also known as meta analysis, is widely used to combine knowledge on parameters shared in common (e.g., average treatment effect) between multiple studies. In this paper, we introduce an attractive data aggregation scheme…

Methodology · Statistics 2023-05-10 Snigdha Panigrahi , Jingshen Wang , Xuming He

Multistate process data are common in studies of chronic diseases such as cancer. These data are ideal for precision medicine purposes as they can be leveraged to improve more refined health outcomes, compared to standard survival outcomes,…

Methodology · Statistics 2022-11-28 Giorgos Bakoyannis