CATE Estimation With Potential Outcome Imputation From Local Regression
Not on the map. Why?
Abstract (arXiv)
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 for CATE estimation. First, we derive regret bounds for general data augmentation methods suggesting that a small imputation error may be necessary for accurate CATE estimation. Inspired by this idea, we propose a contrastive learning approach that reliably imputes missing potential outcomes for a selected subset of individuals formed using a similarity measure. We augment the original dataset with these reliable imputations to reduce the discrepancy between different treatment groups while inducing minimal imputation error. The augmented dataset can subsequently be employed to train standard CATE estimation models. We provide both theoretical guarantees and extensive numerical studies demonstrating the effectiveness of our approach in improving the accuracy and robustness of numerous CATE estimation models.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 20 of its 49 references that are in the map. First posted on arXiv 7 Nov 2023.
Builds on
20 of 49 references.
References · 20
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
- 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,513
- 2020Unsupervised Learning of Visual Features by Contrasting Cluster Assignmentscited by 2,857
- 2018Unsupervised Feature Learning via Non-Parametric Instance-level Discriminationcited by 1,845
- 2020What Makes for Good Views for Contrastive Learning?cited by 846
- 2019Self-Supervised Learning of Pretext-Invariant Representationscited by 808
- 2017Meta-learners for Estimating Heterogeneous Treatment Effects using Machine Learningcited by 478
- 2017Causal Effect Inference with Deep Latent-Variable Modelscited by 410
- 2021A Survey of Data Augmentation Approaches for NLPcited by 409
- 2016Learning Representations for Counterfactual Inferencecited by 329
- 2014Discriminative Unsupervised Feature Learning with Exemplar Convolutional Neural Networkscited by 300
- 2015Recursive Partitioning for Heterogeneous Causal Effectscited by 255
8 more
- 2017Bayesian Inference of Individualized Treatment Effects using Multi-task Gaussian Processescited by 158
- 2020GridMask Data Augmentationcited by 123
- 2017Unsupervised Learning by Predicting Noisecited by 98
- 2019Uniform Error Bounds for Gaussian Process Regression with Application to Safe Controlcited by 64
- 2018Perfect Match: A Simple Method for Learning Representations For Counterfactual Inference With Neural Networkscited by 56
- 2019Selfie: Self-supervised Pretraining for Image Embeddingcited by 46
- 2021RieszNet and ForestRiesz: Automatic Debiased Machine Learning with Neural Nets and Random Forestscited by 42
- 2023Estimating Causal Effects using a Multi-task Deep Ensemblecited by 7
Built on by
Citing papers per year: 2024: 1; 2025: 1.
2 papers.