Minimum Wasserstein distance estimator under covariate shift: closed-form, super-efficiency and irregularity

Read full text
Authors: Junjun Lang, Qiong Zhang, Yukun Liu
Cited by: 0

Abstract

Covariate shift arises when covariate distributions differ between source and target populations while the conditional distribution of the response remains invariant, and it underlies problems in missing data and causal inference. We propose a minimum Wasserstein distance estimation framework for inference under covariate shift that avoids explicit modeling of outcome regressions or importance weights. The resulting W-estimator admits a closed-form expression and is numerically equivalent to the classical 1-nearest neighbor estimator, yielding a new optimal transport interpretation of nearest neighbor methods. We establish root-$n$ asymptotic normality and show that the estimator is not asymptotically linear, leading to super-efficiency relative to the semiparametric efficient estimator under covariate shift in certain regimes, and uniformly in missing data problems. Numerical simulations, along with an analysis of a rainfall dataset, underscore the exceptional performance of our W-estimator.

Fields of Study & Subjects

Advanced Causal Inference Techniques Statistical Methods and Inference Markov Chains and Monte Carlo Methods

Publication details

Journal / source
arXiv (Cornell University)
Publication date
12 January 2026
Document type
Preprint