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.

On Learning Invariant Representation for Domain Adaptation

Trust-aware Conditional Adversarial Domain Adaptation with Feature Norm Alignment.

Learning Domain-Invariant Relationship with Instrumental Variable for Domain Generalization

Domain Generalization Via Conditional Invariant Representations

The Causal Structure of Domain Invariant Supervised Representation Learning

Invariant and Sufficient Supervised Representation Learning

Robust Domain Adaptation: Representations, Weights and Inductive Bias

A General Upper Bound for Unsupervised Domain Adaptation

On Learning Domain-Invariant Representations for Transfer Learning with Multiple Sources

Hidden Covariate Shift: A Minimal Assumption For Domain Adaptation

Domain-Invariant Adversarial Learning for Unsupervised Domain Adaption.

Learning Invariant Representations and Risks for Semi-supervised Domain Adaptation

Learning explicitly transferable representations for domain adaptation

Label-Noise Robust Domain Adaptation

Domain Generalization In Robust Invariant Representation

Domain-Adversarial Training of Neural Networks

Respecting Domain Relations: Hypothesis Invariance for Domain Generalization

Domain Invariant and Class Discriminative Feature Learning for Visual Domain Adaptation

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

Invariant and Transportable Representations for Anti-Causal Domain Shifts

Model-Induced Generalization Error Bound for Information-Theoretic Representation Learning in Source-Data-Free Unsupervised Domain Adaptation