An effective convolutional neural network for classification of benign and malignant breast and thyroid tumors from ultrasound images

Phys Eng Sci Med. 2023 Sep;46(3):995-1013. doi: 10.1007/s13246-023-01262-3. Epub 2023 May 17.

Abstract

Breast and thyroid cancers are the two most common cancers among women worldwide. The early clinical diagnosis of breast and thyroid cancers often utilizes ultrasonography. Most of the ultrasound images of breast and thyroid cancer lack specificity, which reduces the accuracy of ultrasound clinical diagnosis. This study attempts to develop an effective convolutional neural network (E-CNN) for the classification of benign and malignant breast and thyroid tumors from ultrasound images. The 2-Dimension (2D) ultrasound images of 1052 breast tumors were collected, and 8245 2D tumor images were obtained from 76 thyroid cases. We performed tenfold cross-validation on breast and thyroid data, with a mean classification accuracy of 0.932 and 0.902, respectively. In addition, the proposed E-CNN was applied to classify and evaluate 9297 mixed images (breast and thyroid images). The mean classification accuracy was 0.875, and the mean area under the curve (AUC) was 0.955. Based on data in the same modality, we transferred the breast model to classify typical tumor images of 76 patients. The finetuning model achieved a mean classification accuracy of 0.945, and a mean AUC of 0.958. Meanwhile, the transfer thyroid model realized a mean classification accuracy of 0.932, and a mean AUC of 0.959, on 1052 breast tumor images. The experimental results demonstrate the ability of the E-CNN to learn the features and classify breast and thyroid tumors. Besides, it is promising to classify benign and malignant tumors from ultrasound images with the transfer model under the same modality.

Keywords: Breast cancer; Classification; Convolutional neural network (CNN); Deep learning (DL); Thyroid cancer; Transfer learning (TF).

MeSH terms

  • Breast* / diagnostic imaging
  • Diagnosis, Computer-Assisted
  • Female
  • Humans
  • Neural Networks, Computer
  • Thyroid Neoplasms*
  • Ultrasonography