期刊
MACHINE LEARNING
卷 111, 期 5, 页码 1651-1670出版社
SPRINGER
DOI: 10.1007/s10994-021-06088-2
关键词
-
资金
- OATMIL of the French, LEAUDS [ANR-17-CE23-0012, ANR-18-CE23]
- French National Research Agency (ANR) [ANR-19-P3IA-0002]
- 3rd Programme d'Investissements d'Avenir [ANR-18-EUR-0006-02]
- Chair Challenging Technology for Responsible Energy
- Fondation de l'Ecole polytechnique - TOTAL
This study addresses the problem of unsupervised domain adaptation under generalized target shift, showing the importance of aligning marginals and class-conditional distributions across domains for good generalization. The proposed learning framework minimizes importance weighted loss in the source domain and utilizes a Wasserstein distance between weighted marginals, with an estimator of target label proportion blending mixture estimation and optimal matching by optimal transport. Experimental results demonstrate the method outperforms competitors on various domain adaptation problems.
We address the problem of unsupervised domain adaptation under the setting of generalized target shift (joint class-conditional and label shifts). For this framework, we theoretically show that, for good generalization, it is necessary to learn a latent representation in which both marginals and class-conditional distributions are aligned across domains. For this sake, we propose a learning problem that minimizes importance weighted loss in the source domain and a Wasserstein distance between weighted marginals. For a proper weighting, we provide an estimator of target label proportion by blending mixture estimation and optimal matching by optimal transport. This estimation comes with theoretical guarantees of correctness under mild assumptions. Our experimental results show that our method performs better on average than competitors across a range domain adaptation problems including digits,VisDA and Office. Code for this paper is available at https://github.com/arakotom/mars_domain_adaptation.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据