Augmented inverse-probability-weighted estimator Created 2026-09-24 Updated 2026-11-05
An augmented inverse-probability-weighted estimator averages a regression prediction plus an inverse-probability-weighted residual. In the treatment setting of doubly robust estimation, let estimate the conditional outcome mean and the probability of receiving treatment . Then
The first term predicts outcomes for everyone; the weighted residual corrects prediction errors using observed outcomes from the corresponding treatment group. The conditional-expectation calculation in doubly robust estimation makes the remaining bias a product of outcome-regression and propensity-score errors. An exactly correct outcome regression makes the residual correction mean zero even with an incorrect propensity score; a correct propensity score makes that correction repair the regression bias.
The nuisance estimates are commonly obtained by cross-fitting, with each observation evaluated using functions trained on other observations. Both treatment probabilities must remain positive on the population being studied, and small fitted probabilities can make the variance very large. Double robustness concerns consistency under one correct nuisance model; it does not automatically deliver valid finite-sample or asymptotic confidence intervals.
Double machine learning Created 2026-09-24 Updated 2026-11-05
Double machine learning combines Neyman orthogonality with cross-fitting to estimate a low-dimensional statistical parameter while using machine learning for nuisance functions. Orthogonality means that the expected estimating score has zero first derivative with respect to nuisance perturbations at the truth. Small nuisance errors therefore affect the target equation only at second order, under the required smoothness assumptions. cross-fitting separately limits overfitting effects by evaluating scores on observations excluded from their nuisance training.
For the augmented inverse-probability-weighted estimator, a typical sufficient condition is
together with nuisance consistency, suitable moments and treatment probabilities bounded away from zero and one. Each error being is sufficient for this product condition, but one may converge more slowly if the other converges faster. Under the full regularity conditions the target estimate has asymptotic normality at scale , enabling confidence intervals. Machine learning prediction accuracy alone does not supply these assumptions or Causal identification.[ref]
Estimate the propensity score and the untreated outcome regression , preferably with cross-fitting when flexible methods are used, and set
With , the resulting one-step estimator of the ATT is
Equivalently,
This is an augmented inverse-probability-weighted estimator; it is consistent when either the propensity model or the untreated outcome model is correct, subject to the usual regularity and positivity conditions.
The tower property of conditional expectation gives
and another expectation gives .
Use the augmented inverse-probability-weighted estimator
Condition on the independently trained nuisance estimators. Subtracting the oracle influence variable
produces a conditional empirical fluctuation with variance after multiplication by , using , overlap, and the bounded conditional variance. Its conditional bias is
whose absolute value is at most by Cauchy-Schwarz inequality. Thus
The central limit theorem and Slutsky theorem give the claimed limit. Without auxiliary data, use cross-fitting: split the sample into folds, train both nuisance estimators away from each observation's fold, and average the same score over held-out observations.