Transfer Learning-Based DCE-MRI Method for Identifying Differentiation Between Benign and Malignant Breast Tumors

Abstract

In this paper, we propose a transfer learning-based dynamic contrast-enhanced magnetic resonance imaging (DCE-MRI) method for classifying fibroadenoma and invasive ductal carcinoma (IDC) in breast tumors. A total of 207 breast tumors from patients were collected and identified by pathologic diagnosis within 15 days after enhanced DCE-MRI examination; 119 patients (average age 50.52±10.33 years) had fibroadenomas, and 88 patients (average age 42.20±10.10 years) had IDCs. Two lesion-level models were built based on the InceptionV3 and VGG19 models, which were pretrained with the ImageNet dataset. The effects of different depths of transfer learning were examined. The network's performance was evaluated through five-fold cross validation. In the lesion-level models, the model based on Inception V3 obtained better results (area under the receiver operating characteristic curve (AUC) = 0.89) when the weights were frozen before layer-276. The other model based on VGG19 obtained better results (AUC = 0.87) when the weights were frozen before layer-13. Compared with the image-level models, both lesion-level models displayed better discrimination (accuracy increased by 13% and 14%) based on the VGG19 and Inception V3 models, respectively. Our research confirms that transfer learning can be utilized to classify fibroadenomas and IDCs in DCE-MRI images. Different depths of transfer learning result in different performances, and our proposed lesion-level model notably improves the classification accuracy.

Keywords

• Magnetic resonance imaging
• invasive ductal carcinoma
• fibroadenoma
• transfer learning
• lesion classification