Abstract:In this article, we aim to provide a general and complete understanding of semi-supervised (SS) causal inference for treatment effects. Specifically, we consider two such estimands: (a) the average treatment effect and (b) the quantile treatment effect, as prototype cases, in an SS setting, characterized by two available data sets: (i) a labeled data set of size $n$, providing observations for a response and a set of high dimensional covariates, as well as a binary treatment indicator; and (ii) an unlabeled data set of size $N$, much larger than $n$, but without the response observed. Using these two data sets, we develop a family of SS estimators which are ensured to be: (1) more robust and (2) more efficient than their supervised counterparts based on the labeled data set only. Beyond the 'standard' double robustness results (in terms of consistency) that can be achieved by supervised methods as well, we further establish root-n consistency and asymptotic normality of our SS estimators whenever the propensity score in the model is correctly specified, without requiring specific forms of the nuisance functions involved. Such an improvement of robustness arises from the use of the massive unlabeled data, so it is generally not attainable in a purely supervised setting. In addition, our estimators are shown to be semi-parametrically efficient as long as all the nuisance functions are correctly specified. Moreover, as an illustration of the nuisance estimators, we consider inverse-probability-weighting type kernel smoothing estimators involving unknown covariate transformation mechanisms, and establish in high dimensional scenarios novel results on their uniform convergence rates, which should be of independent interest. Numerical results on both simulated and real data validate the advantage of our methods over their supervised counterparts with respect to both robustness and efficiency.

Balancing Method for High Dimensional Causal Inference

Estimating individual treatment effect: generalization bounds and algorithms

Causal Inference with High-dimensional Discrete Covariates

Bounds on Causal Effects and Application to High Dimensional Data

Program Evaluation and Causal Inference with High-Dimensional Data

Causal effect estimation for multivariate continuous treatments

Variational Causal Inference

Causal Inference for a Hidden Treatment

Toward Unique and Unbiased Causal Effect Estimation From Data With Hidden Variables

Contrastive representations of high-dimensional, structured treatments

Optimal Statistical Inference for Individualized Treatment Effects in High-dimensional Models

A General Framework for Treatment Effect Estimation in Semi-Supervised and High Dimensional Settings

Causal Inference under Data Restrictions

Scalable Counterfactual Distribution Estimation in Multivariate Causal Models

Causal Inference: A Missing Data Perspective

Causal Inference with Complex Treatments: A Survey

Statistical Approaches for Causal Inference

Sufficient dimension reduction for average causal effect estimation

Deep End-to-end Causal Inference

DeepMatch: Balancing Deep Covariate Representations for Causal Inference Using Adversarial Training

Bayesian Framework for Causal Inference with Principal Stratification and Clusters