Skip to main page content
U.S. flag

An official website of the United States government

Dot gov

The .gov means it’s official.
Federal government websites often end in .gov or .mil. Before sharing sensitive information, make sure you’re on a federal government site.

Https

The site is secure.
The https:// ensures that you are connecting to the official website and that any information you provide is encrypted and transmitted securely.

Access keys NCBI Homepage MyNCBI Homepage Main Content Main Navigation
. 2024 Jul 26;14(8):792.
doi: 10.3390/jpm14080792.

Breast Cancer Detection and Analytics Using Hybrid CNN and Extreme Learning Machine

Affiliations

Breast Cancer Detection and Analytics Using Hybrid CNN and Extreme Learning Machine

Vidhushavarshini Sureshkumar et al. J Pers Med. .

Abstract

Early detection of breast cancer is essential for increasing survival rates, as it is one of the primary causes of death for women globally. Mammograms are extensively used by physicians for diagnosis, but selecting appropriate algorithms for image enhancement, segmentation, feature extraction, and classification remains a significant research challenge. This paper presents a computer-aided diagnosis (CAD)-based hybrid model combining convolutional neural networks (CNN) with a pruned ensembled extreme learning machine (HCPELM) to enhance breast cancer detection, segmentation, feature extraction, and classification. The model employs the rectified linear unit (ReLU) activation function to enhance data analytics after removing artifacts and pectoral muscles, and the HCPELM hybridized with the CNN model improves feature extraction. The hybrid elements are convolutional and fully connected layers. Convolutional layers extract spatial features like edges, textures, and more complex features in deeper layers. The fully connected layers take these features and combine them in a non-linear manner to perform the final classification. ELM performs classification and recognition tasks, aiming for state-of-the-art performance. This hybrid classifier is used for transfer learning by freezing certain layers and modifying the architecture to reduce parameters, easing cancer detection. The HCPELM classifier was trained using the MIAS database and evaluated against benchmark methods. It achieved a breast image recognition accuracy of 86%, outperforming benchmark deep learning models. HCPELM is demonstrating superior performance in early detection and diagnosis, thus aiding healthcare practitioners in breast cancer diagnosis.

Keywords: breast cancer; computer-aided diagnosis (CAD); convoluted neural networks; extreme learning machine; mammogram; pectoral muscle removal.

PubMed Disclaimer

Conflict of interest statement

The authors declare no conflicts of interest.

Figures

Figure 1
Figure 1
Raw mammogram image and pectoral muscle removed image.
Figure 2
Figure 2
Before and after removal of pectoral muscles.
Figure 3
Figure 3
Pectoral muscle removal.
Figure 4
Figure 4
Results of pectoral muscle removal.
Figure 5
Figure 5
Accuracy of individual versus ensemble ELM.
Figure 6
Figure 6
Training versus validation accuracy of the proposed HCPELM model.
Figure 7
Figure 7
Confusion matrix for HCPELM model.
Figure 8
Figure 8
ROC for HCPELM model.
Figure 9
Figure 9
Comparison of proposed HCPELM and other models.
Figure 10
Figure 10
McNemar test results.

References

    1. Ming C., Viassolo V., Probst-Hensch N., Chappuis P.O., Dinov I.D., Katapodi M.C. Machine learning techniques for personalized breast cancer risk prediction: Comparison with the BCRAT and BOADICEA models. Breast Cancer Res. 2019;21:75. doi: 10.1186/s13058-019-1158-4. - DOI - PMC - PubMed
    1. Shen L., Margolies L.R., Rothstein J.H., Fluder E., McBride R., Sieh W. Deep Learning to Improve Breast Cancer Detection on Screening Mammography. Sci. Rep. 2019;9:12495. doi: 10.1038/s41598-019-48995-4. - DOI - PMC - PubMed
    1. Debelee T.G., Gebreselasie A., Schwenker F., Amirian M., Yohannes D. Classification of Mammograms Using Texture and CNN Based Extracted Features. J. Biomim. Biomater. Biomed. Eng. 2019;42:79–97. doi: 10.4028/www.scientific.net/JBBBE.42.79. - DOI
    1. Debelee T.G., Schwenker F., Ibenthal A., Yohannes D. Survey of deep learning in breast cancer image analysis. Evol. Syst. 2019;11:143–163. doi: 10.1007/s12530-019-09297-2. - DOI
    1. Debelee T.G., Amirian M., Ibenthal A., Palm G., Schwenker F. Classification of Mammograms Using Convolutional Neural Network Based Feature Extraction. LNICST. 2018;244:89–98.

LinkOut - more resources