Current browse context:
cs.LG
Change to browse by:
References & Citations
Computer Science > Machine Learning
Title: Optimal Representations for Covariate Shift
(Submitted on 31 Dec 2021 (v1), last revised 14 Mar 2022 (this version, v2))
Abstract: Machine learning systems often experience a distribution shift between training and testing. In this paper, we introduce a simple variational objective whose optima are exactly the set of all representations on which risk minimizers are guaranteed to be robust to any distribution shift that preserves the Bayes predictor, e.g., covariate shifts. Our objective has two components. First, a representation must remain discriminative for the task, i.e., some predictor must be able to simultaneously minimize the source and target risk. Second, the representation's marginal support needs to be the same across source and target. We make this practical by designing self-supervised objectives that only use unlabelled data and augmentations to train robust representations. Our objectives give insights into the robustness of CLIP, and further improve CLIP's representations to achieve SOTA results on DomainBed.
Submission history
From: Yangjun Ruan [view email][v1] Fri, 31 Dec 2021 21:02:24 GMT (3439kb,D)
[v2] Mon, 14 Mar 2022 19:04:38 GMT (2191kb,D)
Link back to: arXiv, form interface, contact.