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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Person Re-Identification | Market1501 (test) | Rank-1 Accuracy60.9 | 1264 | |
| Person Re-Identification | Duke MTMC-reID (test) | Rank-150.6 | 1018 | |
| Image Classification | PACS (test) | Average Accuracy83.1 | 254 | |
| Image Classification | PACS | Overall Average Accuracy83.1 | 230 | |
| Domain Generalization | PACS (test) | Average Accuracy83.1 | 225 | |
| Domain Generalization | PACS | Accuracy (Art)84.7 | 221 | |
| Image Classification | Office-Home (test) | Mean Accuracy65.5 | 199 | |
| Domain Generalization | PACS (leave-one-domain-out) | Art Accuracy84.94 | 146 | |
| Image Classification | OfficeHome | Average Accuracy65.5 | 131 | |
| object recognition | PACS (leave-one-domain-out) | Acc (Art painting)84.2 | 112 |