Unsupervised Domain Adaptation Semantic Segmentation on Thyroid Ultrasound Images Based on Task-Oriented Feature Disentanglement

Zijian Zhang,Ruiguo Yu,Xi Wei,Jie Gao,Mei Yu,Xuewei Li,Zhiqiang Liu
DOI: https://doi.org/10.1109/icme57554.2024.10687567
2024-01-01
Abstract:Unsupervised Domain Adaptation (UDA) methods have become essential for computer-aided diagnostic analysis on medical images due to the advantage of improving the model generalization ability with fewer annotations. The disentanglement-based UDA methods improve the segmentation performance significantly by disentangling the features into domain-specific and domain-invariant components. However, such methods neglect detailed texture in ultrasound images, resulting in inaccurate segmentations of nodules. To address the above problem, we propose a Task-Oriented Feature Disentanglement (TOFD) method to achieve better UDA segmentation on images from different ultrasound machines. TOFD achieves finer-grained disentanglement by the Feature Constraint Module based on maximum entropy. Furthermore, the Domain Label Encoding Module can improve the stability of features by integrating the latent feature of domain bias. Extensive experiments demonstrate that TOFD can effectively enhance the model’s generalization capability and confirm the performance of TOFD on ultrasound images exceeds that of state-of-the-art UDA semantic segmentation.
What problem does this paper attempt to address?