Breast Cancer Diagnosis Method Based on Multitask Learning and BI-RADS Assessments

Chaoyun Wang,Hongwei Mo
DOI: https://doi.org/10.1145/3433996.3434045
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 used the deep transfer learning network to classify breast ultrasound images as benign and malignant. To improve diagnosis performance, we designed a multitask deep learning classification network for breast tumours and integrated the BI-RADS label information of breast ultrasound images during the training process. In BI-RADS classification accuracy, the multitask deep learning network reached the AUC value of 0.836, superior to the single-task network of 0.826. In terms of benign and malignant tumours, the classification accuracy reached 90.7%, which was 2.9% higher than that of a single-task network. In the combined diagnosis, the model prediction results combined with the BI-RADS category reviewed by the Reader achieved a classification accuracy of 93.4%, which was improved by 2.7% compared with the multitask model the AUC value reached 0.982, achieving an excellent diagnostic effect. Our results show that the multitask deep learning network combined with the BI-RADS assessments can improve the training accuracy. The strategy of deep learning model combined with Reader's assessment of BI-RADS labels can significantly improve the diagnostic accuracy.
What problem does this paper attempt to address?