Breast Ultrasound Image Analysis Based on Transfer Learning

Chaoyun Wang,Hongwei Mo
DOI: https://doi.org/10.1145/3445815.3445835
2020-01-01
Abstract:Breast cancer is the most common cancer in women, and ultrasound diagnosis is one of the most common diagnostic methods. This paper adopted the transfer learning method to classify benign and malignant breast tumours by ultrasound images and predict molecular subtypes. To study the effect of the transfer learning model on breast cancer diagnosis in the benign and malignant tumour classification, we tested whether different models were transferred learning. After transfer learning, the Xception model reached 87.75% in tumour classification accuracy, superior to other models without transfer learning. In predicting and diagnosing molecular subtypes, the multiple-transfer learning Xception model's recognition accuracy reached 74.3%, and the AUC value reached 0.8801. Compared with the Xception model without transfer learning, the accuracy was improved by 4.02%, and the single-transfer learning Xception model was improved by 3.22%, achieving a relatively excellent classification effect. Our results show that the application of transfer learning can improve breast cancer ultrasound images' diagnostic accuracy. Multiple transfer learning can improve the accuracy of molecular subtypes prediction in a small sample of breast cancer.
What problem does this paper attempt to address?