Mar, 2024

基于不确定性引导的对比学习在单源领域泛化中的应用

TL;DR通过引入一种新型模型 Contrastive Uncertainty Domain Generalisation Network (CUDGNet) ,本文提出了在单一域上训练的模型能够在面对多个陌生领域时展现出强大性能的方法。该方法通过虚拟域生成器增强输入和标签空间的源容量,并通过对比学习来联合学习每个类别的域不变表示。在两个 Single Source Domain Generalisation (SSDG) 数据集上的大量实验证明了我们方法的有效性,超过了最先进的单 - DG 方法高达 7.08%。我们的方法还能够通过生成子网络的单次前向传递有效地进行不确定性估计。