Inverse probability weighting

Inverse probability weighting is a statistical technique for estimating quantities related to a population other than the one from which the data was collected. Study designs with a disparate sampling population and population of target inference (target population) are common in application. There may be prohibitive factors barring researchers from directly sampling from the target population such as cost, time, or ethical concerns. A solution to this problem is to use an alternate design strategy, e.g. stratified sampling. Weighting, when correctly applied, can potentially improve the efficiency and reduce the bias of unweighted estimators.

One very early weighted estimator is the Horvitz–Thompson estimator of the mean. When the sampling probability is known, from which the sampling population is drawn from the target population, then the inverse of this probability is used to weight the observations. This approach has been generalized to many aspects of statistics under various frameworks. In particular, there are weighted likelihoods, weighted estimating equations, and weighted probability densities from which a majority of statistics are derived. These applications codified the theory of other statistics and estimators such as marginal structural models, the standardized mortality ratio, and the EM algorithm for coarsened or aggregate data.

Inverse probability weighting is also used to account for missing data when subjects with missing data cannot be included in the primary analysis. With an estimate of the sampling probability, or the probability that the factor would be measured in another measurement, inverse probability weighting can be used to inflate the weight for subjects who are under-represented due to a large degree of missing data.

Inverse Probability Weighted Estimator (IPWE)
The inverse probability weighting estimator can be used to demonstrate causality when the researcher cannot conduct a controlled experiment but has observed data to model. Because it is assumed that the treatment is not randomly assigned, the goal is to estimate the counterfactual or potential outcome if all subjects in population were assigned either treatment.

Suppose observed data are $$\{\bigl(X_i,A_i,Y_i\bigr)\}^{n}_{i=1}$$ drawn i.i.d (independent and identically distributed) from unknown distribution P, where The goal is to estimate the potential outcome, $$Y^{*}\bigl(a\bigr)$$, that would be observed if the subject were assigned treatment $$a$$. Then compare the mean outcome if all patients in the population were assigned either treatment: $$\mu_{a} = \mathbb{E}Y^{*}(a)$$. We want to estimate $$\mu_a$$ using observed data $$\{\bigl(X_i,A_i,Y_i\bigr)\}^{n}_{i=1}$$.
 * $$X \in \mathbb{R}^{p}$$ covariates
 * $$A \in \{0, 1\}$$ are the two possible treatments.
 * $$Y \in \mathbb{R}$$ response
 * We do not assume treatment is randomly assigned.

Estimator Formula
"$\hat{\mu}^{IPWE}_{a,n} = \frac{1}{n}\sum^{n}_{i=1}Y_{i} \frac{\mathbf 1_{A_{i}=a}}{\hat{p}_{n}(A_{i}|X_{i})}$|undefined"

Constructing the IPWE
With the mean of each treatment group computed, a statistical t-test or ANOVA test can be used to judge difference between group means and determine statistical significance of treatment effect.
 * 1) $$\mu_{a} = \mathbb{E}\left[\frac{\mathbf{1}_{A=a} Y}{p(A|X)} \right]$$   where $$p(a|x) = \frac{P(A=a,X=x)}{P(X=x)}$$
 * 2) construct $$\hat{p}_{n}(a|x)$$ or $$p(a|x)$$ using any propensity model (often a logistic regression model)
 * 3) $$\hat{\mu}^{IPWE}_{a,n} = \sum^{n}_{i=1}\frac{Y_{i} 1_{A_{i}=a}}{n\hat{p}_{n}(A_{i}|X_{i})}$$

Assumptions
Recall the joint probability model $$(X,A,Y)\sim P$$ for the covariate $$X$$, action $$A$$, and response $$Y$$. If $$X$$ and $$A$$ are known as $$x$$ and $$a$$, respectively, then the response $$Y(X=x, A=a)=Y(x, a)$$ has the distribution

