Tutorials › Propensity Score Matching and Instrumental Variables › Targeted Maximum Likelihood Estimation as Targeted Updating

Doubly Robust Estimation · Tutorial 311 of 699

Targeted Maximum Likelihood Estimation as Targeted Updating

See how a target-specific update turns an initial outcome regression into a plug-in causal estimate, and how to check whether the update is supported by the data.

Advanced 12 min read

What You'll Learn

  • Define an initial outcome regression and a parameter-specific targeting update.
  • Explain how the target parameter determines what an update should accomplish.
  • Work through a numerical additive fluctuation and its plug-in ATE.
  • Distinguish a sample support problem from structural positivity failure.
  • Compare targeted updating with standardization and AIPW.
  • Identify diagnostics and limitations of a targeted update.

Why Update an Outcome Regression?

A fitted outcome regression can predict outcomes well on average without estimating a particular causal contrast accurately. Targeted maximum likelihood estimation (TMLE) addresses this distinction by starting with an initial outcome regression and then updating it in a way chosen for a specified parameter. The update is not an attempt to improve every prediction equally: it is intended to make the fitted outcome regression more informative for the causal quantity being estimated.

For a binary treatment, let \(O=(W,A,Y)\) denote one observation, where \(W\) is a vector of pretreatment covariates, \(A\in\{0,1\}\) is treatment, and \(Y\) is the outcome. Write \(Q_0(a,w)=E(Y\mid A=a,W=w)\) for the initial outcome regression, and \(g_0(w)=P(A=1\mid W=w)\) for the treatment mechanism. In practice, their fitted versions are estimated from observed data. Here the focus is how an initial outcome fit is targeted, not how to build those nuisance fits.

For example, the average treatment effect (ATE) in a target population with covariate distribution \(P_W\) is

$$ \psi(P)=E_W\{Q(1,W)-Q(0,W)\}. $$

An estimate of this parameter can be obtained by predicting each person's outcome under both treatment levels and averaging the predicted differences. TMLE changes the outcome regression before making these predictions. Thus the final estimate remains a plug-in estimate: it is calculated by substituting the updated regression into the parameter formula.

Definition: A targeted update is a data-informed modification of an initial estimate of the outcome regression, constructed so that the updated regression is better aligned with a specified parameter's estimating information. The resulting plug-in estimate is the target parameter evaluated using that updated regression and the empirical distribution of covariates.

What Makes an Update Targeted?

The update has two essential ingredients. First, a fluctuation model specifies a restricted way to modify the initial regression. Second, the update parameter is estimated from the observed data using a loss or estimating equation related to the target. The fluctuation is deliberately narrower than refitting an unrestricted prediction model: it changes the initial fit along a chosen direction that is relevant to the target parameter.

In a simple additive illustration, let \(H(a,w)\) denote a prescribed update direction and define

$$ Q_{\epsilon}(a,w)=Q_0(a,w)+\epsilon H(a,w). $$

The scalar \(\epsilon\) controls the size of the change. One possible fitting rule chooses \(\epsilon\) to make the empirical residual score in this direction equal to zero:

$$ \sum_{i=1}^{n} H(A_i,W_i)\{Y_i-Q_{\epsilon}(A_i,W_i)\}=0. $$

This equation is an illustrative fluctuation, not a universal specification of every TMLE. Actual choices of fluctuation model and fitting loss depend on the outcome type, parameter, and statistical setup. The direction must be chosen to connect the update to the target; an arbitrary adjustment is not made a TMLE merely by calling it a targeting step.

The basic arithmetic behind the additive example follows directly by substitution. Since \(Q_{\epsilon}=Q_0+\epsilon H\), the residual score equals

$$ \sum_{i=1}^{n}H_i(Y_i-Q_{0i})-\epsilon\sum_{i=1}^{n}H_i^2. $$

Setting this expression to zero gives \(\epsilon=\sum_i H_i(Y_i-Q_{0i})/\sum_i H_i^2\), provided \(\sum_i H_i^2>0\). The update therefore removes the empirical residual component in the selected direction. In a parameter-specific TMLE, the direction and fitting procedure are chosen so that this kind of calibration is relevant to the target's estimating information.

Worked Example: A Simple Additive Targeting Update

Suppose four hypothetical participants have binary treatment and a continuous outcome. For illustration, use the additive fluctuation above with the prespecified direction \(H(1,w)=1\) and \(H(0,w)=-1\). The table gives each participant's observed outcome and initial fitted outcome under the treatment actually received.

