arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2102.05082cs.LG
收藏

基于域密度变换的域不变表示学习

Domain Invariant Representation Learning with Domain Density Transformations

University of Oxford(牛津大学);VinAI Research(越南人工智能研究院)

A. Tuan Nguyen, Toan Tran, Yarin Gal, Atılım Güneş Baydin

摘要:

领域泛化是指我们的目标是根据来自一组源领域的数据训练模型,以便模型可以泛化到看不见的目标领域的问题。在聚合数据集(从所有源域汇集)上简单地训练模型已经被证明是次优的,因为该模型学习到的信息可能是特定于域的,并且不能完美地推广到目标域。为了解决这个问题,一个主要的方法是找到和学习一些领域不变的信息,以便将其用于预测任务。在本文中,我们提出了一个理论上接地的方法来学习域不变的表示,通过强制表示网络是不变的域之间的所有转换函数。我们还展示了如何使用生成对抗网络来学习这种域转换,以在实践中实现我们的方法。我们证明了我们的方法在几个广泛使用的数据集的领域泛化问题的有效性,在所有这些方面,我们取得了竞争力的结果与国家的最先进的模型。

英文摘要:

Domain generalization refers to the problem where we aim to train a model on data from a set of source domains so that the model can generalize to unseen target domains. Naively training a model on the aggregate set of data (pooled from all source domains) has been shown to perform suboptimally, since the information learned by that model might be domain-specific and generalize imperfectly to target domains. To tackle this problem, a predominant approach is to find and learn some domain-invariant information in order to use it for the prediction task. In this paper, we propose a theoretically grounded method to learn a domain-invariant representation by enforcing the representation network to be invariant under all transformation functions among domains. We also show how to use generative adversarial networks to learn such domain transformations to implement our method in practice. We demonstrate the effectiveness of our method on several widely used datasets for the domain generalization problem, on all of which we achieve competitive results with state-of-the-art models.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