Uncertainty-Guided Dual Task Framework for Semi-Supervised Segmentation of Thyroid Nodules
Xiang Ying,Jizhe Zhang,Jie Gao,Mei Yu,Ruixuan Zhang,Jialin Zhu,Xuewei Li,Ruiguo Yu
DOI: https://doi.org/10.3233/faia240527
2024-01-01
Abstract:Since ultrasound imaging technique is convenient and real-time, it plays a crucial role in diagnosing thyroid nodules. With the development of deep learning, computer-aided diagnosis models have been widely applied to diagnose thyroid nodules, in which thyroid nodule segmentation is a basic yet essential task. Semi-supervised learning is a popular topic for thyroid ultrasound images segmentation under limited annotations. However, current mainstream semi-supervised methods (e.g., for nature image scenes) commonly produce poor segmentation results when facing thyroid ultrasound images. We analyze that there are two main reasons: First, since these methods lack the targeted learning for the ambiguous regions of images that may contain complementary clues for segmentation, the model will likely be over-fitting in the regions that are easy to predict, leading to fail to make full use of the unlabeled data. Second, these methods lack the shape constraint on thyroid nodules, resulting in incomplete segmentation shape on the nodule boundary. To address the above issues, we propose a novel Uncertainty-guided Dual Task Framework (UDTF). Concretely, we propose an Uncertainty Region Selection Module (URSM) to guide the segmentation task to learn from the ambiguous regions calculated by prototype under the constraint of consistency regularization. Additionally, to further keep the integrity of the nodules, we propose a Dual Task Module (DTM) to impose the shape constraint on thyroid nodules by exploring the task-level consistency between the segmentation and auxiliary reconstruction task. Extensive experiments are conducted on two thyroid ultrasound image datasets, including a private Philips Thyroid Ultrasound dataset and a public TN3K dataset. The results show that UDTF achieves superior performance compared to several other state-of-the-art methods.