ParticipantWAYQ0(A,W)H(A,W)
1010.900.701
2110.800.701
3000.600.40-1
4100.200.30-1

The numerator of the update is the sum of direction-weighted residuals: \(1(0.90-0.70)+1(0.80-0.70)+(-1)(0.60-0.40)+(-1)(0.20-0.30)=0.20\). The denominator is \(1^2+1^2+(-1)^2+(-1)^2=4\). Therefore \(\hat\epsilon=0.20/4=0.05\).

The updated observed-treatment predictions are \(0.70+0.05(1)=0.75\), \(0.70+0.05(1)=0.75\), \(0.40+0.05(-1)=0.35\), and \(0.30+0.05(-1)=0.25\). Their direction-weighted residuals are \(0.15\), \(0.05\), \(-0.25\), and \(0.05\), respectively, which sum to zero. This verifies the fluctuation equation.

Suppose the initial predictions under both treatment levels are \(Q_0(1,0)=0.70\), \(Q_0(0,0)=0.40\), \(Q_0(1,1)=0.70\), and \(Q_0(0,1)=0.30\). Under the stated direction, the updated predictions are \(0.75,0.35,0.75,0.25\) for those same counterfactual treatment-covariate combinations. With one participant at each covariate value, the initial plug-in ATE is \(\{(0.70-0.40)+(0.70-0.30)\}/2=0.35\) outcome units. The updated plug-in ATE is \(\{(0.75-0.35)+(0.75-0.25)\}/2=0.45\) outcome units. The update changes the estimated contrast by \(0.10\) outcome units. This constructed example illustrates the arithmetic of a targeted additive fluctuation; its simple direction is not a general prescription for a binary-outcome ATE TMLE.

Assumptions and Interpretation

Targeting changes estimation; it does not by itself make a causal parameter identifiable. For the ATE to have a causal interpretation from observed data, common assumptions include consistency, conditional exchangeability, and positivity. Consistency says that the observed outcome equals the outcome under the treatment actually received. Conditional exchangeability says that, given measured pretreatment covariates \(W\), treatment assignment is independent of the potential outcomes. Positivity requires that each treatment level have positive probability at covariate values in the target population.

$$ 0<P(A=1\mid W=w)<1 $$

for covariate values \(w\) relevant to that population. These assumptions, together with suitable estimation conditions, support the causal interpretation of the target; the update does not establish that they hold. In particular, a fluctuation cannot create observed outcome information for treatment-covariate combinations that are absent from the data.

It is also important to distinguish population positivity from finite-sample support. If a sample stratum contains no treated participants, that sample cannot directly estimate the treated outcome there without additional modeling or data. This alone does not prove that the population treatment probability is zero in the stratum: a finite sample can omit treated participants even when population positivity holds. If treatment is structurally never assigned in that stratum, however, the ATE there is not identified from the observed data without additional assumptions or a change in target population.

Under appropriate conditions, TMLE is designed to retain important properties of augmented estimators, including double robustness: broadly, consistency may hold if either the outcome regression or the treatment mechanism is consistently estimated, subject to the required identification and regularity conditions. This is not a guarantee against every modeling or data problem. The earlier material on double robustness emphasizes that misspecification, poor overlap, and finite-sample instability can still matter.

Worked Example: Interpreting an Empty Treatment Stratum

Consider a hypothetical sample with 12 participants in covariate stratum \(W=1\), all of whom received control. A fitted treatment probability from the broader data is \(g(1)=0.40\). What can be concluded?

From the sample: there are no observed treated outcomes among the 12 participants in this stratum. A purely empirical treated-outcome average for this stratum cannot be computed, and an update relying on observed treated outcomes there is unsupported by these data. Extrapolating a fitted outcome regression into that cell requires modeling assumptions.

About the population: the sample pattern alone does not show that \(P(A=1\mid W=1)=0\). If the stated population probability really is \(0.40\), positivity holds at \(W=1\), although this particular finite sample has no treated observations there. If instead treatment is structurally never assigned in this stratum, then \(P(A=1\mid W=1)=0\), positivity fails there, and the stratum-specific treated mean needed for the ATE is not identified without additional assumptions or a changed target population.

Why the Target Parameter Matters

