A Deep Learning-Based Approach for Breast BI-RADS Prediction on Shear Wave Elastography Images

Background: Breast cancer is the most common type of cancer among women. About one of every eight women is diagnosed with breast cancer during her lifetime. Malignant tissue is stiffer than normal and benign tissues. This stiffness could be evaluated by elastography. The American College of Radiology (ACR) has published a quality assurance tool named Breast Imaging-Reporting and Data System (BI-RADS) to standardize breast cancer reporting. Although it was originally designed to use with mammography, it now contains several features for various imaging modalities. Among technologies, shear wave elastography (SWE) has shown promising results in breast lesion classification. Objectives: In this paper, we present the capability of the convolutional neural network in the prediction of B-RADS using SWE images. Methods: A comprehensive dataset of SWE images of breast tissue was provided using Esaote MyLab™ 9 and Aupersonic Aixplorer systems. Two hundred images related to breasts with different BI-RADS stages were gathered from the Cancer Institute, Imam Khomeini Medical Center (UICC). The data augmentation with a factor of 10 was applied to the prepared dataset. Some patients had multiple lesions and for each lesion, one or two images were acquired and stored in DICOM standards. The gold standard for the evaluation of the proposed algorithm was a biopsy, which was performed on all the examined lesions. A novel convolutional neural network was applied to the dataset to extract the visual features of images. The architecture was based on Densenet architecture, which was modified for our purpose. We used the network in both pre-training and end-to-end training strategies and the results were compared. The network was pre-trained on the Imagenet dataset due to the lack of a sufficient dataset. On the other hand, with data augmentation, the network underwent a full training strategy. Finally, the classification layer was a softmax layer, which was used to decide on the benignity or malignancy of the lump. The training and testing procedures for tumor classification were employed with five-fold cross-validation. The entire dataset was randomly divided into five equal-sized subsets on the premise that multiple images acquired from the same patient were assigned to the same subset. Four subsets together were used for training and the remaining one for testing and this process was repeated five times such that each subset was used once as the test set. Results: The processing