From an Outcome Update to an Estimating Equation
For a binary treatment and binary outcome, TMLE updates an initial estimate of the outcome regression in a direction chosen to target the average treatment effect (ATE). The update is not just a way to obtain new outcome predictions: its score equation is designed to make the empirical residual part of the efficient influence function equal zero. This gives a precise connection to AIPW while preserving the plug-in form of the TMLE estimate.
Let \(O_i=(X_i,A_i,Y_i)\), \(i=1,\ldots,n\), denote independent observations, with \(A_i\in\{0,1\}\) and \(Y_i\in\{0,1\}\). Write \(g(X)=P(A=1\mid X)\) for the propensity score and \(Q(a,X)=E(Y\mid A=a,X)\) for the outcome regression. The empirical average of a quantity \(f(O_i)\) is \(P_n f=n^{-1}\sum_{i=1}^n f(O_i)\). The target parameter is \(\psi_0=E\{Q(1,X)-Q(0,X)\}\).
Here \(\operatorname{logit}(u)=\log\{u/(1-u)\}\), and \(\operatorname{expit}(v)=1/(1+e^{-v})\). The update moves the prediction upward for treated observations and downward for controls when \(\epsilon\) is positive. To form the plug-in ATE, apply the fitted fluctuation to both treatment levels for every observed \(X_i\): \(\hat Q^\epsilon(a,X_i)=\operatorname{expit}[\operatorname{logit}\hat Q(a,X_i)+\epsilon\{a/\hat g(X_i)-(1-a)/(1-\hat g(X_i))\}]\). Then \(\hat\psi_{\mathrm{TMLE}}=P_n\{\hat Q^{\hat\epsilon}(1,X)-\hat Q^{\hat\epsilon}(0,X)\}\).
Why the Score Makes the Empirical Influence Function Vanish
For the ATE, the efficient influence function evaluated at nuisance functions \(Q\) and \(g\), and at a candidate parameter \(\psi\), is \[ D^*(Q,g,\psi)(O)=\frac{A}{g(X)}\{Y-Q(1,X)\} -\frac{1-A}{1-g(X)}\{Y-Q(0,X)\} +Q(1,X)-Q(0,X)-\psi. \] The first two terms are the weighted outcome residuals; the third is the plug-in contrast. This formula is the bridge between the targeting score and the AIPW estimator.
In the empirical data, the targeting score at the updated regression is exactly the empirical average of the two weighted residual terms. To see this, substitute the definition of \(H_{\hat g}\): for \(A=1\) the score contribution is \(\{Y-\hat Q^{\hat\epsilon}(1,X)\}/\hat g(X)\); for \(A=0\) it is \(-\{Y-\hat Q^{\hat\epsilon}(0,X)\}/\{1-\hat g(X)\}\). Thus, when the score equation is solved and \(\hat\psi_{\mathrm{TMLE}}\) is the empirical plug-in contrast,
The second zero follows from the definition of the empirical plug-in estimate, not from an additional model fit. This is an algebraic finite-sample identity when the targeting score is solved exactly. If numerical optimization stops with a nonzero score, the empirical influence-function average will generally be nonzero by that same amount.
Worked Example: Solving a Targeting Equation
Consider eight hypothetical observations, four treated and four controls, with \(\hat g(X_i)=0.5\) and initial predictions \(\hat Q(1,X_i)=\hat Q(0,X_i)=0.5\). Suppose three treated outcomes and one control outcome equal 1; the other outcomes equal 0. The treatment and outcome data are therefore compatible with the stated counts, and the fitted propensity scores are strictly between 0 and 1.
For treated observations \(H_{\hat g}=1/0.5=2\); for controls \(H_{\hat g}=-1/(1-0.5)=-2\). The initial logit is zero for both treatment levels. If \(\epsilon\) is the fluctuation parameter, then the updated treated prediction is \(p=\operatorname{expit}(2\epsilon)\), while the updated control prediction is \(1-p=\operatorname{expit}(-2\epsilon)\).
The score equation, multiplied by the common positive factor \(n\), is \(2(3-4p)-2\{1-4(1-p)\}=0\). Simplifying gives \(12-16p=0\), so \(p=0.75\). Since \(p=\operatorname{expit}(2\hat\epsilon)\), the correct inversion is \(2\hat\epsilon=\operatorname{logit}(0.75)=\log(3)\). Therefore \(\hat\epsilon=\tfrac12\log(3)\approx0.5493\). The updated control prediction is \(1-p=0.25\).
The score can be checked directly: the treated residual sum is \(3-4(0.75)=0\), and the control residual sum is \(1-4(0.25)=0\). Hence the weighted residual score is \(2(0)-2(0)=0\). The plug-in contrast is \(0.75-0.25=0.50\), so \(\hat\psi_{\mathrm{TMLE}}=0.50\) in this hypothetical sample.
As an additional check, the initial AIPW estimate is \(0+8^{-1}[2\{3-4(0.5)\}-2\{1-4(0.5)\}]=4/8=0.50\). It happens to match the TMLE here. That agreement is not a general finite-sample identity between TMLE and AIPW based on the initial outcome regression.
The Exact Link to AIPW
For any fitted outcome regression \(Q\) and propensity score \(g\), the empirical AIPW estimate is the empirical average of the plug-in contrast plus the weighted residual correction. If the outcome regression is the targeted one, the correction is zero by the targeting equation. Consequently, TMLE equals the AIPW expression evaluated at the updated outcome regression and the same propensity score:
The equality follows because the last two terms have empirical average zero. It does not say that TMLE must equal AIPW computed with the initial \(\hat Q\). Updating \(\hat Q\) changes both the plug-in contrast and the residual correction; the targeting score constrains the correction at the updated fit, not at the initial fit. Nor does the identity require the propensity model to be correct as an algebraic matter. Causal interpretation and statistical validity require the relevant identification and estimation conditions.
Worked Example: Checking the Updated AIPW Identity
Use the hypothetical fitted values from the preceding example: the updated treatment predictions are 0.75, the updated control predictions are 0.25, and \(\hat g=0.5\). The empirical plug-in contrast is \(0.75-0.25=0.50\). The weighted residual correction is \(8^{-1}[2\{3-4(0.75)\}-2\{1-4(0.25)\}]=0\).
Thus AIPW evaluated at the updated regression is \(0.50+0=0.50\), exactly the TMLE estimate. The calculation also checks the empirical influence function: its weighted residual part is zero, and its plug-in part averages to \(P_n(0.50-0.50)=0\).
For contrast, the initial regression in this example produced an initial AIPW estimate of 0.50 as well. That equality is a feature of these particular values. In general, the expression with initial \(\hat Q\) uses different residuals and a different plug-in contrast, so it need not equal the targeted estimate.
Conditions and Numerical Diagnostics
The score-to-influence-function identity is direct algebra, but estimating a causal ATE requires more than solving that equation. The usual identification conditions include consistency of the observed outcome with the relevant potential outcome, conditional exchangeability of treatment given \(X\), and positivity: both treatment levels must have positive probability at covariate values represented in the target population. The observations are assumed independent for the simple empirical formulas here. For the binary-outcome logistic fluctuation, the initial predictions should be strictly between 0 and 1, and the estimated propensities should avoid 0 and 1 so that logits and weights are defined and stable.
A finite maximum-likelihood fluctuation estimate is not guaranteed. Complete or near separation can make the logistic likelihood increase toward a boundary rather than attain a finite maximizer. Extreme propensity scores can also make the clever covariate very large, so a small prediction residual may contribute strongly to the score. In computation, check the fitted \(\hat\epsilon\), the range of \(\hat g\), and the post-update score \(P_n[H_{\hat g}\{Y-\hat Q^{\hat\epsilon}(A,X)\}]\). A small score is evidence that the numerical targeting equation was solved; it is not proof that positivity, exchangeability, or the nuisance fits are adequate.
If optimization reports nonconvergence, or the score remains appreciably away from zero, do not describe the empirical influence-function equation as having been made to vanish. Check the likelihood optimization, the prediction bounds, and the propensity values. Altering or truncating extreme weights can improve stability but changes the procedure and may introduce bias; it should not be treated as an automatic cure.
Estimate \(\hat Q(a,X)\) and \(\hat g(X)\), checking that the predictions and propensity scores permit the planned update.
Use the clever covariate \(H_{\hat g}\) in the logistic offset model and solve its score equation, verifying the numerical residual.
Predict each treatment-level outcome for every observed covariate vector and average their difference.
Evaluate the weighted residual correction using the updated regression; it should be zero up to numerical tolerance.
How to Interpret the Connection
AIPW starts with a plug-in contrast and adds a weighted residual correction. TMLE instead updates the outcome regression so that the correction, evaluated at that updated fit, is zero in the observed sample; the resulting plug-in estimate is therefore also an AIPW estimate at the updated fit. AIPW is commonly expressed directly as a one-step correction to an initial regression, whereas TMLE uses a targeted update and retains an estimate in the parameter’s plug-in form. Neither method’s finite-sample identity alone establishes causal identification or guarantees good inference.
The empirical equation is also distinct from a population statement. A zero empirical influence-function average is imposed by the targeting procedure, subject to optimization accuracy. It does not mean the population influence function is identically zero, that the estimator has no sampling variability, or that a confidence interval has been established. Inference requires appropriate variance and nuisance-estimation considerations.
Check Your Understanding
Answer each question, then check the solutions below.
- For \(\hat g(X)=0.25\), what is the clever covariate for a treated observation and for a control observation?
- What two empirical terms vanish in \(P_nD^*(\hat Q^{\hat\epsilon},\hat g,\hat\psi_{\mathrm{TMLE}})\), and why?
- In the first worked example, why is \(2\hat\epsilon=\log(3)\) rather than \(\operatorname{logit}(2\hat\epsilon)=\log(3)\)?
- Does the TMLE/AIPW identity show that AIPW computed from the initial outcome regression equals TMLE in every sample?
- Name two checks that are useful if the fitted targeting score is not close to zero.
Solutions. (1) For a treated observation, \(H=1/0.25=4\); for a control, \(H=-1/0.75=-4/3\). (2) The weighted residual terms average to zero by the targeting score; the empirical plug-in contrast minus \(\hat\psi_{\mathrm{TMLE}}\) averages to zero by the definition of the estimate. (3) Because \(p=\operatorname{expit}(2\hat\epsilon)\), applying the inverse link gives \(2\hat\epsilon=\operatorname{logit}(p)\); the logit is applied to \(p\), not to \(2\hat\epsilon\). (4) No. The exact equality is to AIPW evaluated at the updated outcome regression. (5) Check the optimization convergence and post-update score, and inspect extreme propensity scores and outcome predictions at the boundaries.