The same initial outcome regression can require different updates for different targets. For the ATE, the plug-in formula averages treatment-specific prediction differences over the target covariate distribution. For the average treatment effect on the treated (ATT), the target averages a contrast over the covariate distribution of treated individuals instead. Since the averaging population changes, the information that the update should emphasize also changes.

This is why an update chosen for one estimand should not automatically be reused for another. The target parameter determines which parts of the outcome regression matter and how the update should be constructed. More generally, targeting can apply to parameters other than a mean contrast, but each target requires a compatible fluctuation and plug-in calculation.

Worked Example: Changing the Target Population Changes the Plug-In Estimate

Suppose a hypothetical fitted regression gives treatment contrasts of 4 and 10 outcome units for covariate groups \(W=0\) and \(W=1\), respectively. In the full population, \(P(W=1)=0.25\), while among treated individuals, \(P(W=1\mid A=1)=0.60\).

For the ATE, the standardized contrast is \(0.75(4)+0.25(10)=3+2.5=5.5\) outcome units. For the ATT, the contrast averaged over the treated covariate distribution is \(0.40(4)+0.60(10)=1.6+6=7.6\) outcome units. These are different target quantities because they average the same conditional contrasts over different populations.

A TMLE for the ATE must target the population-average contrast, whereas an ATT TMLE must target the treated-population contrast. The numerical difference here is not evidence that one calculation is wrong; it follows from the distinct estimands.

Diagnostics, Alternatives, and Limitations

Targeted updating is useful because it combines an initial outcome model with a parameter-focused adjustment while retaining a plug-in form. But the fitted update and its implications should be checked rather than assumed to work automatically.

  • Check support and overlap. Examine treatment probabilities and the representation of treatment levels across important covariate regions. Extreme probabilities can make target-relevant updates unstable; an empty sample cell calls for caution, not an automatic conclusion of structural nonpositivity.
  • Check the fluctuation fit. Confirm that the intended empirical score or loss condition is satisfied and that the update converged. A numerical failure or near-zero variation in the update direction may mean the proposed fluctuation is not estimable as specified.
  • Check predictions. Ensure updated predictions are plausible for the outcome scale. A simple additive update can produce values outside a known outcome range; a suitable model and link function may be needed for bounded outcomes.
  • Check the target definition. Confirm that the empirical averaging distribution matches the intended population. Averaging over all participants when the target is the treated population changes the estimand.

Standardization uses an outcome regression to predict outcomes under alternative treatment levels and averages the resulting contrasts, without necessarily adding a targeted update. AIPW combines outcome regression and treatment weighting through an augmented estimating expression. TMLE also uses outcome and treatment information, but modifies the outcome regression and then evaluates the target as a plug-in parameter. In suitable implementations, the methods can have closely related large-sample behavior while differing in finite-sample estimates, constraints, and computational details.

The update is not a general cure for poor nuisance fits, lack of overlap, unmeasured confounding, or an ill-defined target. A targeted fit may shift the estimate substantially, and that shift should be interpreted alongside the data support and modeling choices. Finally, a fitted score equation does not alone provide valid uncertainty quantification: inference requires appropriate variance estimation and regularity conditions.

Key takeaway: TMLE is a parameter-specific update followed by a plug-in calculation. Its defining logic is not simply “fit an outcome model,” but “modify that fit along a direction chosen to improve estimation of the stated target,” while preserving the need for causal identification assumptions and adequate data support.

Check Your Understanding

Answer each question, then compare with the brief solutions.

  1. In one sentence, distinguish an initial outcome regression from a targeted update.
  2. In an additive fluctuation, suppose \(\sum_i H_i(Y_i-Q_{0i})=0.30\) and \(\sum_i H_i^2=6\). What value of \(\hat\epsilon\) solves the empirical score equation?
  3. A sample has no treated participants in one covariate stratum. Does this alone prove a population positivity failure? Explain.
  4. Why can the ATE and ATT plug-in estimates differ even when they use the same fitted conditional treatment contrasts?
  5. Name one problem that targeting cannot solve by itself.

Solutions. (1) The initial regression is the starting outcome fit; the targeted update modifies it along a direction selected for the parameter of interest. (2) \(\hat\epsilon=0.30/6=0.05\). (3) No. It is a finite-sample support limitation; structural positivity failure requires the population treatment probability to be zero in that stratum. (4) They average contrasts over different covariate distributions: the full population for the ATE and the treated population for the ATT. (5) Examples include unmeasured confounding, structural positivity failure, or an inappropriate target definition.

Clinical Biostats