ReTaSA: A Nonparametric Functional Estimation Approach for Addressing Continuous Target Shift
Hwanwoo Kim, Xin Zhang, Jiwei Zhao, Qinglong Tian
TL;DR
The paper tackles continuous target shift in regression under unsupervised domain adaptation by recasting the problem as estimating a continuous importance weight $\omega(y)=p_t(y)/p_s(y)$ from an ill-posed Fredholm integral equation. It develops ReTaSA, a nonparametric regularization framework using Tikhonov regularization to stabilize the solution, with a kernel-density-estimation–based data-driven implementation that yields a simple linear system to solve for $\rho(y)=\omega(y)-1$ and thus $\omega(y)$. The authors establish identifiability under a completeness condition, prove consistency and convergence rates for the estimator in a weighted $L^2$ sense, and demonstrate superior performance over baselines on synthetic and real regression datasets. The work provides theoretical guarantees and a practical, computation-friendly method to adapt regression models to target-domain shifts without requiring distribution matching, with extensions to high-dimensional features via black-box mappings. This advances robust regression under distributional shifts by bridging nonparametric inverse problems with domain adaptation theory.
Abstract
The presence of distribution shifts poses a significant challenge for deploying modern machine learning models in real-world applications. This work focuses on the target shift problem in a regression setting (Zhang et al., 2013; Nguyen et al., 2016). More specifically, the target variable y (also known as the response variable), which is continuous, has different marginal distributions in the training source and testing domain, while the conditional distribution of features x given y remains the same. While most literature focuses on classification tasks with finite target space, the regression problem has an infinite dimensional target space, which makes many of the existing methods inapplicable. In this work, we show that the continuous target shift problem can be addressed by estimating the importance weight function from an ill-posed integral equation. We propose a nonparametric regularized approach named ReTaSA to solve the ill-posed integral equation and provide theoretical justification for the estimated importance weight function. The effectiveness of the proposed method has been demonstrated with extensive numerical studies on synthetic and real-world datasets.
