Tutorials › Causal Inference › Multiple Imputation for Causal Effects

Missing Data and Measurement Error · Tutorial 905 of 972

Multiple Imputation for Causal Effects

Build causal multiple-imputation analyses that preserve treatment–confounder–outcome relationships and correctly pool effect estimates and their uncertainty.

Advanced 12 min read

What You'll Learn

  • State the causal and missing-data assumptions needed for multiple imputation.
  • Specify imputation models for missing outcomes and baseline confounders.
  • Estimate a common causal contrast separately in each completed dataset.
  • Calculate pooled estimates, standard errors, and confidence intervals using Rubin’s rules.
  • Distinguish within-imputation uncertainty from between-imputation variability.
  • Recognize why a single imputed dataset or an incompatible imputation model can mislead.

Why Multiple Imputation Is Used in Causal Analysis

When outcomes or confounders are missing, analyzing only records with complete data can change the population being compared and can induce selection bias. Multiple imputation addresses this by replacing each missing value with several plausible values, creating \(m\) completed datasets. The same causal analysis is then performed in every dataset, and the estimates and their uncertainty are combined.

The goal is not to make missing values known. Rather, the variation across completed datasets represents uncertainty about those values. A causal effect estimate must still rely on assumptions about confounding, positivity, and missingness; imputation does not by itself make an observational comparison causal.

Definition: In multiple imputation, each missing value is drawn or otherwise generated from an imputation model conditional on appropriate observed information. This produces \(m\) completed datasets. Let \(Q_i\) be the causal-effect estimate and \(U_i\) its estimated sampling variance in completed dataset \(i\), for \(i=1,\ldots,m\). Rubin’s rules combine these quantities into one estimate and variance.

Specify the Causal Target and Required Assumptions

Let \(A\) denote treatment, \(Y\) the outcome, and \(C\) measured baseline confounders. For a binary treatment, one common target is the average treatment effect (ATE), \(\psi=E[Y(1)-Y(0)]\), where \(Y(a)\) is the potential outcome under treatment level \(a\). Under consistency, conditional exchangeability \(Y(a)\perp A\mid C\), and positivity \(P(A=a\mid C=c)>0\) for covariate patterns in the target population, this estimand can be identified by standardization:

$$ \psi = E_C\{E(Y\mid A=1,C)-E(Y\mid A=0,C)\}. $$

Missing data add assumptions. A common working condition is missing at random (MAR): conditional on variables observed for a person, the probability a value is missing does not depend on its unobserved value. For example, an outcome may be MAR if its missingness depends on treatment, observed baseline severity, and observed auxiliary variables, but not on the unobserved outcome after conditioning on those variables. MAR is an assumption about the data-generating and observation processes, not a conclusion established merely by fitting an imputation model.

A valid analysis also requires imputation models that adequately represent the relevant conditional distributions and preserve the relationships needed by the causal analysis. Imputation should reflect uncertainty in model parameters and missing values; simply replacing each value by a fitted conditional mean is generally not proper multiple imputation. The causal identification assumptions and the missing-data assumptions do different work: the former identify the effect in the absence of missingness, while the latter support recovering the necessary information despite missing values.

Constructing Imputations for Causal Variables

The imputation model should include treatment, outcome, confounders, and useful observed auxiliary variables that predict missingness or the missing values. If a confounder is missing, treatment and outcome information can be important predictors of that confounder. If the outcome is missing, treatment and the confounders used in the causal analysis should generally be included. Leaving out treatment or important confounders can weaken the association structure needed to estimate the causal contrast.

The model also needs to preserve the form of the substantive analysis. For example, if the outcome model includes a treatment-by-confounder interaction or a nonlinear covariate effect, an imputation procedure that imposes incompatible relationships can distort the resulting treatment contrast. Categorical variables need models suitable for categorical values; continuous variables require distributions appropriate to their support and shape. With several variables incomplete, imputations may be generated jointly or through a sequence of conditional models. In either case, the resulting completed datasets should represent uncertainty rather than a single “best guess.”

Once each dataset is complete, apply the same estimand, target population, and analysis procedure in every dataset. For an ATE, that might mean fitting the same outcome regression and standardizing predicted outcomes to the same covariate distribution in each dataset. Use a variance estimator \(U_i\) that corresponds to that completed-data estimator. Do not average raw imputations first and analyze only the averaged data: that discards the between-imputation uncertainty.

Worked Example: Imputing Missing Outcomes Before Estimating an Effect

