Tutorials › Causal Inference › AIPW and TMLE Under Nuisance-Model Misspecification

Doubly Robust and Semiparametric Methods · Tutorial 485 of 972

AIPW and TMLE Under Nuisance-Model Misspecification

See how nuisance-model misspecification affects AIPW and TMLE consistency, targeted substitution, and efficiency for the average treatment effect.

Advanced 10 min read

What You'll Learn

  • Define double robustness for the ATE and distinguish it from efficiency.
  • Derive the AIPW probability limit when nuisance estimates converge to fixed limits.
  • Explain why targeting yields a substitution estimate but cannot repair two wrong nuisance models.
  • Work through cases with a correct propensity score, a correct outcome regression, and both models misspecified.
  • Identify when an EIF-based efficiency claim requires more than consistency.

Why Misspecification Requires More Than a “Doubly Robust” Label

AIPW and TMLE combine an outcome regression with a propensity score to estimate a causal effect. Their double-robustness property is important: under standard conditions, either nuisance model can be misspecified while the average treatment effect (ATE) estimator remains consistent, provided the other nuisance model is correctly estimated. But this statement is not a guarantee that any combination of working models will succeed. If both nuisance limits are wrong, the estimators can converge to different, incorrect values. And consistency under one correct model does not, by itself, establish that an estimator is efficient.

The distinction is especially visible in TMLE. It updates an outcome regression so its predictions satisfy a targeting equation, then calculates the effect by substituting those updated predictions into the estimand. This substitution structure has useful properties, but the targeting equation is not a mechanism that automatically corrects two misspecified nuisance models.

Setup and the Two Estimators

Let \(A\) be a binary treatment, \(Y\) an outcome, and \(W\) baseline covariates. Write \(Y(1)\) and \(Y(0)\) for the potential outcomes under treatment and control. The target is the population ATE, \(\psi_0=E[Y(1)-Y(0)]\). For identification from observed data, assume consistency, conditional exchangeability \((Y(1),Y(0))\perp A\mid W\), and positivity: \(0<P(A=1\mid W)<1\) for covariate values in the target population. These assumptions identify the ATE as \(E[Q_0(1,W)-Q_0(0,W)]\), where \(Q_0(a,W)=E[Y\mid A=a,W]\).

Definition: The propensity score is \(g_0(W)=P(A=1\mid W)\), and the outcome regression is \(Q_0(a,W)=E[Y\mid A=a,W]\). A nuisance model is correctly specified if its fitted estimate converges to the corresponding true function. It is misspecified if it instead converges to a different limit.

For fitted nuisance estimates \(\hat g\) and \(\hat Q\), the AIPW estimator is

$$ \hat\psi_{\mathrm{AIPW}} = \frac{1}{n}\sum_{i=1}^n \left[ \hat Q(1,W_i)-\hat Q(0,W_i) +\frac{A_i}{\hat g(W_i)}\{Y_i-\hat Q(1,W_i)\} -\frac{1-A_i}{1-\hat g(W_i)}\{Y_i-\hat Q(0,W_i)\} \right]. $$

TMLE starts with an initial estimate of \(Q_0\), updates it using a fluctuation model involving the fitted propensity score, and reports the plug-in estimate

$$ \hat\psi_{\mathrm{TMLE}} = \frac{1}{n}\sum_{i=1}^n \{\hat Q^*(1,W_i)-\hat Q^*(0,W_i)\}, $$

where \(\hat Q^*\) is the targeted outcome regression. In a standard ATE targeting step, the clever covariate is \(A/\hat g(W)-(1-A)/(1-\hat g(W))\). The update is chosen to make the empirical mean of this covariate times the outcome residual \(Y-\hat Q^*(A,W)\) equal to zero, or to solve the corresponding likelihood score equation. The precise fluctuation model depends on the outcome type.

What Double Robustness Says

Consider the probability limit of AIPW when the fitted nuisance functions converge to fixed limits \(g^*\) and \(Q^*\). Conditional on \(W\), the expectation of the AIPW expression is

$$ Q^*(1,W)-Q^*(0,W) +\frac{g_0(W)}{g^*(W)}\{Q_0(1,W)-Q^*(1,W)\} -\frac{1-g_0(W)}{1-g^*(W)}\{Q_0(0,W)-Q^*(0,W)\}. $$

If \(g^*=g_0\), the residual terms correct the difference between the fitted and true outcome regressions, and this conditional expectation reduces to \(Q_0(1,W)-Q_0(0,W)\). If \(Q^*=Q_0\), both residual terms have conditional expectation zero, regardless of whether \(g^*=g_0\). Taking expectations over \(W\) gives the true ATE in either case.

