MIC: Masked Image Consistency for Context-Enhanced Domain Adaptation
About
In unsupervised domain adaptation (UDA), a model trained on source data (e.g. synthetic) is adapted to target data (e.g. real-world) without access to target annotation. Most previous UDA methods struggle with classes that have a similar visual appearance on the target domain as no ground truth is available to learn the slight appearance differences. To address this problem, we propose a Masked Image Consistency (MIC) module to enhance UDA by learning spatial context relations of the target domain as additional clues for robust visual recognition. MIC enforces the consistency between predictions of masked target images, where random patches are withheld, and pseudo-labels that are generated based on the complete image by an exponential moving average teacher. To minimize the consistency loss, the network has to learn to infer the predictions of the masked regions from their context. Due to its simple and universal concept, MIC can be integrated into various UDA methods across different visual recognition tasks such as image classification, semantic segmentation, and object detection. MIC significantly improves the state-of-the-art performance across the different recognition tasks for synthetic-to-real, day-to-nighttime, and clear-to-adverse-weather UDA. For instance, MIC achieves an unprecedented UDA performance of 75.9 mIoU and 92.8% on GTA-to-Cityscapes and VisDA-2017, respectively, which corresponds to an improvement of +2.1 and +3.0 percent points over the previous state of the art. The implementation is available at https://github.com/lhoyer/MIC.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Semantic segmentation | GTA5 → Cityscapes (val) | mIoU75.9 | 533 | |
| Semantic segmentation | SYNTHIA to Cityscapes (val) | Rider IoU73 | 435 | |
| Object Detection | Cityscapes to Foggy Cityscapes (test) | mAP47.6 | 196 | |
| Image Classification | Office-Home | Average Accuracy86.2 | 142 | |
| Semantic segmentation | Cityscapes adaptation from Synthia 1.0 (val) | Person IoU81 | 114 | |
| Object Detection | Foggy Cityscapes (test) | mAP (Mean Average Precision)47.6 | 108 | |
| Semantic segmentation | GTA5 to Cityscapes 1.0 (val) | Road IoU97.4 | 98 | |
| Semantic segmentation | BDD100K night | mIoU41.3 | 65 | |
| Semantic segmentation | Cityscapes trained on SYNTHIA (val) | Road IoU86.6 | 60 | |
| Object Detection | Foggy Cityscapes | mAP47.6 | 47 |