Uncertainty-guided Contrastive Learning for Single Source Domain Generalisation
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.
Code (0)
등록된 구현이 없습니다.
Tasks
Contrastive LearningSimilar Papers 제목 키워드 기반
Uncertainty-guided Model Generalization to Unseen Domains
We study a worst-case scenario in generalization: Out-of-domain generalization from a single source. The goal is to learn a robust model from a single source and expect it to generalize over many unknown distributions. T…
Domain Generalizationimage-classificationImage ClassificationMeta-Learning+6Guiding Pseudo-labels with Uncertainty Estimation for Source-free Unsupervised Domain Adaptation
Standard Unsupervised Domain Adaptation (UDA) methods assume the availability of both source and target data during the adaptation. In this work, we investigate Source-free Unsupervised Domain Adaptation (SF-UDA), a spec…
Domain AdaptationSource-Free Domain AdaptationUnsupervised Domain AdaptationUncertainty-guided Open-Set Source-Free Unsupervised Domain Adaptation with Target-private Class Segregation
Standard Unsupervised Domain Adaptation (UDA) aims to transfer knowledge from a labeled source domain to an unlabeled target but usually requires simultaneous access to both source and target data. Moreover, UDA approach…
Contrastive LearningDomain AdaptationNovel Class DiscoveryUnsupervised Domain AdaptationGrad-CL: Source Free Domain Adaptation with Gradient Guided Feature Disalignment
Accurate segmentation of the optic disc and cup is critical for the early diagnosis and management of ocular diseases such as glaucoma. However, segmentation models trained on one dataset often suffer significant perform…
Source-Free Domain AdaptationContrastive LearningWhat Has Been Overlooked in Contrastive Source-Free Domain Adaptation: Leveraging Source-Informed Latent Augmentation within Neighborhood Context
Source-free domain adaptation (SFDA) involves adapting a model originally trained using a labeled dataset ({\em source domain}) to perform effectively on an unlabeled dataset ({\em target domain}) without relying on any …
Continual LearningContrastive LearningDomain AdaptationInformativeness+1