Treatment Effect Estimation using Invariant Risk Minimization
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 the domain generalization framework of invariant risk minimization (IRM). IRM uses data from multiple domains, learns predictors that do not exploit spurious domain-dependent factors, and generalizes better to unseen domains. We propose an IRM-based ITE estimator aimed at tackling treatment assignment bias when there is little support overlap between the control group and the treatment group. We accomplish this by creating diversity: given a single dataset, we split the data into multiple domains artificially. These diverse domains are then exploited by IRM to more effectively generalize regression-based models to data regions that lack support overlap. We show gains over classical regression approaches to ITE estimation in settings when support mismatch is more pronounced.
Code (2)
Tasks
DiversityDomain GeneralizationregressionSimilar Papers 제목 키워드 기반
Invariant Representation Learning for Treatment Effect Estimation
The defining challenge for causal inference from observational data is the presence of `confounders', covariates that affect both treatment assignment and the outcome. To address this challenge, practitioners collect and…
Causal IdentificationCausal InferenceRepresentation LearningvalidRevisiting Invariant Learning for Out-of-Domain Generalization on Multi-Site Mammogram Datasets
Despite significant progress in robust deep learning techniques for mammogram breast cancer classification, their reliability in real-world clinical development settings remains uncertain. The translation of these models…
Cancer ClassificationDomain GeneralizationThe Risks and Rewards of Invariant Risk Minimization
Spurious correlations are one of the most prominent pain points for building and deploying machine learning models. Invariant Risk Minimization (IRM) is a learning algorithm designed to mitigate the effect of spurious fe…
BIG-bench Machine LearningKernelized Heterogeneous Risk Minimization
The ability to generalize under distributional shifts is essential to reliable machine learning, while models optimized with empirical risk minimization usually fail on non-$i.i.d$ testing data. Recently, invariant learn…
Heterogeneous Risk Minimization
Machine learning algorithms with empirical risk minimization usually suffer from poor generalization performance due to the greedy exploitation of correlations among the training data, which are not stable under distribu…