Double Machine Learning Density Estimation for Local Treatment Effects with Instruments
Yonghan Jung, Jin Tian, Elias Bareinboim
Abstract
Randomized Controlled Trials constitute a powerful tool to learn cause and effect relationships found throughout a wide range of applied settings. In practice, the treatment assignment's compliance is hard to ascertain in many settings since patients may not feel compelled to take the treatment for various reasons. One typical quantity investigated in these settings is the local treatment effect (LTE, for short). The LTE measures the causal effect among compliers, which usually comes under the assumption of monotonicity (only the ones offered the treatment are allowed to take it). In this paper, we investigate the challenge of estimating the LTE density function (instead of its expected value) of a binary treatment on a continuous outcome given a binary instrumental variable in the presence of both observed and unobserved confounders. Specifically, we develop two families of methods for this task, kernel-smoothing and model-based approximationsthe former smoothes the density by convoluting with a smooth kernel function; the latter projects the density onto a finite-dimensional density class. For both approaches, we derive double/debiased machine learning (DML) based estimators. We study the asymptotic convergence rates of the estimators and show that they are robust to the biases in nuisance function estimation. We illustrate the proposed methods on synthetic data and a real dataset called 401(k).
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext c3e4ceae-5e11-4edc-999c-78446b0a19a4Cited by top-tier papers5
- Empirical Gateaux Derivatives for Causal InferenceMichael I. Jordan, Yixin Wang, Angela ZhouNeurIPS 2022 · 13 citations
- Mitigating Adversarial Vulnerability through Causal Parameter Estimation by Adversarial Double Machine LearningByung-Kwan Lee, Junho Kim, Yong Man RoICCV 2023 · 12 citations
- Fast Instrument Learning with Faster RatesZiyu Wang, Yuhao Zhou, Jun ZhuNeurIPS 2022 · 6 citations
- Automated Efficient Estimation using Monte Carlo Efficient Influence FunctionsRaj Agrawal, Sam Witty, Andy Zane, Elias BinghamNeurIPS 2024 · 6 citations
- Transportable Representations for Domain GeneralizationKasra Jalaldoust, Elias BareinboimAAAI 2024 · 5 citations
Builds on1
Related papers
- Estimating individual treatment effects under unobserved confounding using binary instrumentsDennis Frauen, Stefan FeuerriegelICLR 2023 · 3 citations
- Estimating Heterogeneous Treatment Effects by Combining Weak Instruments and Observational DataMiruna Oprescu, Nathan KallusNeurIPS 2024 · 4 citations
- Beyond the Average: Distributional Causal Inference under Imperfect ComplianceUndral Byambadalai, Tomu Hirata, Tatsushi Oka, Shota YasuiNeurIPS 2025 · 3 citations
- A Non-parametric Direct Learning Approach to Heterogeneous Treatment Effect Estimation under Unmeasured ConfoundingXinhai Zhang, Xingye QiaoNeurIPS 2024 · 1 citation
- Estimating Distributional Treatment Effects in Randomized Experiments: Machine Learning for Variance ReductionUndral Byambadalai, Tatsushi Oka, Shota YasuiICML 2024 · 7 citations
