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
1023
Image ClassificationOffice-Home (test)
Mean Accuracy65.5
328
Image ClassificationPACS (test)
Average Accuracy83.1
279
Image ClassificationPACS
Overall Average Accuracy83.1
270
Domain GeneralizationPACS--
263
Domain GeneralizationPACS (test)
Average Accuracy83.1
225
Image ClassificationOfficeHome
Average Accuracy65.5
161
Domain GeneralizationPACS (leave-one-domain-out)
Art Accuracy84.94
152
Domain GeneralizationOffice-Home (test)
Average Accuracy65.5
121
Showing 10 of 45 rows

Other info

Code

Follow for update