Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Deep Domain-Adversarial Image Generation for Domain Generalisation

About

Machine learning models typically suffer from the domain shift problem when trained on a source dataset and evaluated on a target dataset of different distribution. To overcome this problem, domain generalisation (DG) methods aim to leverage data from multiple source domains so that a trained model can generalise to unseen domains. In this paper, we propose a novel DG approach based on \emph{Deep Domain-Adversarial Image Generation} (DDAIG). Specifically, DDAIG consists of three components, namely a label classifier, a domain classifier and a domain transformation network (DoTNet). The goal for DoTNet is to map the source training data to unseen domains. This is achieved by having a learning objective formulated to ensure that the generated data can be correctly classified by the label classifier while fooling the domain classifier. By augmenting the source training data with the generated unseen domain data, we can make the label classifier more robust to unknown domain changes. Extensive experiments on four DG datasets demonstrate the effectiveness of our approach.

Kaiyang Zhou, Yongxin Yang, Timothy Hospedales, Tao Xiang• 2020

Related benchmarks

TaskDatasetResultRank
Person Re-IdentificationMarket1501 (test)
Rank-1 Accuracy60.9
1264
Person Re-IdentificationDuke MTMC-reID (test)
Rank-150.6
1018
Image ClassificationPACS (test)
Average Accuracy83.1
254
Image ClassificationPACS
Overall Average Accuracy83.1
230
Domain GeneralizationPACS (test)
Average Accuracy83.1
225
Domain GeneralizationPACS
Accuracy (Art)84.7
221
Image ClassificationOffice-Home (test)
Mean Accuracy65.5
199
Domain GeneralizationPACS (leave-one-domain-out)
Art Accuracy84.94
146
Image ClassificationOfficeHome
Average Accuracy65.5
131
object recognitionPACS (leave-one-domain-out)
Acc (Art painting)84.2
112
Showing 10 of 45 rows

Other info

Code

Follow for update