Distributionally Robust Learning for Multi-source Unsupervised Domain Adaptation

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Wang, Zhenyu, Bühlmann, Peter, Guo, Zijian
Formato: Preprint
Publicado: 2023
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866916659805028352
author Wang, Zhenyu
Bühlmann, Peter
Guo, Zijian
author_facet Wang, Zhenyu
Bühlmann, Peter
Guo, Zijian
contents Empirical risk minimization often performs poorly when the distribution of the target domain differs from those of source domains. To address such potential distribution shifts, we develop an unsupervised domain adaptation approach that leverages labeled data from multiple source domains and unlabeled data from the target domain. We introduce a distributionally robust model that optimizes an adversarial reward based on the explained variance across a class of target distributions, ensuring generalization to the target domain. We show that the proposed robust model is a weighted average of conditional outcome models from source domains. This formulation allows us to compute the robust model through the aggregation of source models, which can be estimated using various machine learning algorithms of the users' choice, such as random forests, boosting, and neural networks. Additionally, we introduce a bias-correction step to obtain a more accurate aggregation weight, which is effective for various machine learning algorithms. Our framework can be interpreted as a distributionally robust federated learning approach that satisfies privacy constraints while providing insights into the importance of each source for prediction on the target domain. The performance of our method is evaluated on both simulated and real data.
format Preprint
id arxiv_https___arxiv_org_abs_2309_02211
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Distributionally Robust Learning for Multi-source Unsupervised Domain Adaptation
Wang, Zhenyu
Bühlmann, Peter
Guo, Zijian
Machine Learning
Methodology
Empirical risk minimization often performs poorly when the distribution of the target domain differs from those of source domains. To address such potential distribution shifts, we develop an unsupervised domain adaptation approach that leverages labeled data from multiple source domains and unlabeled data from the target domain. We introduce a distributionally robust model that optimizes an adversarial reward based on the explained variance across a class of target distributions, ensuring generalization to the target domain. We show that the proposed robust model is a weighted average of conditional outcome models from source domains. This formulation allows us to compute the robust model through the aggregation of source models, which can be estimated using various machine learning algorithms of the users' choice, such as random forests, boosting, and neural networks. Additionally, we introduce a bias-correction step to obtain a more accurate aggregation weight, which is effective for various machine learning algorithms. Our framework can be interpreted as a distributionally robust federated learning approach that satisfies privacy constraints while providing insights into the importance of each source for prediction on the target domain. The performance of our method is evaluated on both simulated and real data.
title Distributionally Robust Learning for Multi-source Unsupervised Domain Adaptation
topic Machine Learning
Methodology
url https://arxiv.org/abs/2309.02211