arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2403.07514cs.CV

不确定性引导的对比学习用于单源域泛化

Uncertainty-guided Contrastive Learning for Single Source Domain Generalisation

Hellenic Drones S.A.(希腊无人机公司)

Anastasios Arsenos, Dimitrios Kollias, Evangelos Petrongonas, Christos Skliros, Stefanos Kollias

AI总结:

本文提出对比不确定性域泛化网络(CUDGNet),通过虚构域生成器增强源数据容量并利用对比学习学习域不变表示,在单源域泛化任务中超越现有方法,最高提升7.08%,同时实现高效不确定性估计。

摘要:

在单域泛化的背景下,目标是让仅在单一领域数据上训练的模型在面对各种不熟悉的领域时展现出强大的性能。在本文中,我们介绍了一种新颖的模型,称为对比不确定性域泛化网络(CUDGNet)。其关键思想是通过虚构域生成器在输入和标签空间中增强源容量,并通过对比学习联合学习每个类别的域不变表示。在两个单源域泛化(SSDG)数据集上的大量实验证明了我们方法的有效性,其性能优于最先进的单域泛化方法,最高提升达7.08%。我们的方法还通过生成器子网络的单次前向传播在推理时提供了高效的不确定性估计。

英文摘要:

In the context of single domain generalisation, the objective is for models that have been exclusively trained on data from a single domain to demonstrate strong performance when confronted with various unfamiliar domains. In this paper, we introduce a novel model referred to as Contrastive Uncertainty Domain Generalisation Network (CUDGNet). The key idea is to augment the source capacity in both input and label spaces through the fictitious domain generator and jointly learn the domain invariant representation of each class through contrastive learning. Extensive experiments on two Single Source Domain Generalisation (SSDG) datasets demonstrate the effectiveness of our approach, which surpasses the state-of-the-art single-DG methods by up to $7.08\%$. Our method also provides efficient uncertainty estimation at inference time from a single forward pass through the generator subnetwork.

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

↑