Skip to content

Doubly robust learner

Learn a conditional average effect by regressing augmented inverse-probability-weighted pseudo-outcomes, and see what double robustness does and does not protect.

A doubly robust learner (DR-learner) estimates how a treatment's average effect varies with observed characteristics by regressing an augmented inverse-probability-weighted pseudo-outcome on those characteristics. A quant researcher comparing two execution policies can use it to estimate which order profiles have different average cost effects. The causal interpretation still depends on how the policies were assigned and what confounders were recorded.
Its target is the conditional average treatment effect, or CATE,
τ(x)=E[Y(1)−Y(0)∣X=x].\tau(x)=\mathbb{E}[Y(1)-Y(0)\mid X=x].

Here D∈{0,1}D\in\{0,1\} is the policy used, XX contains pretreatment characteristics, and Y(d)Y(d) is the outcome under policy dd. For execution cost measured in basis points, a positive effect means policy 1 raises cost relative to policy 0. Specify the execution rules, horizon and treatment of unfilled orders before estimating this quantity.

Build a response, then learn its conditional mean

Let μd(x)=EY∣D=d,X=x]\mu_d(x)=\mathbb{E}Y\mid D=d,X=x] and let e(x)=P(D=1∣X=x)e(x)=\mathbb{P}(D=1\mid X=x) be the [propensity score. Fit estimates of these nuisance functions on a separate training sample. For an observation excluded from that fit, construct

ϕ^i=μ^1(Xi)−μ^0(Xi)+Di{Yi−μ^1(Xi)}e^(Xi)−(1−Di){Yi−μ^0(Xi)}1−e^(Xi).\widehat\phi_i= \widehat\mu_1(X_i)-\widehat\mu_0(X_i) +\frac{D_i\{Y_i-\widehat\mu_1(X_i)\}}{\widehat e(X_i)} -\frac{(1-D_i)\{Y_i-\widehat\mu_0(X_i)\}}{1-\widehat e(X_i)}.
The baseline difference is corrected using the observed policy's prediction residual and an inverse-probability weight. The second stage regresses these pseudo-outcomes on XX to obtain τ^(x)\widehat\tau(x). Averaging them instead targets an overall average effect. Kennedy's DR-learner construction, Algorithm 1, sets out this separation; cross-fitting can rotate the training and estimation roles.

The regression learns the effect function directly, which can be simpler than either outcome surface. It does not treat each pseudo-outcome as an observed individual causal effect.

One row gives 4; the conditional effect can be 2

Consider a fictional order profile. Suppose the nuisance predictions are μ^1=5\widehat\mu_1=5, μ^0=3\widehat\mu_0=3 basis points and e^=0.5\widehat e=0.5. An order receives policy 1 and its observed cost is 6 basis points. Its pseudo-outcome is

ϕ^=(5−3)+6−50.5=4 basis points.\widehat\phi=(5-3)+\frac{6-5}{0.5}=4\ \text{basis points}.

The untreated correction is zero because this order received policy 1. We have not observed what its cost would have been under policy 0. If that unobserved cost were 4, its individual effect would be 2, despite the pseudo-outcome of 4.

To see what the second stage targets, stipulate a complete fictional population at this same profile: policy assignment is independent with probability one half; policy-1 costs are equally likely to be 4 or 6, and policy-0 costs equally likely to be 2 or 4. The two potential costs are independent. The true conditional means are therefore 5 and 3, and the CATE is 2 basis points. Using those correct nuisances gives:

Observed policyObserved cost (bp)Population probabilityPseudo-outcome (bp)
161/44
141/40
041/40
021/44

Thus the pseudo-outcome's conditional mean is (4+0+0+4)/4=2(4+0+0+4)/4=2. This is a population calculation, not a fitted four-observation estimate.

Fictional execution-cost example: a baseline difference of 2 basis points plus a treated residual correction of 2 gives pseudo-outcome 4. Four equally probable observed states have pseudo-outcomes 4, 0, 0 and 4; their conditional mean is 2 basis points.
Fictional execution-cost example: a baseline difference of 2 basis points plus a treated residual correction of 2 gives pseudo-outcome 4. Four equally probable observed states have pseudo-outcomes 4, 0, 0 and 4; their conditional mean is 2 basis points.
Original fictional population. A noisy row-level response and its conditional mean answer different questions; all cost differences are in basis points.
For the averaging step, Conditional Expectation develops conditional means and the tower property. The public page previews the lesson; full teaching requires membership.

What “doubly robust” protects

Hold the fitted nuisance functions fixed by conditioning on their training data. Write them as m1(x),m0(x),p(x)m_1(x),m_0(x),p(x) and use ϕm,p\phi_{m,p} for the resulting pseudo-outcome. Directly taking expectations over the observed treatment and outcome gives

E[ϕm,p∣X=x]−{μ1(x)−μ0(x)}={p(x)−e(x)}{m1(x)−μ1(x)p(x)+m0(x)−μ0(x)1−p(x)}.\begin{aligned} \mathbb{E}[\phi_{m,p}\mid X=x]-\{\mu_1(x)-\mu_0(x)\} ={}&\{p(x)-e(x)\}\left\{ \frac{m_1(x)-\mu_1(x)}{p(x)} +\frac{m_0(x)-\mu_0(x)}{1-p(x)} \right\}. \end{aligned}
Assume the denominators are nonzero. This identity vanishes if the propensity is correct, or if both outcome means are correct. This is double robustness of the conditional mean. If both nuisance components are wrong, the product need not vanish. Even when it does vanish, an unsuitable or poorly estimated second-stage regression can still miss the conditional mean.
To interpret that mean as causal requires consistency, no unmeasured confounding given XX, and overlap. Neither nuisance model can repair an unidentified comparison.

Near-zero propensity denominators amplify residual noise. Finite outcome second moments and propensities bounded away from zero and one are useful sufficient controls when the fitted functions are also well behaved; bare positivity alone need not give finite pseudo-outcome variance. Clipping changes the score and can introduce bias.

Finally, consistency of a conditional mean is weaker than a fast convergence rate or valid confidence interval. Kennedy's Theorem 2 requires second-stage stability and control of pseudo-outcome error; matching its oracle additionally requires the smoothed product of nuisance errors to be negligible at the oracle regression's rate. There is no universal CATE guarantee from a particular nuisance rate alone. Ordinary random folds also need reconsideration for dependent, dated execution data.

Sources

Last updated
  • causal inference
  • heterogeneous treatment effects
  • doubly robust estimation
  • pseudo outcome
  • execution research
← Back to the glossaryBack to top ↑