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.

Limits of Model Selection under Transfer Learning

Transfer Learning and Deep Domain Adaptation

Transferable Calibration with Lower Bias and Variance in Domain Adaptation.

A survey on domain adaptation theory: learning bounds and theoretical guarantees

On the Generalization for Transfer Learning: An Information-Theoretic Analysis

An Information-Theoretic Analysis for Transfer Learning: Error Bounds and Applications

Scaling Laws for Transfer

Features are fate: a theory of transfer learning in high-dimensional regression

Transfer Learning for Nonparametric Classification: Minimax Rate and Adaptive Classifier

A More Unified Theory of Transfer Learning

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

Domain Adaptation with Conditional Transferable Components.

Information-theoretic Analysis for Transfer Learning.

An introduction to domain adaptation and transfer learning

The Common Intuition to Transfer Learning Can Win or Lose: Case Studies for Linear Regression

Tight Rates in Supervised Outlier Transfer Learning

An Empirical Study of Scaling Laws for Transfer

Learning Transferable Parameters for Unsupervised Domain Adaptation

Transfer Learning Beyond Bounded Density Ratios

Super-model ecosystem: A domain-adaptation perspective

Transfer Learning with Dynamic Distribution Adaptation