Key result: Under identification, positivity, and appropriate regularity conditions, AIPW and a standard, properly targeted ATE TMLE are consistent if either the propensity score or the outcome regression is consistently estimated. Double robustness does not say that an estimator is consistent when both nuisance limits are wrong.

The same two cases explain TMLE’s double robustness. If the initial outcome regression is correct, its population residual has mean zero conditional on \(A,W\), so the targeting fluctuation has no population residual signal to correct; the regular targeting solution leaves the outcome regression at its truth. If the propensity score is correct, the population targeting equation has a different role: it makes the difference between the targeted plug-in contrast and the true ATE equal to zero. The calculation is given below.

Worked Example: Correct Propensity Score, Misspecified Outcome Regression

Worked Example: Correct Propensity Score, Misspecified Outcome Regression

Suppose there are no covariates, treatment is assigned with probability \(g_0=0.50\), and the hypothetical outcome means are \(Q_0(1)=8\) and \(Q_0(0)=4\). Thus the true ATE is \(8-4=4\) outcome units. Suppose the fitted outcome model converges instead to \(Q^*(1)=7\) and \(Q^*(0)=5\), while the propensity estimate is correct at \(g^*=0.50\).

The AIPW probability limit is

$$ (7-5) +\frac{0.50}{0.50}(8-7) -\frac{0.50}{0.50}(4-5) =2+1+1=4. $$

So the outcome model is wrong, but the AIPW limit equals the true ATE. For TMLE, the population targeting equation under the correct propensity score implies

$$ \{Q_0(1)-Q^*(1)\}-\{Q_0(0)-Q^*(0)\}=0. $$

Consequently, the targeted plug-in contrast equals \(Q_0(1)-Q_0(0)=4\). This is a population consistency argument, not a claim that a finite-sample estimate will equal 4.

Why TMLE’s Targeting Equation Helps—and Where It Stops

For the ATE, let \(Q^*\) denote the population limit of the targeted outcome regression, and suppose the targeting step solves the population version of its score equation:

$$ E\left[ \frac{A}{g^*(W)}\{Y-Q^*(1,W)\} -\frac{1-A}{1-g^*(W)}\{Y-Q^*(0,W)\} \right]=0. $$

If \(g^*=g_0\), apply consistency and exchangeability to the two residual terms. Their expectations become \(E[Q_0(1,W)-Q^*(1,W)]\) and \(E[Q_0(0,W)-Q^*(0,W)]\), respectively. Therefore the targeting equation is

$$ E[Q_0(1,W)-Q^*(1,W)] - E[Q_0(0,W)-Q^*(0,W)] =0. $$

Rearranging shows that \(E[Q^*(1,W)-Q^*(0,W)]=E[Q_0(1,W)-Q_0(0,W)]\). Thus, with a correct propensity score, a targeted plug-in estimate is consistent even if the initial outcome regression was not correct, assuming the targeting procedure reaches a suitable solution.

If \(g^*\) is wrong as well as the initial outcome regression, the weighted residual equation no longer equals the difference in outcome-regression errors. It still imposes a score constraint, but that constraint need not make the plug-in contrast correct. This is the precise limit of the targeting argument: a solved equation is not the same as a correctly identified causal contrast.

Worked Example: Correct Outcome Regression, Misspecified Propensity Score

Worked Example: Correct Outcome Regression, Misspecified Propensity Score

Use the same hypothetical setting: \(g_0=0.50\), \(Q_0(1)=8\), and \(Q_0(0)=4\). Now suppose the outcome regression is correct, but the fitted propensity score converges to \(g^*=0.25\).

The AIPW limit is

$$ (8-4) +\frac{0.50}{0.25}(8-8) -\frac{0.50}{0.75}(4-4) =4+0-0=4. $$

The residual corrections vanish because the outcome regression is correct. For TMLE, the true outcome regression has conditional mean-zero residuals given \(A,W\). Under a regular fluctuation whose population solution is stable, the targeting update therefore does not move the correct regression away from its truth. The plug-in contrast remains 4. In finite samples, the update need not be exactly zero, since observed residuals fluctuate around zero.

Worked Example: Both Nuisance Limits Are Wrong

Worked Example: Both Nuisance Limits Are Wrong

Again take the hypothetical true values \(g_0=0.50\), \(Q_0(1)=8\), and \(Q_0(0)=4\), so the ATE is 4. Let both fitted nuisance limits be wrong: \(g^*=0.25\), \(Q^*(1)=7\), and \(Q^*(0)=5\).

The AIPW probability limit is

