arXiv Machine Learning

Conditional Distributional Treatment Effects: Doubly Robust Estimation and Testing

The paper introduces a new estimand for conditional distributional treatment effects that captures how treatments influence the entire outcome distribution, including variance and tail risks, in a covariate-dependent manner. It presents a doubly robust estimator that is minimax optimal locally and uses it to construct a test for global homogeneity of conditional potential outcome distributions. The test accommodates discrepancies beyond the maximum mean discrepancy, guarantees valid type‑1 error, is consistent against fixed alternatives, and includes a computationally efficient, permutation‑free algorithm with exact closed‑form expressions for two natural discrepancies.

arXiv Machine Learning
Sep 4

Reliable Selection of Heterogeneous Treatment Effect Estimators

The paper introduces a method for selecting the best heterogeneous treatment effect (HTE) estimator from a set of candidates when the true treatment effect is unobserved. It frames estimator selection as a multiple testing problem and proposes a cross‑fitted, exponentially weighted test statistic that uses a two‑way sample splitting scheme to separate nuisance estimation from weight learning, ensuring stability for inference. The authors prove asymptotic familywise error rate control under mild conditions and demonstrate empirically that their procedure reduces false selections compared to common methods on ACIC 2016, IHDP, and Twins benchmarks.

By Jiayi Guo, Zijun Gao
arXiv Statistics ML
Aug 25

Model-Agnostic Covariate-Assisted Inference on Partially Identified Causal Effects

The paper introduces a model‑agnostic inference framework for partially identified causal effects that leverages covariate information without requiring discrete covariates or accurate conditional distribution estimates. Using duality theory for optimal transport, the method delivers uniformly valid inference in randomized experiments, is doubly robust in observational settings, achieves asymptotic unbiasedness when nuisance parameters converge semiparametrically, and allows multiplier‑bootstrap selection of covariates and models while remaining computationally efficient. Empirical applications show the approach consistently narrows identified sets and confidence intervals without imposing extra structural assumptions.

By Wenlong Ji, Lihua Lei, Asher Spector
arXiv Machine Learning
Jun 8

Automatic, Debiased, and Invariant Counterfactual Generation under General Interventions

arXiv:2606. 07399v1 Announce Type: cross Abstract: Generative models for counterfactual outcomes have great potential to support decision-making under complex interventions, but existing approaches are limited by unstable estimation, poor generalization across environments, and bias from nuisance model misspecification.

By Raphael C Kim, Jingsen Zhu, Ramin Zabih, Michele Santacatterina
arXiv Machine Learning
Jun 4

Orthogonal Learner for Estimating Heterogeneous Long-Term Treatment Effects

arXiv:2604. 00915v2 Announce Type: replace Abstract: Estimation of heterogeneous long-term treatment effects (HLTEs) is relevant for personalized decision-making in marketing, economics, and medicine, where short-term observational datasets are often combined with long-term observational datasets.

By Haorui Ma, Dennis Frauen, Valentyn Melnychuk, Stefan Feuerriegel