Share your thoughts, 1 month free Claude Pro on usSee more
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
271
Image ClassificationPACS
Overall Average Accuracy83.1
241
Domain GeneralizationPACS--
231
Domain GeneralizationPACS (test)
Average Accuracy83.1
225
Image ClassificationOffice-Home (test)
Mean Accuracy65.5
199
Domain GeneralizationPACS (leave-one-domain-out)
Art Accuracy84.94
152
Image ClassificationOfficeHome
Average Accuracy65.5
137
object recognitionPACS (leave-one-domain-out)
Acc (Art painting)84.2
112
Showing 10 of 45 rows

Other info

Code

Follow for update