ArXiv

Semiparametric Efficient Test for Interpretable Distributional Treatment Effects

Authors
Houssam Zenati, Arthur Gretton
Categories
stat.ML, cs.LG
arXiv
https://arxiv.org/abs/2605.08034v1
PDF
https://arxiv.org/pdf/2605.08034v1

Brief

Distributional treatment effects that leave means unchanged are targeted by DR-ME, a semiparametrically efficient finite-location test (Zenati & Gretton, arXiv:2605.08034v1, 2026-05-08). From observational data it derives orthogonal doubly robust kernel features whose centered oracle is the canonical gradient; for fixed locations the test is chi-square calibrated and has noncentral chi-square local power with covariance whitening optimizing local SNR. Sample splitting preserves post-selection validity; experiments show near-nominal Type-I error and competitive power, with learned locations that localize effects in a semi-synthetic medical-imaging study.

Source evidence

Abstract

Distributional treatment effects can be invisible to means: a treatment may preserve average outcomes while changing tails, modes, dispersion, or rare-event probabilities. Kernel tests can detect discrepancies between interventional outcome laws, but global tests do not reveal where the laws differ. We propose DR-ME, to our knowledge the first semiparametrically efficient finite-location test for interpretable distributional treatment effects. DR-ME evaluates an interventional kernel witness at learned outcome locations, returning causal-discrepancy coordinates rather than only a global rejection. From observational data, we derive orthogonal doubly robust kernel features whose centered oracle form is the canonical gradient of this finite witness. For fixed locations, we characterize the local testing limit: DR-ME is chi-square calibrated under the null, has noncentral chi-square local power, and uses the covariance whitening that optimizes local signal-to-noise for discrepancies visible through the selected coordinates. This efficient local-power geometry yields a principled location-learning criterion, with sample splitting preserving post-selection validity. Experiments show near-nominal type-I error, competitive power against global doubly robust kernel tests, and interpretable learned locations that localize distributional effects in a semi-synthetic medical-imaging study.