Integrate Domain Knowledge in Training CNN for Ultrasonography Breast Cancer Diagnosis

Jiali Liu,Wanyu Li,Ningbo Zhao,Kunlin Cao,Youbing Yin,Qi Song,Hanbo Chen,Xuehao Gong
DOI: https://doi.org/10.1007/978-3-030-00934-2_96
2018-01-01
Abstract:Breast cancer is the most common cancer in women, and ultrasound imaging is one of the most widely used approach for diagnosis. In this paper, we proposed to adopt Convolutional Neural Network (CNN) to classify ultrasound images and predict tumor malignancy. CNN is a successful algorithm for image recognition tasks and has achieved human-level performance in real applications. To improve the performance of CNN in breast cancer diagnosis, we integrated domain knowledge and conducted multi-task learning in the training process. After training, a radiologist visually inspected the class activation map of the last convolutional layer of trained network to evaluate the result. Our result showed that CNN classifier can not only give reasonable performance in predicting breast cancer, but also propose potential lesion regions which can be integrated into the breast ultrasound system in the future.
What problem does this paper attempt to address?