Conditional Distributional Treatment Effects: Doubly Robust Estimation and Testing
Not on the map. Why?
Abstract (arXiv)
Beyond conditional average treatment effects, treatments may impact the entire outcome distribution in covariate-dependent ways, for example, by altering the variance or tail risks for specific subpopulations. We propose a novel estimand to capture such conditional distributional treatment effects, and develop a doubly robust estimator that is minimax optimal in the local asymptotic sense. Using this, we develop a test for the global homogeneity of conditional potential outcome distributions that accommodates discrepancies beyond the maximum mean discrepancy (MMD), has provably valid type 1 error, and is consistent against fixed alternatives---the first test, to our knowledge, with such guarantees in this setting. We then provide a test that aggregates evidence across a grid of kernel-bandwidth choices. Furthermore, we derive exact closed-form expressions for two natural discrepancies (including the MMD), and provide a computationally efficient, permutation-free algorithm for our test.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 17 of its 55 references that are in the map. First posted on arXiv 17 Mar 2026.
Builds on
17 of 55 references.
References · 17
- 2019Optuna: A Next-generation Hyperparameter Optimization Frameworkcited by 2,455
- 2015Estimation and Inference of Heterogeneous Treatment Effects using Random Forestscited by 597
- 2017Meta-learners for Estimating Heterogeneous Treatment Effects using Machine Learningcited by 475
- 2020A Measure-Theoretic Approach to Kernel Conditional Mean Embeddingscited by 86
- 2021Conditional Distributional Treatment Effect with Kernel Conditional Mean Embeddings and U-Statistic Regressioncited by 40
- 2018Counterfactual Mean Embeddingscited by 32
- 2021A Witness Two-Sample Testcited by 20
- 2017On the Optimality of Kernel-Embedding Based Goodness-of-Fit Testscited by 19
- 2022Robust and Agnostic Learning of Conditional Distributional Treatment Effectscited by 18
- 2020Dimension-agnostic inference using cross U-statisticscited by 14
- 2023One-Step Estimation of Differentiable Hilbert-Valued Parameterscited by 11
- 2023An Efficient Doubly-Robust Test for the Kernel Treatment Effectcited by 10
5 more
- 2024Neural-Kernel Conditional Mean Embeddingscited by 7
- 2023Confidence and Uncertainty Assessment for Distributional Random Forestscited by 5
- 2024Causal-DRF: Conditional Kernel Treatment Effect Estimation using Distributional Random Forestcited by 5
- 2018Network and Panel Quantile Effects Via Distribution Regressioncited by 3
- 2025Minimax Optimal Kernel Two-Sample Tests with Random Featurescited by 2
Built on by
Citing papers per year: 2026: 1.
1 paper.
- 2026Wasserstein Causal Forests for Distribution-Valued Outcomes27 Sep 2026