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

by   Junhyung Park, et al.

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.



There are no comments yet.


page 1

page 2

page 3

page 4


A U-statistic-based test of treatment effect heterogeneity

Many studies include a goal of determining whether there is treatment ef...

A Fundamental Measure of Treatment Effect Heterogeneity

In this paper we offer an asymptotically efficient, non-parametric way t...

Treatment Effect Risk: Bounds and Inference

Since the average treatment effect (ATE) measures the change in social w...

Estimation of Partially Conditional Average Treatment Effect by Hybrid Kernel-covariate Balancing

We study nonparametric estimation for the partially conditional average ...

Adapting Neural Networks for Uplift Models

Uplift is a particular case of individual treatment effect modeling. Suc...

Efficient Discovery of Heterogeneous Treatment Effects in Randomized Experiments via Anomalous Pattern Detection

The randomized experiment is an important tool for inferring the causal ...

Higher-Order Orthogonal Causal Learning for Treatment Effect

Most existing studies on the double/debiased machine learning method con...
This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.