arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2103.01134cs.LGcs.CV
收藏

基于推理时间标签保持目标投影的领域泛化

Domain Generalization via Inference-time Label-Preserving Target Projections

IIT Delhi(德里印度理工学院)

Prashant Pandey, Mrigank Raman, Sumanth Varambally, Prathosh AP

摘要:

在一组具有不同统计数据的源域上训练的机器学习模型在看不见的目标域上的推广是一个具有挑战性的问题。虽然已经提出了许多方法来解决这个问题,但它们仅在训练期间利用源数据,而没有利用在推理时可获得单个目标示例的事实。受此启发,我们提出了一种方法,有效地使用目标样本在推理过程中超越单纯的分类。我们的方法有三个组成部分-(i)对源数据的标签保留特征或度量变换,使得源样本根据它们的类而不考虑它们的域而被聚类(ii)在这些特征上训练的生成模型(iii)目标点在源上的标签保留投影-通过使用所学习的度量在所述生成模型的输入空间上求解优化问题,在推理期间对所述特征流形进行优化。最后,将投影目标用于分类器。由于投影的目标特征来自源流形,并且通过设计具有与真实目标相同的标签,因此分类器期望在其上比真实目标表现得更好。我们证明,我们的方法优于国家的最先进的领域泛化方法在多个数据集和任务。

英文摘要:

Generalization of machine learning models trained on a set of source domains on unseen target domains with different statistics, is a challenging problem. While many approaches have been proposed to solve this problem, they only utilize source data during training but do not take advantage of the fact that a single target example is available at the time of inference. Motivated by this, we propose a method that effectively uses the target sample during inference beyond mere classification. Our method has three components - (i) A label-preserving feature or metric transformation on source data such that the source samples are clustered in accordance with their class irrespective of their domain (ii) A generative model trained on the these features (iii) A label-preserving projection of the target point on the source-feature manifold during inference via solving an optimization problem on the input space of the generative model using the learned metric. Finally, the projected target is used in the classifier. Since the projected target feature comes from the source manifold and has the same label as the real target by design, the classifier is expected to perform better on it than the true target. We demonstrate that our method outperforms the state-of-the-art Domain Generalization methods on multiple datasets and tasks.

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

↑