Automatic deep learning method for detection and classification of breast lesions in dynamic contrast-enhanced magnetic resonance imaging.

ABSTRACT

Background: The purpose of this study was to develop a deep learning-based system with a cascade feature pyramid network for the detection and classification of breast lesions in dynamic contrast-enhanced magnetic resonance imaging (DCE-MRI). Methods: This retrospective study enrolled 191 consecutive patients with pathological confirmed breast lesions who underwent preoperative magnetic resonance imaging (MRI) at the Second Affiliated Hospital of Xi'an Jiaotong University. Patients were randomly divided into a training set comprising 153 patients with 126 malignant and 27 benign lesions and a validation set containing 38 patients with 31 malignant and 7 benign lesions under 5-fold cross-validation. Two radiologists annotated the location and classification of all lesions. After augmentation with pseudo-color image fusion, MRI images were fed into the developed cascade feature pyramid network system, feature pyramid network, and faster region-based convolutional neural network (CNN) for breast lesion detection and classification, respectively. The performance on large (>2 cm) and small (≤2 cm) breast lesions was further evaluated. Average precision (AP), mean AP, F1-score, sensitivity, and false positives were used to evaluate different systems. Cohen's kappa scores were calculated to assess agreement between different systems, and Student's t-test and the Holm-Bonferroni method were used to compare multiple groups. Results: The cascade feature pyramid network system outperformed the other systems with a mean AP and highest sensitivity of 0.826±0.051 and 0.970±0.014 (at 0.375 false positives), respectively. The F1-score of the cascade feature pyramid network in real detection was also superior to that of the other systems at both the slice and patient levels. The mean AP values of the cascade feature pyramid network reached 0.779±0.152 and 0.790±0.080 in detecting large and small lesions, respectively. Especially for small lesions, the cascade feature pyramid network achieved the best performance in detecting benign and malignant breast lesions at both the slice and patient levels. Conclusions: The deep learning-based system with the developed cascade feature pyramid network has the potential to detect and classify breast lesions on DCE-MRI, especially small lesions.