Classification of Breast Cancer Using Radiological Society of North America Data by EfficientNet

Abstract

Breast cancer is a common cancer that affects women all over the world. Therefore, detection at an early stage is crucial for reducing the mortality rate linked to this disease. Mammography is the primary screening method for breast cancer. However, it has drawbacks, including high rates of false-positive and negative results, inter-observer variability, and limited sensitivity with dense breast tissue. To solve such problems, breast cancer was analyzed and classified using mammography images and deep learning models from the Radiological Society of North America (RSNA) database. This database contains processed and raw images from the RSNA that consist of annotated malignancies and clinical data. Using deep learning models based on convolutional neural network (CNN) models such as visual geometry group (VGG), Googlenet, EfficientNet, and Residual Networks, mammograms were classified into cancer or non-cancer categories. In this study, a novel architecture was proposed by combining CNNs and attention mechanisms, which extracted and highlighted the relevant features. A dataset of 8000 patients with 47,000 photographs was used to train and assess the model via 5-fold cross-validation. The results outperformed prior methods using the same database and reached an average accuracy rate of 95%. The results showed that mammography with deep learning methods considerably improved breast cancer detection and diagnosis.

Keywords

• breast cancer
• EfficientNet
• RSNA database
• 5-fold cross-validation
• mammography