# Conditional Distributional Treatment Effect with Kernel Conditional Mean Embeddings and U-Statistic Regression

arXiv: https://arxiv.org/abs/2102.08208 · HTML: https://mlsanity.com/paper/2102.08208/

## Key facts

- Cited by 40 papers in the map (data through 14 Oct 2026).
- Builds on 20 of its 83 references that are in the map.
- First posted on arXiv 16 Feb 2021.

## Builds on (most cited first, 20 of 20)

- [Estimation and Inference of Heterogeneous Treatment Effects using Random Forests](https://mlsanity.com/paper/1510.04342/) (2015)
- [Meta-learners for Estimating Heterogeneous Treatment Effects using Machine Learning](https://mlsanity.com/paper/1706.03461/) (2017)
- [Causal Effect Inference with Deep Latent-Variable Models](https://mlsanity.com/paper/1705.08821/) (2017)
- [Kernel Mean Embedding of Distributions: A Review and Beyond](https://mlsanity.com/paper/1605.09522/) (2016)
- [Learning Representations for Counterfactual Inference](https://mlsanity.com/paper/1605.03661/) (2016)
- [Adapting Neural Networks for the Estimation of Treatment Effects](https://mlsanity.com/paper/1906.02120/) (2019)
- [Bayesian Inference of Individualized Treatment Effects using Multi-task Gaussian Processes](https://mlsanity.com/paper/1704.02801/) (2017)
- [Kernel methods through the roof: handling billions of points efficiently](https://mlsanity.com/paper/2006.10350/) (2020)
- [A Measure-Theoretic Approach to Kernel Conditional Mean Embeddings](https://mlsanity.com/paper/2002.03689/) (2020)
- [FALKON: An Optimal Large Scale Kernel Method](https://mlsanity.com/paper/1705.10958/) (2017)
- [Operator-valued Kernels for Learning from Functional Response Data](https://mlsanity.com/paper/1510.08231/) (2015)
- [Identifying Causal-Effect Inference Failure with Uncertainty-Aware Models](https://mlsanity.com/paper/2007.00163/) (2020)
- [Some methods for heterogeneous treatment effect estimation in high-dimensions](https://mlsanity.com/paper/1707.00102/) (2017)
- [Kernel Distribution Embeddings: Universal Kernels, Characteristic Kernels and Kernel Metrics on Distributions](https://mlsanity.com/paper/1604.05251/) (2016)
- [Globally Convergent Newton Methods for Ill-conditioned Generalized Self-concordant Losses](https://mlsanity.com/paper/1907.01771/) (2019)
- [Bayesian Nonparametric Causal Inference: Information Rates and Learning Algorithms](https://mlsanity.com/paper/1712.08914/) (2017)
- [Causal effects based on distributional distances](https://mlsanity.com/paper/1806.02935/) (2018)
- [Robust Recursive Partitioning for Heterogeneous Treatment Effects with Uncertainty Quantification](https://mlsanity.com/paper/2006.07917/) (2020)
- [Regularised Least-Squares Regression with Infinite-Dimensional Output Space](https://mlsanity.com/paper/2010.10973/) (2020)
- [Network and Panel Quantile Effects Via Distribution Regression](https://mlsanity.com/paper/1803.08154/) (2018)

## Built on by (most cited first, 30 of 40)

- [Nonparametric Identifiability of Causal Representations from Unknown Interventions](https://mlsanity.com/paper/2306.00542/) (2023)
- [Causal Discovery in Heterogeneous Environments Under the Sparse Mechanism Shift Hypothesis](https://mlsanity.com/paper/2206.02013/) (2022)
- [Robust and Agnostic Learning of Conditional Distributional Treatment Effects](https://mlsanity.com/paper/2205.11486/) (2022)
- [On the Relationship Between Explanation and Prediction: A Causal View](https://mlsanity.com/paper/2212.06925/) (2022)
- [Identifiable Causal Representation Learning: Unsupervised, Multi-View, and Multi-Environment](https://mlsanity.com/paper/2406.13371/) (2024)
- [An Efficient Doubly-Robust Test for the Kernel Treatment Effect](https://mlsanity.com/paper/2304.13237/) (2023)
- [Quantifying Aleatoric Uncertainty of the Treatment Effect: A Novel Orthogonal Learner](https://mlsanity.com/paper/2411.03387/) (2024)
- [Context-Aware Drift Detection](https://mlsanity.com/paper/2203.08644/) (2022)
- [Estimating Potential Outcome Distributions with Collaborating Causal Networks](https://mlsanity.com/paper/2110.01664/) (2021)
- [Early Concept Drift Detection via Prediction Uncertainty](https://mlsanity.com/paper/2412.11158/) (2024)
- [Feature Selection for Discovering Distributional Treatment Effect Modifiers](https://mlsanity.com/paper/2206.00516/) (2022)
- [Neural-Kernel Conditional Mean Embeddings](https://mlsanity.com/paper/2403.10859/) (2024)
- [On Efficient Estimation of Distributional Treatment Effects under Covariate-Adaptive Randomization](https://mlsanity.com/paper/2506.05945/) (2025)
- [Confidence and Uncertainty Assessment for Distributional Random Forests](https://mlsanity.com/paper/2302.05761/) (2023)
- [iSCAN: Identifying Causal Mechanism Shifts among Nonlinear Additive Noise Models](https://mlsanity.com/paper/2306.17361/) (2023)
- [Estimating Distributional Treatment Effects in Randomized Experiments: Machine Learning for Variance Reduction](https://mlsanity.com/paper/2407.16037/) (2024)
- [Heterogeneous Treatment Effect with Trained Kernels of the Nadaraya-Watson Regression](https://mlsanity.com/paper/2207.09139/) (2022)
- [Optimizing Estimators of Squared Calibration Errors in Classification](https://mlsanity.com/paper/2410.07014/) (2024)
- [Doubly-Robust Estimation of Counterfactual Policy Mean Embeddings](https://mlsanity.com/paper/2506.02793/) (2025)
- [Conditional Distributional Treatment Effects: Doubly Robust Estimation and Testing](https://mlsanity.com/paper/2603.16829/) (2026)
- [Doubly Robust Kernel Statistics for Testing Distributional Treatment Effects](https://mlsanity.com/paper/2212.04922/) (2022)
- [Asymptotically Unbiased Synthetic Control Methods by Moment Matching](https://mlsanity.com/paper/2307.11127/) (2023)
- [Frugal, Flexible, Faithful: Causal Data Simulation via Frengression](https://mlsanity.com/paper/2508.01018/) (2025)
- [Beyond the Average: Distributional Causal Inference under Imperfect Compliance](https://mlsanity.com/paper/2509.15594/) (2025)
- [Kernel Treatment Effects with Adaptively Collected Data](https://mlsanity.com/paper/2510.10245/) (2025)
- [Inference on Variable Importance for Treatment Effect Heterogeneity: Shapley Values and Beyond](https://mlsanity.com/paper/2510.18843/) (2025)
- [BENK: The Beran Estimator with Neural Kernels for Estimating the Heterogeneous Treatment Effect](https://mlsanity.com/paper/2211.10793/) (2022)
- [I-trustworthy Models. A framework for trustworthiness evaluation of probabilistic classifiers](https://mlsanity.com/paper/2501.15617/) (2025)
- [GDR-learners: Orthogonal Learning of Generative Models for Potential Outcomes](https://mlsanity.com/paper/2509.22953/) (2025)
- [ActiveCQ: Active Estimation of Causal Quantities](https://mlsanity.com/paper/2509.24293/) (2025)

## Abstract (arXiv)

We propose to analyse the conditional distributional treatment effect (CoDiTE), which, in contrast to the more common conditional average treatment effect (CATE), is designed to encode a treatment's distributional aspects beyond the mean. We first introduce a formal definition of the CoDiTE associated with a distance function between probability measures. Then we discuss the CoDiTE associated with the maximum mean discrepancy via kernel conditional mean embeddings, which, coupled with a hypothesis test, tells us whether there is any conditional distributional effect of the treatment. Finally, we investigate what kind of conditional distributional effect the treatment has, both in an exploratory manner via the conditional witness function, and in a quantitative manner via U-statistic regression, generalising the CATE to higher-order moments. Experiments on synthetic, semi-synthetic and real datasets demonstrate the merits of our approach.