$$\begin{align} Y(x, a) \sim \frac{P(x,a,\cdot)}{\int P(x,a,y) \, dy}. \end{align}$$

We make the following assumptions. \qquad \mathbb{E}_{(A,Y)}\left[ f(Y(X,a))\, g(A)\,|\, X\right] =\mathbb{E}_{Y}\left[ f(Y(X,a))\,|\, X\right] \, \mathbb{E}_{A}\left[ g(A)\,|\, X\right]. \end{align}$$This means that treatment assignment is based solely on covariate data and independent of potential outcomes.
 * (A1) Consistency: $$Y = Y^{*}(A)$$
 * (A2) No unmeasured confounders: $$\{Y^{*}(0), Y^{*}(1)\} \perp A|X$$. More formally, for each bounded and measurable functions $$f$$ and $$g$$, $$\begin{align}
 * (A3) Positivity: $$P(A=a|X=x)=\mathbb{E}_{A}[\mathbf{1}(A=a)\,|\, X=x] >0 $$ for all $$a$$ and $$x$$.

Formal derivation
Under the assumptions (A1)-(A3), we will derive the following identities

$$\begin{align} \mathbb{E}\left[ Y^{*}(a) \right] = \mathbb{E}_{(X,Y)}\left[ Y(X,a)\right]  =   \mathbb{E}_{(X,A,Y)}\left[ \frac{  Y \mathbf{1}(A=a) }{ P(A=a|X)} \right]. \qquad \cdots \cdots (*) \end{align}$$

The first equality follows from the definition and (A1). For the second equality, first use the iterated expectation to write

$$\begin{align} \mathbb{E}_{(X,Y)}\left[ Y(X,a)\right] = \mathbb{E}_{X}\left[ \mathbb{E}_{Y}\left[ Y(X,a) \,|\, X \right] \right]. \end{align}$$

By (A3), $$\mathbb{E}_{A}[\mathbf{1}(A=a)\,|\, X] >0 $$ almost surely. Then using (A2), note that

$$\begin{align} \mathbb{E}_{Y}\left[ Y(X,a) \,|\, X \right] &= \frac{ \mathbb{E}_{Y}\left[ Y(X,a) \,|\, X \right] \, \mathbb{E}_{A}[\mathbf{1}(A=a)\,|\, X ]}{ \mathbb{E}_{A}[\mathbf{1}(A=a)\,|\, X ]} \\ &= \frac{ \mathbb{E}_{(A,Y)}\left[ Y(X,a) \mathbf{1}(A=a)\,|\, X \right]}{ \mathbb{E}[\mathbf{1}(A=a)\,|\, X ]} \qquad \cdots \cdots (A2) \\ &= \mathbb{E}_{(A,Y)}\left[ \frac{ Y(X,a) \mathbf{1}(A=a) }{ \mathbb{E}[\mathbf{1}(A=a)\,|\, X ]} \,\bigg|\, X\right]. \qquad \cdots \cdots (\text{denominator is a function of X}) \end{align}$$

Hence integrating out the last expression with respect to $$X$$ and noting that $$Y(X,a) \mathbf{1}(A=a) = Y(X,A)\mathbf{1}(A=a)= Y\,\mathbf{1}(A=a)$$ almost surely, the second equality in $$(*) $$ follows.

Variance reduction
The Inverse Probability Weighted Estimator (IPWE) is known to be unstable if some estimated propensities are too close to 0 or 1. In such instances, the IPWE is dominated by a small number of subjects with large weights. Recently developed smoothed IPW estimators by employing Rao-Blackwellization, however, reduce the variance of IPWE by up to 7-fold and can also protect the augmented inverse probability weighted estimator from model misspecification.

Augmented Inverse Probability Weighted Estimator (AIPWE)
An alternative estimator is the augmented inverse probability weighted estimator (AIPWE) combines both the properties of the regression based estimator and the inverse probability weighted estimator. It is therefore a 'doubly robust' method in that it only requires either the propensity or outcome model to be correctly specified but not both. This method augments the IPWE to reduce variability and improve estimate efficiency. This model holds the same assumptions as the Inverse Probability Weighted Estimator (IPWE).

