Abstract:Practitioners often deploy a learned prediction model in a new environment where the joint distribution of covariate and response has shifted. In observational data, the distribution shift is often driven by unobserved confounding factors lurking in the environment, with the underlying mechanism unknown. Confounding can obfuscate the definition of the best prediction model (concept shift) and shift covariates to domains yet unseen (covariate shift). Therefore, a model maximizing prediction accuracy in the source environment could suffer a significant accuracy drop in the target environment. This motivates us to study the domain adaptation problem with observational data: given labeled covariate and response pairs from a source environment, and unlabeled covariates from a target environment, how can one predict the missing target response reliably? We root the adaptation problem in a linear structural causal model to address endogeneity and unobserved confounding. We study the necessity and benefit of leveraging exogenous, invariant covariate representations to cure concept shifts and improve target prediction. This further motivates a new representation learning method for adaptation that optimizes for a lower-dimensional linear subspace and, subsequently, a prediction model confined to that subspace. The procedure operates on a non-convex objective-that naturally interpolates between predictability and stability/invariance-constrained on the Stiefel manifold. We study the optimization landscape and prove that, when the regularization is sufficient, nearly all local optima align with an invariant linear subspace resilient to both concept and covariate shift. In terms of predictability, we show a model that uses the learned lower-dimensional subspace can incur a nearly ideal gap between target and source risk. Three real-world data sets are investigated to validate our method and theory.

Low-Dimensional Density Ratio Estimation for Covariate Shift Correction

Adapting to Continuous Covariate Shift via Online Density Ratio Estimation

Double Debiased Covariate Shift Adaptation Robust to Density-Ratio Estimation

Robust Covariate Shift Adaptation for Density-Ratio Estimation

Discriminative Density-ratio Estimation

Clustering and Prediction with Variable Dimension Covariates

Tailoring Density Ratio Weight for Covariate Shift Adaptation

Transfer Learning under Covariate Shift: Local $k$-Nearest Neighbours Regression with Heavy-Tailed Design

Estimation of prediction error with known covariate shift

Maximum Likelihood Estimation is All You Need for Well-Specified Covariate Shift

Contrastive Identification of Covariate Shift in Image Data

TransFusion: Covariate-Shift Robust Transfer Learning for High-Dimensional Regression

A One-step Approach to Covariate Shift Adaptation

Double-Weighting for Covariate Shift Adaptation

Covariate-Shift Generalization Via Random Sample Weighting.

High-Dimensional Kernel Methods under Covariate Shift: Data-Dependent Implicit Regularization

Nearest Neighbor Sampling for Covariate Shift Adaptation

Transfer Learning with General Estimating Equations

Covariate shift in nonparametric regression with Markovian design

On a Dimension Reduction Regression with Covariate Adjustment

Learning When the Concept Shifts: Confounding, Invariance, and Dimension Reduction