Efficient Inference under Label Shift in Unsupervised Domain Adaptation
Authors
Research Topics
Paper Information
-
Journal:
Journal of Machine Learning Research -
Added to Tracker:
Sep 08, 2026
Abstract
In many real-world applications, researchers aim to deploy models trained in a source domain to a target domain, where obtaining labeled data is often expensive, time-consuming, or even infeasible. While most existing literature assumes that the source and target data follow the same joint distribution, distribution shifts are common in practice. This paper considers a particular type of distribution shift, label shift, and develops an efficient inference procedure for general parameters characterizing the unlabeled target population. A central idea is to model the outcome density ratio between the labeled source data and unlabeled target data. To this end, we propose a progressive estimation strategy that unfolds in three stages: an initial heuristic guess, a consistent estimation, and ultimately, an efficient estimation. This self-evolving process is novel in the statistical literature and of independent interest. We also highlight the connection between our approach and prediction-powered inference (PPI), which uses machine learning models to improve statistical inference in related settings. We rigorously establish the asymptotic properties of the proposed estimators and demonstrate their superior performance compared to existing methods. Through simulation studies and multiple real-world applications, we illustrate both the theoretical contributions and practical benefits of our approach.
Author Details
Jiwei Zhao
AuthorSeong-ho Lee
AuthorYanyuan Ma
AuthorResearch Topics & Keywords
Machine Learning
Research AreaCitation Information
APA Format
Jiwei Zhao
,
Seong-ho Lee
&
Yanyuan Ma
.
Efficient Inference under Label Shift in Unsupervised Domain Adaptation.
Journal of Machine Learning Research
.
BibTeX Format
@article{paper1590,
title = { Efficient Inference under Label Shift in Unsupervised Domain Adaptation },
author = {
Jiwei Zhao
and Seong-ho Lee
and Yanyuan Ma
},
journal = { Journal of Machine Learning Research },
url = { https://www.jmlr.org/papers/v27/25-2717.html }
}