FDS: Feedback-guided Domain Synthesis with Multi-Source Conditional Diffusion Models for Domain Generalization

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Noori, Mehrdad, Cheraghalikhani, Milad, Bahri, Ali, Hakim, Gustavo Adolfo Vargas, Osowiechi, David, Yazdanpanah, Moslem, Ayed, Ismail Ben, Desrosiers, Christian
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915070679711744
author Noori, Mehrdad
Cheraghalikhani, Milad
Bahri, Ali
Hakim, Gustavo Adolfo Vargas
Osowiechi, David
Yazdanpanah, Moslem
Ayed, Ismail Ben
Desrosiers, Christian
author_facet Noori, Mehrdad
Cheraghalikhani, Milad
Bahri, Ali
Hakim, Gustavo Adolfo Vargas
Osowiechi, David
Yazdanpanah, Moslem
Ayed, Ismail Ben
Desrosiers, Christian
contents Domain Generalization techniques aim to enhance model robustness by simulating novel data distributions during training, typically through various augmentation or stylization strategies. However, these methods frequently suffer from limited control over the diversity of generated images and lack assurance that these images span distinct distributions. To address these challenges, we propose FDS, Feedback-guided Domain Synthesis, a novel strategy that employs diffusion models to synthesize novel, pseudo-domains by training a single model on all source domains and performing domain mixing based on learned features. By incorporating images that pose classification challenges to models trained on original samples, alongside the original dataset, we ensure the generation of a training set that spans a broad distribution spectrum. Our comprehensive evaluations demonstrate that this methodology sets new benchmarks in domain generalization performance across a range of challenging datasets, effectively managing diverse types of domain shifts. The code can be found at: \url{https://github.com/Mehrdad-Noori/FDS.git}.
format Preprint
id arxiv_https___arxiv_org_abs_2407_03588
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle FDS: Feedback-guided Domain Synthesis with Multi-Source Conditional Diffusion Models for Domain Generalization
Noori, Mehrdad
Cheraghalikhani, Milad
Bahri, Ali
Hakim, Gustavo Adolfo Vargas
Osowiechi, David
Yazdanpanah, Moslem
Ayed, Ismail Ben
Desrosiers, Christian
Computer Vision and Pattern Recognition
Domain Generalization techniques aim to enhance model robustness by simulating novel data distributions during training, typically through various augmentation or stylization strategies. However, these methods frequently suffer from limited control over the diversity of generated images and lack assurance that these images span distinct distributions. To address these challenges, we propose FDS, Feedback-guided Domain Synthesis, a novel strategy that employs diffusion models to synthesize novel, pseudo-domains by training a single model on all source domains and performing domain mixing based on learned features. By incorporating images that pose classification challenges to models trained on original samples, alongside the original dataset, we ensure the generation of a training set that spans a broad distribution spectrum. Our comprehensive evaluations demonstrate that this methodology sets new benchmarks in domain generalization performance across a range of challenging datasets, effectively managing diverse types of domain shifts. The code can be found at: \url{https://github.com/Mehrdad-Noori/FDS.git}.
title FDS: Feedback-guided Domain Synthesis with Multi-Source Conditional Diffusion Models for Domain Generalization
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2407.03588