Saved in:
Bibliographic Details
Main Authors: Testa, Lorenzo, Xu, Qi, Lei, Jing, Roeder, Kathryn
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2505.06452
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909031990296576
author Testa, Lorenzo
Xu, Qi
Lei, Jing
Roeder, Kathryn
author_facet Testa, Lorenzo
Xu, Qi
Lei, Jing
Roeder, Kathryn
contents In modern scientific applications, large volumes of covariate data are readily available, while outcome labels are costly, sparse, and often subject to distribution shift. This asymmetry has spurred interest in semi-supervised (SS) learning, but most existing approaches rely on strong assumptions -- such as missing completely at random (MCAR) labeling or strict positivity -- that put substantial limitations on their practical usefulness. In this work, we introduce a general semiparametric framework for estimation, inference, and efficiency benchmarking in SS settings where labels are missing at random (MAR) and the overlap may vanish as sample size increases. Our framework, that we label D2S3, accommodates a wide range of smooth statistical targets -- including means, linear regression coefficients, quantiles, and causal effects -- and remains valid under high-dimensional nuisance estimation and distributional shift between labeled and unlabeled samples. We extend the theoretical guarantees of augmented inverse probability weighting estimators to preserve double robustness, asymptotic normality, and semiparametric efficiency under this challenging D2S3 regime. A key insight is that classical root-n convergence fails under vanishing overlap; we instead provide corrected asymptotic rates that capture the impact of the decay in overlap. We validate our theory through simulations and demonstrate practical utility in real-world applications on the internet of things and public health where labeled data are scarce.
format Preprint
id arxiv_https___arxiv_org_abs_2505_06452
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Semiparametric semi-supervised learning for general targets under distribution shift and decaying overlap
Testa, Lorenzo
Xu, Qi
Lei, Jing
Roeder, Kathryn
Statistics Theory
In modern scientific applications, large volumes of covariate data are readily available, while outcome labels are costly, sparse, and often subject to distribution shift. This asymmetry has spurred interest in semi-supervised (SS) learning, but most existing approaches rely on strong assumptions -- such as missing completely at random (MCAR) labeling or strict positivity -- that put substantial limitations on their practical usefulness. In this work, we introduce a general semiparametric framework for estimation, inference, and efficiency benchmarking in SS settings where labels are missing at random (MAR) and the overlap may vanish as sample size increases. Our framework, that we label D2S3, accommodates a wide range of smooth statistical targets -- including means, linear regression coefficients, quantiles, and causal effects -- and remains valid under high-dimensional nuisance estimation and distributional shift between labeled and unlabeled samples. We extend the theoretical guarantees of augmented inverse probability weighting estimators to preserve double robustness, asymptotic normality, and semiparametric efficiency under this challenging D2S3 regime. A key insight is that classical root-n convergence fails under vanishing overlap; we instead provide corrected asymptotic rates that capture the impact of the decay in overlap. We validate our theory through simulations and demonstrate practical utility in real-world applications on the internet of things and public health where labeled data are scarce.
title Semiparametric semi-supervised learning for general targets under distribution shift and decaying overlap
topic Statistics Theory
url https://arxiv.org/abs/2505.06452