Hypothetical setting. A randomized study assigns 120 people equally to treatment and control. The outcome is symptom improvement in points, with higher values indicating greater improvement. Some follow-up outcomes are missing. Because assignment is randomized, an unadjusted difference in mean improvement estimates the intention-to-treat contrast, subject to appropriate handling of missing outcomes. Suppose MAR is plausible conditional on assignment, baseline symptom score, and an observed early symptom measure, and a suitably specified predictive model includes these variables.

For each missing outcome, the imputation procedure draws a value from its predictive distribution, including uncertainty in the model. Repeat the procedure to create \(m=5\) completed datasets. In each dataset, estimate the treatment-minus-control mean difference and its sampling variance. Suppose the five completed-data analyses give the following results:

Dataset \(i\)Effect estimate \(Q_i\) (points)Variance \(U_i\) (points squared)
12.01.00
23.01.00
32.01.00
44.01.00
53.01.00

The pooled estimate is the average of the five estimates:

$$ \bar{Q}=\frac{2+3+2+4+3}{5}=\frac{14}{5}=2.8\text{ points}. $$

The average within-imputation variance is \(\bar{U}=(1+1+1+1+1)/5=1.00\) points squared. The between-imputation variance is the sample variance of the five estimates:

$$ B=\frac{(2-2.8)^2+(3-2.8)^2+(2-2.8)^2+(4-2.8)^2+(3-2.8)^2}{5-1} =\frac{2.80}{4}=0.700\text{ points}^2. $$

The total variance is \(\bar{U}+(1+1/m)B=1.00+1.2(0.700)=1.84\) points squared, so the pooled standard error is \(\sqrt{1.84}=1.356\) points. Using the conventional large-complete-data-df Rubin approximation, the degrees of freedom are approximately \( (m-1)[1+\bar{U}/\{(1+1/m)B\}]^2=4[1+1/(1.2\times0.700)]^2=19.2\). A \(t\)-critical value of about 2.093 gives the 95% interval \(2.8\pm2.093(1.356)=(-0.039,5.639)\) points. This interval includes zero, so the imputed analysis does not rule out no average treatment effect at the 5% level. The analysis estimates an assignment effect under the stated assumptions; the numerical result does not establish those assumptions.

Rubin’s Rules and Why They Have This Form

Rubin’s rules: Given \(m\) estimates \(Q_i\) and corresponding variances \(U_i\), define the pooled estimate \(\bar{Q}\), within-imputation variance \(\bar{U}\), between-imputation variance \(B\), and total variance \(T\) as follows.
$$ \bar{Q}=\frac{1}{m}\sum_{i=1}^{m}Q_i,\qquad \bar{U}=\frac{1}{m}\sum_{i=1}^{m}U_i,\qquad B=\frac{1}{m-1}\sum_{i=1}^{m}(Q_i-\bar{Q})^2, $$
$$ T=\bar{U}+\left(1+\frac{1}{m}\right)B,\qquad SE(\bar{Q})=\sqrt{T}. $$

The within-imputation component, \(\bar{U}\), reflects sampling uncertainty that would remain if the completed data were observed. The between-imputation component, \(B\), estimates how much the effect estimate varies across imputations because the missing values are uncertain. The factor \(1+1/m\) incorporates the additional finite-imputation contribution to the variance of the pooled estimate. In particular, \(B\) is the sample variance across the \(m\) estimates; it is not, on average, a downward-biased estimate of the underlying imputation-distribution variance.

The variance formula follows from a simple decomposition. If the complete-data estimate has conditional variance represented by \(U_i\), uncertainty in the missing data contributes variation in the conditional estimate across imputations. Averaging \(m\) imputed estimates leaves finite simulation uncertainty, represented by \(B/m\), in addition to the between-imputation component \(B\). Combining the within-data variance and this uncertainty yields:

$$ T=\bar{U}+B+\frac{B}{m} =\bar{U}+\left(1+\frac{1}{m}\right)B. $$

For inference, a common large-complete-data-sample approximation uses a \(t\) distribution with degrees of freedom

$$ \nu=(m-1)\left(1+\frac{\bar{U}}{(1+1/m)B}\right)^2. $$

A confidence interval is then \(\bar{Q}\pm t_{0.975,\nu}\sqrt{T}\) for a 95% interval. When \(B=0\), this expression for degrees of freedom is not used as written; uncertainty from imputation is absent in the observed set of imputations, and inference needs an appropriate complete-data or finite-sample degrees-of-freedom treatment. In small samples, refinements that account for finite complete-data degrees of freedom may be preferable.

Worked Example: Imputing a Missing Confounder

