Uncertainty-guided Contrastive Learning for Single Source Domain Generalisation
Fuente:
arXiv
Guardado en:
| Autores principales: | , , , , |
|---|---|
| Formato: | Preprint |
| Publicado: |
2024
|
| Materias: | |
| Acceso en línea: | |
| Etiquetas: |
Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
|
| _version_ | 1866917614420230144 |
|---|---|
| author | Arsenos, Anastasios Kollias, Dimitrios Petrongonas, Evangelos Skliros, Christos Kollias, Stefanos |
| author_facet | Arsenos, Anastasios Kollias, Dimitrios Petrongonas, Evangelos Skliros, Christos Kollias, Stefanos |
| contents | 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. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2403_07514 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Uncertainty-guided Contrastive Learning for Single Source Domain Generalisation Arsenos, Anastasios Kollias, Dimitrios Petrongonas, Evangelos Skliros, Christos Kollias, Stefanos Computer Vision and Pattern Recognition 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. |
| title | Uncertainty-guided Contrastive Learning for Single Source Domain Generalisation |
| topic | Computer Vision and Pattern Recognition |
| url | https://arxiv.org/abs/2403.07514 |