Detection and Recognition of Ultrasound Breast Nodules Based on Semi-supervised deep learning

Yanhua Gao,Bo Liu,Yuan Zhu,Lin Chen,Miao Tan,Xiaozhou Xiao,Youmin Guo
DOI: https://doi.org/10.1101/2020.04.24.20078816
2020-04-29
Abstract:Abstract Background The successful application of deep learning in medical images requires a large amount of annotation data for supervised training. However, massive labeling of medical data is expensive and time consuming. This paper proposes a semi-supervised deep learning method for the detection and classification of benign and malignant breast nodules in ultrasound images, which include two phases. Methods The nodule position in the ultrasound image is firstly detected using the faster RCNN network. Second, the recognition network is used to identify the benign and malignant types of nodules. The method in this paper uses a semi-supervised learning strategy, using 800 labeled nodules and 4396 unlabeled nodules. Results Based on mean teacher training strategy, the proposed semi-supervised network has obtained excellent results, which is similar to currently used with supervised training networks. On the two test data sets, the AUC of semi-supervised learning and supervised learning were: 93.7% vs 94.2% and 92% vs 92.3%. Conclusions The paper proves that semi-supervised learning strategies have good application potential in medical images. Based on a special learning strategy, the result of semi-supervised learning is expected to achieve close or even achieve similar result of supervised deep learning, which only need a small number of labeled samples and a large number of unlabeled samples. It means deep learning analysis of breast lesion will be more feasible and more efficient.
What problem does this paper attempt to address?