Hypothetical setting. In an observational cohort, treatment \(A\) and outcome \(Y\) are observed for everyone, but baseline severity \(C\), a confounder, is missing for some participants. Suppose \(C\) is binary and its missingness is plausibly MAR conditional on observed treatment, outcome, age, and an auxiliary baseline measure. A logistic imputation model for \(C\) includes those observed predictors and draws missing \(C\) values from the fitted predictive distribution, with parameter uncertainty included.

Create \(m=10\) completed datasets. In each dataset, estimate the ATE by the same prespecified outcome-regression standardization over the cohort, using \(A\), \(C\), and any specified interaction in the same way. Suppose the effect estimates, in outcome points, average to \(\bar{Q}=1.60\), the average complete-data variance is \(\bar{U}=0.36\) points squared, and the sample variance of the ten estimates is \(B=0.09\) points squared. Then

$$ T=0.36+\left(1+\frac{1}{10}\right)(0.09) =0.36+0.099=0.459\text{ points}^2, \qquad SE=\sqrt{0.459}=0.6775\text{ points}. $$

The large-complete-data-df approximation gives \(\nu=9[1+0.36/\{1.1(0.09)\}]^2=9(4.636)^2\approx193.4\). The corresponding 95% interval is approximately \(1.60\pm1.972(0.6775)=(0.264,2.936)\) points. Under the causal and missing-data assumptions, this estimates a positive average treatment effect in the target cohort. The interval reflects both sampling variation within completed datasets and variability across the confounder imputations.

Worked Example: Why One Imputation Does Not Pool Correctly

Hypothetical setting. Suppose a causal analysis produces five estimates \(Q_i=(1.2,1.8,1.5,2.1,1.4)\) points, and each completed-data analysis has variance \(U_i=0.25\) points squared. Their mean is \(\bar{Q}=(1.2+1.8+1.5+2.1+1.4)/5=1.60\) points. The squared deviations sum to \(0.50\), so \(B=0.50/4=0.125\) points squared and \(\bar{U}=0.25\) points squared.

Pooling gives \(T=0.25+1.2(0.125)=0.400\) points squared and \(SE=\sqrt{0.400}=0.6325\) points. If an analyst instead selects one completed dataset and reports only its standard error, the reported variance is \(0.25\) points squared and the standard error is \(0.500\) points. That calculation omits the additional uncertainty indicated by the differences among the five estimates. It is not a valid substitute for pooling; nor does selecting one imputation turn its estimate into the pooled estimate of 1.60 points.

Common Pitfalls and Practical Interpretation

Multiple imputation is useful because it carries missing-value uncertainty into the causal estimate, but it is not a universal correction for missing data. Its conclusions depend on the imputation model, the MAR assumption, and the causal identification assumptions. If missingness depends on an unobserved value even after conditioning on observed information, standard MAR imputations may not represent the missing-data process. That concern requires additional assumptions and analyses, rather than a claim that pooling has resolved it.

  • Imputing without treatment or outcome information. A confounder imputation model that ignores relevant treatment or outcome associations can fail to preserve the structures needed for adjustment. Include variables that are substantively relevant and observed.
  • Using an incompatible model. An imputation model that omits a substantive interaction or imposes an unsuitable distribution can distort the causal contrast. Align imputation and analysis models as far as practicable.
  • Treating imputed values as observed truth. A completed dataset is one plausible version of the missing data, not a recovered record. Use all \(m\) analyses and include between-imputation uncertainty.
  • Assuming MAR because a model was fitted. MAR is not testable from observed data alone in general. A well-fitted model does not prove the assumption.
  • Changing the estimand across imputations. Use the same target population, causal contrast, and analysis procedure in all completed datasets so that the \(Q_i\) estimates the same quantity.

The central practical rule is to define the causal estimand first, specify an imputation strategy compatible with its analysis, estimate that same contrast in every completed dataset, and pool both estimates and uncertainty. The resulting confidence interval describes sampling and imputation uncertainty conditional on the modeling and identification assumptions; it does not include uncertainty about whether those assumptions are true.

Check Your Understanding

Use the definitions and calculations in this tutorial to answer the following questions.

  1. Why should treatment and relevant outcome information generally be considered when imputing a missing baseline confounder?
  2. In Rubin’s rules, what do \(\bar{U}\) and \(B\) represent, and why does the total variance include the factor \(1+1/m\) multiplying \(B\)?
  3. For estimates \(Q_1=1\), \(Q_2=2\), and \(Q_3=3\), calculate \(\bar{Q}\) and \(B\).
  4. Why is the standard error from one completed dataset not a valid pooled standard error for a multiple-imputation analysis?
  5. State one causal assumption and one missing-data assumption required for a causal effect analysis using multiple imputation.
Clinical Biostats