English
Related papers

Related papers: Adversarially Balanced Representation for Continuo…

200 papers

Due to the limited availability of medical data, deep learning approaches for medical image analysis tend to generalise poorly to unseen data. Augmenting data during training with random transformations has been shown to help and became a…

Image and Video Processing · Electrical Eng. & Systems 2022-10-04 Tian Xia , Pedro Sanchez , Chen Qin , Sotirios A. Tsaftaris

Estimating heterogeneous treatment effects in network settings is complicated by interference, meaning that the outcome of an instance can be influenced by the treatment status of others. Existing causal machine learning approaches usually…

Machine Learning · Computer Science 2025-10-27 Daan Caljon , Jente Van Belle , Wouter Verbeke

Methods for estimating heterogeneous treatment effect in observational data have largely focused on continuous or binary outcomes, and have been relatively less vetted with survival outcomes. Using flexible machine learning methods in the…

Applications · Statistics 2021-07-09 Liangyuan Hu , Jiayi Ji , Fan Li

Machine learning can help personalized decision support by learning models to predict individual treatment effects (ITE). This work studies the reliability of prediction-based decision-making in a task of deciding which action $a$ to take…

Machine Learning · Statistics 2019-06-07 Iiris Sundin , Peter Schulam , Eero Siivola , Aki Vehtari , Suchi Saria , Samuel Kaski

Estimating conditional average treatment effect from observational data is highly challenging due to the existence of treatment selection bias. Prevalent methods mitigate this issue by aligning distributions of different treatment groups in…

Machine Learning · Computer Science 2025-05-07 Hao Wang , Zhichao Chen , Jiajun Fan , Haoxuan Li , Tianqiao Liu , Weiming Liu , Quanyu Dai , Yichao Wang , Zhenhua Dong , Ruiming Tang

Given a dataset of individuals each described by a covariate vector, a treatment, and an observed outcome on the treatment, the goal of the individual treatment effect (ITE) estimation task is to predict outcome changes resulting from a…

Machine Learning · Computer Science 2024-06-07 Lokesh Nagalapatti , Pranava Singhal , Avishek Ghosh , Sunita Sarawagi

We study two-way-fixed-effects regressions (TWFE) with several treatment variables. Under a parallel trends assumption, we show that the coefficient on each treatment identifies a weighted sum of that treatment's effect, with possibly…

Econometrics · Economics 2023-04-18 Clément de Chaisemartin , Xavier D'Haultfœuille

A key to causal inference with observational data is achieving balance in predictive features associated with each treatment type. Recent literature has explored representation learning to achieve this goal. In this work, we discuss the…

Machine Learning · Statistics 2021-02-25 Serge Assaad , Shuxi Zeng , Chenyang Tao , Shounak Datta , Nikhil Mehta , Ricardo Henao , Fan Li , Lawrence Carin

Counterfactual prediction is about predicting outcome of the unobserved situation from the data. For example, given patient is on drug A, what would be the outcome if she switch to drug B. Most of existing works focus on modeling…

Machine Learning · Computer Science 2020-10-29 Yanbo Xu , Cao Xiao , Jimeng Sun

Counterfactual estimation over time is important in various applications, such as personalized medicine. However, time-dependent confounding bias in observational data still poses a significant challenge in achieving accurate and efficient…

Machine Learning · Computer Science 2026-03-13 Nghia D. Nguyen , Pablo Robles-Granda , Lav R. Varshney

We propose novel estimators for categorical and continuous treatments by using an optimal covariate balancing strategy for inverse probability weighting. The resulting estimators are shown to be consistent and asymptotically normal for…

Methodology · Statistics 2025-09-08 Seong-ho Lee , Yanyuan Ma , Xavier de Luna

Determining causal effects of interventions onto outcomes from real-world, observational (non-randomized) data, e.g., treatment repurposing using electronic health records, is challenging due to underlying bias. Causal deep learning has…

Machine Learning · Computer Science 2025-10-23 Shantanu Ghosh , Zheng Feng , Jiang Bian , Kevin Butler , Mattia Prosperi

Individualized treatment regimes (ITRs) aim to improve clinical outcomes by assigning treatment based on patient-specific characteristics. However, existing methods often struggle with high-dimensional covariates, limiting accuracy,…

Machine Learning · Statistics 2026-01-13 Sungtaek Son , Eardi Lila , Kwun Chuen Gary Chan

Estimating the individual treatment effect (ITE) from observational data is meaningful and practical in healthcare. Existing work mainly relies on the strong ignorability assumption that no hidden confounders exist, which may lead to bias…

Methodology · Statistics 2020-12-16 Ruoqi Liu , Changchang Yin , Ping Zhang

Causal inference often relies on the counterfactual framework, which requires that treatment assignment is independent of the outcome, known as strong ignorability. Approaches to enforcing strong ignorability in causal analyses of…

Machine Learning · Computer Science 2020-12-04 Amelia J. Averitt , Natnicha Vanitchanant , Rajesh Ranganath , Adler J. Perotte

Matching has become the mainstream in counterfactual inference, with which selection bias between sample groups can be significantly eliminated. However in practice, when estimating average treatment effect on the treated (ATT) via…

Econometrics · Economics 2022-06-14 Boyang You , Kerry Papps

One of the most significant challenges in Conditional Average Treatment Effect (CATE) estimation is the statistical discrepancy between distinct treatment groups. To address this issue, we propose a model-agnostic data augmentation method…

Machine Learning · Computer Science 2025-06-17 Ahmed Aloui , Juncheng Dong , Cat P. Le , Vahid Tarokh

Deep neural networks have achieved remarkable performance in various applications but are extremely vulnerable to adversarial perturbation. The most representative and promising methods that can enhance model robustness, such as adversarial…

Computer Vision and Pattern Recognition · Computer Science 2022-08-30 Faqiang Liu , Rong Zhao

We address the Individualized continuous treatment effect (ICTE) estimation problem where we predict the effect of any continuous-valued treatment on an individual using observational data. The main challenge in this estimation task is the…

Machine Learning · Computer Science 2024-01-30 Lokesh Nagalapatti , Akshay Iyer , Abir De , Sunita Sarawagi

The development of causal prediction models is challenged by the fact that the outcome is only observable for the applied (factual) intervention and not for its alternatives (the so-called counterfactuals); in medicine we only know…

Machine Learning · Computer Science 2023-11-29 Stefan Schrod , Fabian Sinz , Michael Altenbuchinger