Machine Learning Prediction of Lymph Node Metastasis in Breast Cancer: Performance of a Multi-institutional MRI-based 4D Convolutional Neural Network

Purpose: To develop a custom deep convolutional neural network (CNN) for noninvasive prediction of breast cancer nodal metastasis. Materials and Methods: This retrospective study included patients with newly diagnosed primary invasive breast cancer with known pathologic (pN) and clinical nodal (cN) status who underwent dynamic contrast-enhanced (DCE) breast MRI at the authors' institution between July 2013 and July 2016. Clinicopathologic data (age, estrogen receptor and human epidermal growth factor 2 status, Ki-67 index, and tumor grade) and cN and pN status were collected. A four-dimensional (4D) CNN model integrating temporal information from dynamic image sets was developed. The convolutional layers learned prognostic image features, which were combined with clinicopathologic measures to predict cN0 versus cN+ and pN0 versus pN+ disease. Performance was assessed with the area under the receiver operating characteristic curve (AUC), with fivefold nested cross-validation. Results: Data from 350 female patients (mean age, 51.7 years +/- 11.9 [SD]) were analyzed. AUC, sensitivity, and specificity values of the 4D hybrid model were 0.87 (95% CI: 0.83, 0.91), 89% (95% CI: 79%, 93%), and 76% (95% CI: 68%, 88%) for differentiating pN0 versus pN+ and 0.79 (95% CI: 0.76, 0.82), 80% (95% CI: 77%, 84%), and 62% (95% CI: 58%, 67%), respectively, for differentiating cN0 versus cN+. Conclusion: The proposed deep learning model using tumor DCE MR images demonstrated high sensitivity in identifying breast cancer lymph node metastasis and shows promise for potential use as a clinical decision support tool.