Minimizing-Entropy and Fourier Consistency Network for Domain Adaptation on Optic Disc and Cup Segmentation

Abstract

Automated segmentation of the optic disc (OD) and optic cup (OC) from different datasets plays an important role in the diagnosis of glaucoma and greatly saves human resources in both data annotation and image segmentation. However, the domain shift between different datasets suppresses the generalization ability of the segmentation network, especially damaging the performance of segmentation in the target domain, which is unlabeled. Therefore, using a transfer learning algorithm or domain adaptation method to enhance the migration ability of segmentation models has become an essential step and has attracted the attention of many researchers. In this paper, we propose an unsupervised domain adaptation network, called the Minimizing-entropy and Fourier Domain Adaptation network (MeFDA), to narrow the discrepancy between the source and target domains and prevent the degradation of segmentation performance. First, we perform adversarial optimization on the entropy maps of the predicted segmentation results to alleviate the domain shift. Then, direct entropy-minimization optimization is applied to the unlabeled target domain data to improve the credibility of the prediction segmentation maps. To enhance the prediction consistency of the target domain data, we augment the target domain dataset through the Fourier transform by replacing the low-frequency part in the target images with that of the source images. Then, a semantic consistency constraint is imposed on the raw images and augmented images of the target domain to improve the prediction consistency of the segmentation model, thereby further narrowing the discrepancy between the source and target domains. Experiments on several public retinal fundus image datasets prove the superiority of MeFDA compared with state-of-the-art methods, and the ablation study analyzes the importance of the different proposed components.