Adversarial Dropout Regularization
About
We present a method for transferring neural representations from label-rich source domains to unlabeled target domains. Recent adversarial methods proposed for this task learn to align features across domains by fooling a special domain critic network. However, a drawback of this approach is that the critic simply labels the generated features as in-domain or not, without considering the boundaries between classes. This can lead to ambiguous features being generated near class boundaries, reducing target classification accuracy. We propose a novel approach, Adversarial Dropout Regularization (ADR), to encourage the generator to output more discriminative features for the target domain. Our key idea is to replace the critic with one that detects non-discriminative features, using dropout on the classifier network. The generator then learns to avoid these areas of the feature space and thus creates better features. We apply our ADR approach to the problem of unsupervised domain adaptation for image classification and semantic segmentation tasks, and demonstrate significant improvement over the state of the art. We also show that our approach can be used to train Generative Adversarial Networks for semi-supervised learning.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Classification | CIFAR-10 (test) | -- | 3381 | |
| Semantic segmentation | Cityscapes GTA5 to Cityscapes adaptation (val) | mIoU (Overall)33.3 | 352 | |
| Unsupervised Domain Adaptation | Office-Home | Average Accuracy44.5 | 238 | |
| Classification | SVHN (test) | Error Rate6.26 | 182 | |
| Object Classification | VisDA synthetic-to-real 2017 | Mean Accuracy73.5 | 91 | |
| Unsupervised Domain Adaptation | VisDA unsupervised domain adaptation 2017 | Mean Accuracy74.8 | 87 | |
| Image Classification | VisDA-C (test) | Mean Accuracy79.6 | 76 | |
| Domain Adaptation | Office31 (test) | Mean Accuracy73.7 | 74 | |
| Domain Adaptation | VisDA 2017 (val) | Mean Accuracy73.5 | 52 | |
| Semi-supervised Domain Adaptation | DomainNet 3-shot | Mean Accuracy60.4 | 48 |