Uncertainty-guided Contrastive Learning for Single Source Domain Generalisation
ICASSP 2024 - 2024 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP)(2024)
摘要
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.
更多查看译文
关键词
Domain Generalisation,Augmentation,Adversarial & Contrastive Learning,Uncertainty estimation
AI 理解论文
溯源树
样例
生成溯源树,研究论文发展脉络
Chat Paper
正在生成论文摘要