Estimator Formula
$$ \begin{align}

\hat{\mu}^{AIPWE}_{a,n}

&= \frac{1}{n} \sum_{i=1}^n\Biggl(\frac{Y_{i}1_{A_{i}=a}}{\hat{p}_{n}(A_{i}|X_{i})} - \frac{1_{A_{i}=a}-\hat{p}_n(A_i|X_i)}{\hat{p}_n(A_i|X_i)}\hat{Q}_n(X_i,a)\Biggr) \\

&= \frac{1}{n} \sum_{i=1}^n\Biggl(\frac{1_{A_{i}=a}}{\hat{p}_{n}(A_{i}|X_{i})}Y_{i} + (1-\frac{1_{A_{i}=a}}{\hat{p}_{n}(A_{i}|X_{i})})\hat{Q}_n(X_i,a)\Biggr) \\

&= \frac{1}{n}\sum_{i=1}^n\Biggl(\hat{Q}_n(X_i,a)\Biggr) +

\frac{1}{n}\sum_{i=1}^n\frac{1_{A_{i}=a}}{\hat{p}_{n}(A_{i}|X_{i})}\Biggl(Y_{i} - \hat{Q}_n(X_i,a)\Biggr)

\end{align} $$

With the following notations:
 * 1) $$1_{A_{i}=a}$$ is an indicator function if subject i is part of treatment group a (or not).
 * 2) Construct regression estimator $$\hat{Q}_n(x,a)$$ to predict outcome $$Y$$ based on covariates $$X$$ and treatment $$A$$, for some subject i. For example, using ordinary least squares regression.
 * 3) Construct propensity (probability) estimate $$\hat{p}_n(A_i|X_i)$$. For example, using logistic regression.
 * 4) Combine in AIPWE to obtain $$\hat{\mu}^{AIPWE}_{a,n}$$

Interpretation and "double robustness"
The later rearrangement of the formula helps reveal the underlying idea: our estimator is based on the average predicted outcome using the model (i.e.: $$\frac{1}{n}\sum_{i=1}^n\Biggl(\hat{Q}_n(X_i,a)\Biggr)$$). However, if the model is biased, then the residuals of the model will not be (in the full treatment group a) around 0. We can correct this potential bias by adding the extra term of the average residuals of the model (Q) from the true value of the outcome (Y) (i.e.: $$\frac{1}{n}\sum_{i=1}^n\frac{1_{A_{i}=a}}{\hat{p}_{n}(A_{i}|X_{i})}\Biggl(Y_{i} - \hat{Q}_n(X_i,a)\Biggr)$$). Because we have missing values of Y, we give weights to inflate the relative importance of each residual (these weights are based on the inverse propensity, a.k.a. probability, of seeing each subject observations) (see page 10 in ).

The "doubly robust" benefit of such an estimator comes from the fact that it's sufficient for one of the two models to be correctly specified, for the estimator to be unbiased (either $$\hat{Q}_n(X_i,a)$$ or $$\hat{p}_{n}(A_{i}|X_{i})$$, or both). This is because if the outcome model is well specified then its residuals will be around 0 (regardless of the weights each residual will get). While if the model is biased, but the weighting model is well specified, then the bias will be well estimated (And corrected for) by the weighted average residuals.

The bias of the doubly robust estimators is called a second-order bias, and it depends on the product of the difference $$\frac{1}{\hat{p}_{n}(A_{i}|X_{i})} - \frac{1}{{p}_{n}(A_{i}|X_{i})}$$ and the difference $$\hat{Q}_n(X_i,a) - Q_n(X_i,a)$$. This property allows us, when having a "large enough" sample size, to lower the overall bias of doubly robust estimators by using machine learning estimators (instead of parametric models).