Localized Debiased Machine Learning: Efficient Inference on Quantile Treatment Effects and Beyond

Nathan Kallus,Xiaojie Mao,Masatoshi Uehara
DOI: https://doi.org/10.48550/arxiv.1912.12945
2024-01-01
Journal of Machine Learning Research
Abstract:We consider estimating a low-dimensional parameter in an estimating equationinvolving high-dimensional nuisances that depend on the parameter. A centralexample is the efficient estimating equation for the (local) quantile treatmenteffect ((L)QTE) in causal inference, which involves as a nuisance thecovariate-conditional cumulative distribution function evaluated at thequantile to be estimated. Debiased machine learning (DML) is a data-splittingapproach to estimating high-dimensional nuisances using flexible machinelearning methods, but applying it to problems with parameter-dependentnuisances is impractical. For (L)QTE, DML requires we learn the wholecovariate-conditional cumulative distribution function. We instead proposelocalized debiased machine learning (LDML), which avoids this burdensome stepand needs only estimate nuisances at a single initial rough guess for theparameter. For (L)QTE, LDML involves learning just two regression functions, astandard task for machine learning methods. We prove that under lax rateconditions our estimator has the same favorable asymptotic behavior as theinfeasible estimator that uses the unknown true nuisances. Thus, LDML notablyenables practically-feasible and theoretically-grounded efficient estimation ofimportant quantities in causal inference such as (L)QTEs when we must controlfor many covariates and/or flexible relationships, as we demonstrate inempirical studies.
What problem does this paper attempt to address?