$$ (7-5) +\frac{0.50}{0.25}(8-7) -\frac{0.50}{0.75}(4-5) =2+2+\frac{2}{3} =4.667. $$

This limit is not the true ATE of 4. A TMLE targeting update also need not fix the problem. To see this concretely, use a continuous-outcome identity fluctuation \(Q_\epsilon(A)=Q^*(A)+\epsilon H(A)\), with \(H(A)=A/0.25-(1-A)/0.75\). Thus \(H(1)=4\) and \(H(0)=-4/3\). The population score at \(\epsilon=0\) is

$$ E[H(A)\{Y-Q^*(A)\}] =0.50(4)(8-7)+0.50\left(-\frac{4}{3}\right)(4-5) =\frac{8}{3}. $$

The score decreases by \(\epsilon E[H(A)^2]\), where

$$ E[H(A)^2] =0.50(16)+0.50\left(\frac{16}{9}\right) =\frac{80}{9}. $$

Setting the score to zero gives \(\epsilon=(8/3)/(80/9)=0.30\). The targeted predictions are \(Q^*(1)+0.30(4)=8.2\) and \(Q^*(0)+0.30(-4/3)=4.6\). The resulting TMLE plug-in contrast is \(8.2-4.6=3.6\), still not the true ATE of 4. The targeting equation has been solved, but because the propensity score is wrong, that equation does not force the targeted contrast to equal the truth.

Substitution Behavior Is Not the Same as Double Robustness

TMLE is called a substitution estimator because its final estimate substitutes fitted outcome predictions into the identified form of the parameter. For a binary outcome, a logistic fluctuation keeps predictions between 0 and 1, so the resulting ATE lies between -1 and 1. AIPW, by contrast, adds weighted residual corrections to an initial contrast and can produce a value outside this range. Respecting the parameter’s range can be valuable, but it does not guarantee that the estimate is unbiased, consistent, or efficient under misspecification.

There is also a useful algebraic connection. If a TMLE targeting step makes the empirical AIPW residual correction, evaluated at the targeted regression \(\hat Q^*\) and the same \(\hat g\), exactly zero, then the AIPW expression using \(\hat Q^*\) equals the TMLE plug-in estimate. This does not mean the AIPW estimator using the initial regression equals TMLE, nor does it make either estimate correct when both nuisance limits are wrong.

Consistency Is Not an Efficiency Guarantee

Under suitable regularity conditions, when both nuisance estimates converge to their true functions at appropriate rates, the AIPW and TMLE estimators can be asymptotically linear with the efficient influence function for the ATE. That supports efficient estimation and the corresponding influence-function-based variance calculation. When one nuisance limit is misspecified, double robustness may still give consistency, but the estimator’s first-order sampling variation need not be the efficient influence function’s variation. It is therefore not valid to infer efficiency merely from consistency.

In particular, an EIF-based standard error computed using a misspecified nuisance fit is not automatically justified by the double-robustness result. The variance procedure must be valid for the estimator’s actual large-sample behavior under the working models and estimation procedure. Cross-fitting and careful variance estimation address important estimation and inference issues, but they do not make a misspecified nuisance model correct.

Common Mistakes

  • “Doubly robust means safe if both models are wrong.” It means consistency can hold when either one of the two nuisance functions is consistently estimated, under the stated conditions. It gives no general guarantee when both limits are wrong.
  • “TMLE solves its score, so its answer must be right.” A score equation is a constraint. Its connection to the true ATE depends on conditions such as a correct propensity score or a correct outcome regression.
  • “A bounded TMLE is necessarily better.” Substitution can respect parameter bounds, but a bounded estimate can still be biased or inconsistent.
  • “Consistency implies efficiency.” Double robustness is a consistency property. Efficiency requires additional conditions, including suitable nuisance estimation and regularity.
Key takeaway: Assess AIPW and TMLE by separating three questions: Are the identification assumptions credible? Is at least one nuisance function consistently estimated? Are the stronger conditions needed for the claimed efficiency and inference also met?

Check Your Understanding

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

  1. In the AIPW probability-limit expression, what happens when \(g^*=g_0\) but \(Q^*\ne Q_0\)?
  2. Why does a correct outcome regression make the population AIPW residual corrections vanish even when \(g^*\) is wrong?
  3. In the both-wrong example, why does solving the TMLE targeting equation not ensure that the plug-in contrast is 4?
  4. For a binary outcome, what does logistic TMLE’s substitution structure guarantee about the range of its ATE estimate, and what does it not guarantee?
  5. Why is double robustness alone insufficient to claim that an EIF-based standard error is valid under nuisance-model misspecification?
Clinical Biostats