Conditional Distributional Treatment Effects: Doubly Robust Estimation and Testing
Saksham Jain, Alex Luedtke
Abstract
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.
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 006d3f63-d25f-45be-ba18-94186412e7ffBuilds on4
- A Measure-Theoretic Approach to Kernel Conditional Mean EmbeddingsJunhyung Park, Krikamol MuandetNeurIPS 2020 · 123 citations
- Conditional Distributional Treatment Effect with Kernel Conditional Mean Embeddings and U-Statistic RegressionJunhyung Park, Uri Shalit, Bernhard Schölkopf, Krikamol MuandetICML 2021 · 46 citations
- An Efficient Doubly-Robust Test for the Kernel Treatment EffectDiego Martinez-Taboada, Aaditya Ramdas, Edward KennedyNeurIPS 2023 · 17 citations
- Neural-Kernel Conditional Mean EmbeddingsEiki Shimizu, Kenji Fukumizu, Dino SejdinovicICML 2024 · 6 citations
Related papers
- Sinkhorn Treatment Effects: A Causal Optimal Transport MeasureMedha Agarwal, Alex LuedtkeICML 2026
- A Meta-learner for Heterogeneous Effects in Difference-in-DifferencesHui Lan, Haoge Chang, Eleanor Wiske Dillon, Vasilis SyrgkanisICML 2025
- A permutation-free kernel two-sample testShubhanshu Shekhar, Ilmun Kim, Aaditya RamdasNeurIPS 2022 · 40 citations
- Kernel-based Maximum-of-difference Test for Two-sample ComparisonDan Pu, Tianyi Zhu, Yao Yan, Wei LanICML 2026
- Counterfactual Density Estimation using Kernel Stein DiscrepanciesDiego Martinez-Taboada, Edward KennedyICLR 2024 · 8 citations
