Related Experiment Video
Updated: Sep 17, 2025
A Swin Transformer-Based Model for Thyroid Nodule Detection in Ultrasound Images
Published on: April 21, 2023
Dual-stage artificial intelligence-powered screening for accurate classification of thyroid nodules: enhancing fine
Dianhuan Tan1, Yue Zhai1, Zhengming Hu1
1Shenzhen Key Laboratory for Drug Addiction and Medication Safety, Department of Ultrasound, Institute of Ultrasonic Medicine, Peking University Shenzhen Hospital, Shenzhen Peking University-The Hong Kong University of Science and Technology Medical Center, Shenzhen, China.
Background:
Accurately differentiating thyroid conditions, particularly malignant and benign nodules, is crucial for effective treatment planning. Fine needle aspiration biopsy (FNAB), the standard diagnostic method for suspected malignancies detected by ultrasound, is invasive and can be unnecessary. This study introduces a novel dual-stage deep learning architecture for four-class classification of thyroid nodules, aiming to improve diagnostic accuracy and reduce unnecessary procedures. The framework integrates segmentation and classification stages, offering more precise and consistent analyses than traditional binary classification, potentially improving guidance for FNAB decisions and reducing invasiveness.
Methods:
Our dual-stage deep learning framework integrates segmentation and classification. The segmentation stage uses K-Net to delineate nodule boundaries. The classification stage employs MobileViT to classify nodules into four categories: papillary carcinoma, medullary carcinoma, nodular goiter with adenomatous hyperplasia, and chronic lymphocytic thyroiditis (CLT).
Results:
K-Net achieved a mean intersection over union of 87.06% in segmenting the nodule boundaries. MobileViT achieved a mean accuracy of 92.33% on the validation set and 90.27% on the test set. The confusion matrix demonstrated high diagonal values, indicating accurate classification across all categories. The model demonstrated a precision of 91.30% and recall of 89.76% on the validation set. Performance was particularly strong for papillary thyroid carcinoma and CLT.
Conclusions:
This approach offers an efficient tool for analyzing thyroid nodules, potentially improving guidance for FNAB decisions and reducing invasiveness. The framework shows promise for clinical application, offering a non-invasive method to aid in the differential diagnosis of thyroid nodules. By integrating advanced segmentation and classification, the model offers a refined approach to thyroid nodule analysis.
