Double Machine Learning

double-machine-learningcausal-inferencemachine-learningneyman-orthogonalitycross-fittingtreatment-effect

Definition

Double (or debiased) machine learning (DML; Chernozhukov et al. 2018) is a framework for estimating a low-dimensional causal parameter θ0\theta_0 — a treatment effect, an elasticity, a structural coefficient — while using flexible machine-learning methods to estimate high-dimensional nuisance functions (e.g. the outcome regression and the propensity/treatment model), and still obtaining n\sqrt{n}-consistent, asymptotically normal inference with honest standard errors. It rests on three ingredients: a Neyman-orthogonal score, high-quality ML nuisance estimation, and cross-fitting (Bach et al. 2023).

Key Ideas

How It Works

For the partially linear model Y=θ0D+g0(X)+εY = \theta_0 D + g_0(X) + \varepsilon, D=m0(X)+VD = m_0(X) + V: (1) estimate g0g_0 and m0m_0 by any ML learner on a training fold; (2) form residuals Y~=Yg^0(X)\tilde Y = Y - \hat g_0(X) and D~=Dm^0(X)\tilde D = D - \hat m_0(X) on the held-out fold; (3) estimate θ0\theta_0 from the orthogonal moment E[(Y~θD~)D~]=0E[(\tilde Y - \theta \tilde D)\tilde D]=0, i.e. an IV/residual-regression of Y~\tilde Y on D~\tilde D; (4) repeat across folds and average. The variance follows from the influence function of the orthogonal score, giving standard confidence intervals.

Why It Matters

Open Questions

Related