Source-Free Unsupervised Domain Adaptation Fundus Image Segmentation via Entropy Optimization and Anatomical Priors

Abstract

This research focuses on fundus image segmentation within a source-free domain adaptation framework, where the availability of source images during the adaptation phase is limited due to privacy concerns. Although Source-Free Unsupervised Domain Adaptation (SFUDA) methods have seen significant innovative developments in recent years, they still face several challenges which include suboptimal performance due to substantial domain discrepancies, reliance on potentially noisy or inaccurate pseudo-labels during the adaptation process, and a lack of integration with domain-specific prior knowledge. To address these issues, this paper proposes a SFUDA framework via Entropy Optimization and Anatomical Priors (EOAPNet). To alleviate the influence of the divergence between the source and target domains, EOAPNet primarily evaluates the uncertainty (i.e., entropy) of predictions on target domain data and improves the model by focusing on high-entropy pixels or regions. Additionally, a weak-strong augmentation mean-teacher scheme is introduced in EOAPNet, which can enhance the accuracy of pseudo-labels and reduce error propagation. Thirdly, by integrating an anatomical knowledge-based class ratio prior into the overall loss function in the form of a Kullback–Leibler (KL) divergence, EOAPNet also incorporates expert domain knowledge. EOAPNet yields comparable results to several state-of-the-art adaptation techniques in experiments on two retinal image segmentation datasets involving the RIM-ONE-r3 and Drishti-GS datasets.