mlsanityData to 7 Oct 2026

Conditional Distributional Treatment Effects: Doubly Robust Estimation and Testing

arXiv 2603.16829 ↗First posted 17 Mar 2026stat.ML cs.LG math.STcited by 1

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

  1. 2019Optuna: A Next-generation Hyperparameter Optimization Frameworkcited by 2,455
  2. 2015Estimation and Inference of Heterogeneous Treatment Effects using Random Forestscited by 597
  3. 2017Meta-learners for Estimating Heterogeneous Treatment Effects using Machine Learningcited by 475
  4. 2020A Measure-Theoretic Approach to Kernel Conditional Mean Embeddingscited by 86
  5. 2021Conditional Distributional Treatment Effect with Kernel Conditional Mean Embeddings and U-Statistic Regressioncited by 40
  6. 2018Counterfactual Mean Embeddingscited by 32
  7. 2021A Witness Two-Sample Testcited by 20
  8. 2017On the Optimality of Kernel-Embedding Based Goodness-of-Fit Testscited by 19
  9. 2022Robust and Agnostic Learning of Conditional Distributional Treatment Effectscited by 18
  10. 2020Dimension-agnostic inference using cross U-statisticscited by 14
  11. 2023One-Step Estimation of Differentiable Hilbert-Valued Parameterscited by 11
  12. 2023An Efficient Doubly-Robust Test for the Kernel Treatment Effectcited by 10
5 more
  1. 2024Neural-Kernel Conditional Mean Embeddingscited by 7
  2. 2023Confidence and Uncertainty Assessment for Distributional Random Forestscited by 5
  3. 2024Causal-DRF: Conditional Kernel Treatment Effect Estimation using Distributional Random Forestcited by 5
  4. 2018Network and Panel Quantile Effects Via Distribution Regressioncited by 3
  5. 2025Minimax Optimal Kernel Two-Sample Tests with Random Featurescited by 2

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.