We found a match
Your institution may have access to this item. Find your institution then sign in to continue.
- Title
Meta-learning the invariant representation for domain generalization.
- Authors
Jia, Chen; Zhang, Yue
- Abstract
Domain generalization studies how to generalize a machine learning model to unseen distributions. Learning invariant representation across different source distributions has been shown high effectiveness for domain generalization. However, the intrinsic possibility of overfitting in source domains can limit the generalization of invariance when faced with a target domain with large discrepancy to the source domains. To address this problem, we propose a meta-learning algorithm via bilevel optimization for domain generalization, where the inner-loop objective aims to minimize the discrepancy across different source domains while the outer-loop objective aims to minimize the discrepancy between source domains and a potential target domain. We show from a geometric perspective that the proposed algorithm can improve out-of-domain robustness for invariance learning. Empirically, we evaluate on five datasets and achieve the best results among a range of strong domain generalization baselines.
- Subjects
BILEVEL programming; MACHINE learning; GENERALIZATION
- Publication
Machine Learning, 2024, Vol 113, Issue 4, p1661
- ISSN
0885-6125
- Publication type
Article
- DOI
10.1007/s10994-022-06256-y