Estimating Conditional Average Treatment Effects with Missing Treatment
Information
- URL: http://arxiv.org/abs/2203.01422v2
- Date: Sun, 16 Apr 2023 11:09:30 GMT
- Title: Estimating Conditional Average Treatment Effects with Missing Treatment
Information
- Authors: Milan Kuzmanovic, Tobias Hatt, Stefan Feuerriegel
- Abstract summary: Estimating conditional average treatment effects (CATE) is challenging when treatment information is missing.
In this paper, we analyze CATE estimation in the setting with missing treatments.
We develop MTRNet, a novel CATE estimation algorithm.
- Score: 20.83151214072516
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: Estimating conditional average treatment effects (CATE) is challenging,
especially when treatment information is missing. Although this is a widespread
problem in practice, CATE estimation with missing treatments has received
little attention. In this paper, we analyze CATE estimation in the setting with
missing treatments where unique challenges arise in the form of covariate
shifts. We identify two covariate shifts in our setting: (i) a covariate shift
between the treated and control population; and (ii) a covariate shift between
the observed and missing treatment population. We first theoretically show the
effect of these covariate shifts by deriving a generalization bound for
estimating CATE in our setting with missing treatments. Then, motivated by our
bound, we develop the missing treatment representation network (MTRNet), a
novel CATE estimation algorithm that learns a balanced representation of
covariates using domain adaptation. By using balanced representations, MTRNet
provides more reliable CATE estimates in the covariate domains where the data
are not fully observed. In various experiments with semi-synthetic and
real-world data, we show that our algorithm improves over the state-of-the-art
by a substantial margin.
Related papers
- Efficient adjustment for complex covariates: Gaining efficiency with
DOPE [56.537164957672715]
We propose a framework that accommodates adjustment for any subset of information expressed by the covariates.
Based on our theoretical results, we propose the Debiased Outcome-adapted Propensity Estorimator (DOPE) for efficient estimation of the average treatment effect (ATE)
Our results show that the DOPE provides an efficient and robust methodology for ATE estimation in various observational settings.
arXiv Detail & Related papers (2024-02-20T13:02:51Z) - Adversarially Balanced Representation for Continuous Treatment Effect
Estimation [6.469020202994118]
In this paper, we consider the more practical and challenging scenario in which the treatment is a continuous variable.
We propose the adversarial counterfactual regression network (ACFR) that adversarially minimizes the representation imbalance in terms of KL divergence.
Our experimental evaluation on semi-synthetic datasets demonstrates the empirical superiority of ACFR over a range of state-of-the-art methods.
arXiv Detail & Related papers (2023-12-17T00:46:16Z) - Counterfactual Data Augmentation with Contrastive Learning [27.28511396131235]
We introduce a model-agnostic data augmentation method that imputes the counterfactual outcomes for a selected subset of individuals.
We use contrastive learning to learn a representation space and a similarity measure such that in the learned representation space close individuals identified by the learned similarity measure have similar potential outcomes.
This property ensures reliable imputation of counterfactual outcomes for the individuals with close neighbors from the alternative treatment group.
arXiv Detail & Related papers (2023-11-07T00:36:51Z) - B-Learner: Quasi-Oracle Bounds on Heterogeneous Causal Effects Under
Hidden Confounding [51.74479522965712]
We propose a meta-learner called the B-Learner, which can efficiently learn sharp bounds on the CATE function under limits on hidden confounding.
We prove its estimates are valid, sharp, efficient, and have a quasi-oracle property with respect to the constituent estimators under more general conditions than existing methods.
arXiv Detail & Related papers (2023-04-20T18:07:19Z) - Generalization bounds and algorithms for estimating conditional average
treatment effect of dosage [13.867315751451494]
We investigate the task of estimating the conditional average causal effect of treatment-dosage pairs from a combination of observational data and assumptions on the causal relationships in the underlying system.
This has been a longstanding challenge for fields of study such as epidemiology or economics that require a treatment-dosage pair to make decisions.
We show empirically new state-of-the-art performance results across several benchmark datasets for this problem.
arXiv Detail & Related papers (2022-05-29T15:26:59Z) - Robust and Agnostic Learning of Conditional Distributional Treatment
Effects [62.44901952244514]
The conditional average treatment effect (CATE) is the best point prediction of individual causal effects.
In aggregate analyses, this is usually addressed by measuring distributional treatment effect (DTE)
We provide a new robust and model-agnostic methodology for learning the conditional DTE (CDTE) for a wide class of problems.
arXiv Detail & Related papers (2022-05-23T17:40:31Z) - Treatment Effect Risk: Bounds and Inference [58.442274475425144]
Since the average treatment effect measures the change in social welfare, even if positive, there is a risk of negative effect on, say, some 10% of the population.
In this paper we consider how to nonetheless assess this important risk measure, formalized as the conditional value at risk (CVaR) of the ITE distribution.
Some bounds can also be interpreted as summarizing a complex CATE function into a single metric and are of interest independently of being a bound.
arXiv Detail & Related papers (2022-01-15T17:21:26Z) - Assessment of Treatment Effect Estimators for Heavy-Tailed Data [70.72363097550483]
A central obstacle in the objective assessment of treatment effect (TE) estimators in randomized control trials (RCTs) is the lack of ground truth (or validation set) to test their performance.
We provide a novel cross-validation-like methodology to address this challenge.
We evaluate our methodology across 709 RCTs implemented in the Amazon supply chain.
arXiv Detail & Related papers (2021-12-14T17:53:01Z) - SurvITE: Learning Heterogeneous Treatment Effects from Time-to-Event
Data [83.50281440043241]
We study the problem of inferring heterogeneous treatment effects from time-to-event data.
We propose a novel deep learning method for treatment-specific hazard estimation based on balancing representations.
arXiv Detail & Related papers (2021-10-26T20:13:17Z) - Evaluating (weighted) dynamic treatment effects by double machine
learning [0.12891210250935145]
We consider evaluating the causal effects of dynamic treatments in a data-driven way under a selection-on-observables assumption.
We make use of so-called Neyman-orthogonal score functions, which imply the robustness of treatment effect estimation to moderate (local) misspecifications.
We demonstrate that the estimators are regularityally normal and $sqrtn$-consistent under specific conditions.
arXiv Detail & Related papers (2020-12-01T09:55:40Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.