A multi-task model for reliable classification of thyroid nodules in ultrasound images

Thyroid nodules are common, and patients with potential malignant lesions are usually diagnosed using ultrasound imaging to determine further treatment options. This study aims to propose a computer-aided diagnosis method for benign and malignant classification of thyroid nodules in ultrasound images. We propose a novel multi-task framework that combines the advantages of dense connectivity, Squeeze-and-Excitation (SE) connectivity, and Atrous Spatial Pyramid Pooling (ASPP) layer to enhance feature extraction. The Dense connectivity is used to optimize feature reuse, the SE connectivity to optimize feature weights, the ASPP layer to fuse feature information, and a multi-task learning framework to adjust the attention of the network. We evaluate our model using a 10-fold cross-validation approach based on our established Thyroid dataset. We assess the performance of our method using six average metrics: accuracy, sensitivity, specificity, positive predictive value, negative predictive value, and AUC, which are 93.49, 95.54, 91.52, 91.63, 95.47, and 96.84%, respectively. Our proposed method outperforms other classification networks in all metrics, achieving optimal performance. We propose a multi-task model, DSMA-Net, for distinguishing thyroid nodules in ultrasound images. This method can further enhance the diagnostic ability of doctors for suspected cancer patients and holds promise for clinical applications.