Classification of Breast Pathology based on Transfer Learning by MobileNet
Abstract: Breast cancer is the most common cancer among women worldwide. By using artificial intelligent technique, the efficiency of cancer diagnosis can be effectively improved. However, the computer-aided diagnosis (CAD) has problems such as long training time for large-resolution pathological images and insufficient data that can be marked for training. In this article, a transfer learning model for pathological diagnosis of breast cancer is developed to overcome these problems. MobileNet was adopted to train breast pathology images under four different resolutions (40X, 100X, 200X, 400X). A transfer learning framework was established to distinguish benign and malignant breast pathologies and eight subtypes. The accuracy of the two-class model at the best magnification (200X) can reach 91.24%, and the average accuracy is 89.31%. At the same time, the multi-classification model of eight subtypes of pathological slices also achieved quite satisfactory results. It is show that the presented transfer learning framework has great potential in exploring the CAD technique for breast cancer.
Keywords: Breast cancer; Pathological image; Computer aided diagnosis; Transfer learning.