Abstract:Recently, deep neural networks have demonstrated comparable and even better performance than board-certified ophthalmologists in well-annotated datasets. However, the diversity of retinal imaging devices poses a significant challenge: domain shift, which leads to performance degradation when applying the deep learning models trained on one domain to new testing domains. In this paper, we propose a domain adaptation framework comprising of multi-scale inputs along with multiple domain adaptors applied hierarchically in both feature and output spaces. The proposed training strategy and novel unsupervised domain adaptation framework, called Collaborative Adversarial Domain Adaptation (CADA), can effectively overcome the domain shift challenge. Multi-scale inputs can reduce the information loss due to the pooling layers used in the network for feature extraction, while our proposed CADA is an interactive paradigm that presents an exquisite collaborative adaptation through both adversarial learning and ensembling weights at different network layers. In particular, to produce a better prediction for the unlabeled target domain data, we simultaneously achieve domain invariance and model generalizability via adversarial learning at multi-scale outputs from different levels of network layers and maintaining an exponential moving average (EMA) of the historical weights during training. Without annotating any sample from the target domain, multiple adversarial losses in encoder and decoder layers guide the extraction of domain-invariant features to confuse the domain classifier. Meanwhile, the ensembling of weights via EMA reduces the uncertainty of adapting multiple discriminator learning. Comprehensive experimental results demonstrate that our CADA model incorporating multi-scale input training can overcome performance degradation and outperform state-of-the-art domain adaptation methods in segmenting retinal optic disc and cup from fundus images stemming from the REFUGE, Drishti-GS, and Rim-One-r3 datasets. The code is available at https://github.com/cswin/CADA.

Unsupervised domain adaptation of dynamic extension networks based on class decision boundaries

Attention-based Cross-Layer Domain Alignment for Unsupervised Domain Adaptation

Unsupervised Cross-Modality Domain Adaptation of ConvNets for Biomedical Image Segmentations with Adversarial Loss

Unsupervised Domain Adaptation Network with Category-Centric Prototype Aligner for Biomedical Image Segmentation

Dynamic Instance Domain Adaptation

Unsupervised Domain Adaptation with Target Reconstruction and Label Confusion in the Common Subspace

Domain adaptation via Wasserstein distance and discrepancy metric for chest X-ray image classification

Domain Adaptation for Medical Image Classification Without Source Data

Unsupervised domain adaptation via discriminative feature learning and classifier adaptation from center-based distances

Cross-Domain Extreme Learning Machines for Domain Adaptation

A multi-task unsupervised domain adaptation network for medical image segmentation

Collaborative Unsupervised Domain Adaptation for Medical Image Diagnosis

Attention-guided Partial Domain Adaptation for Automated Pneumonia Diagnosis From Chest X-Ray Images

Unsupervised domain adaptation with deep network based on discriminative class-wise MMD

CADA: Multi-scale Collaborative Adversarial Domain Adaptation for unsupervised optic disc and cup segmentation

Unsupervised Domain Adaptation for Anatomical Landmark Detection

Deep Transfer Network: Unsupervised Domain Adaptation

Unsupervised Domain Adaptation for Low-dose CT Reconstruction via Bayesian Uncertainty Alignment

Deep Asymmetric Transfer Network For Unbalanced Domain Adaptation

Unsupervised domain adaptation for the detection of cardiomegaly in cross-domain chest X-ray images

Unsupervised Domain Adaptation for Neuron Membrane Segmentation based on Structural Features