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
. 2022 Aug 8:8:e1054.
doi: 10.7717/peerj-cs.1054. eCollection 2022.

Classification of breast cancer using a manta-ray foraging optimized transfer learning framework

Nadiah A Baghdadi  1 , Amer Malki  2 , Hossam Magdy Balaha  3 , Yousry AbdulAzeem  4 , Mahmoud Badawy  3 , Mostafa Elhosseini  2   3
Affiliations

Classification of breast cancer using a manta-ray foraging optimized transfer learning framework

Nadiah A Baghdadi et al. PeerJ Comput Sci. .

Abstract

Due to its high prevalence and wide dissemination, breast cancer is a particularly dangerous disease. Breast cancer survival chances can be improved by early detection and diagnosis. For medical image analyzers, diagnosing is tough, time-consuming, routine, and repetitive. Medical image analysis could be a useful method for detecting such a disease. Recently, artificial intelligence technology has been utilized to help radiologists identify breast cancer more rapidly and reliably. Convolutional neural networks, among other technologies, are promising medical image recognition and classification tools. This study proposes a framework for automatic and reliable breast cancer classification based on histological and ultrasound data. The system is built on CNN and employs transfer learning technology and metaheuristic optimization. The Manta Ray Foraging Optimization (MRFO) approach is deployed to improve the framework's adaptability. Using the Breast Cancer Dataset (two classes) and the Breast Ultrasound Dataset (three-classes), eight modern pre-trained CNN architectures are examined to apply the transfer learning technique. The framework uses MRFO to improve the performance of CNN architectures by optimizing their hyperparameters. Extensive experiments have recorded performance parameters, including accuracy, AUC, precision, F1-score, sensitivity, dice, recall, IoU, and cosine similarity. The proposed framework scored 97.73% on histopathological data and 99.01% on ultrasound data in terms of accuracy. The experimental results show that the proposed framework is superior to other state-of-the-art approaches in the literature review.

Keywords: Breast cancer; Convolutional neural network (CNN); Deep learning (DL); Manta-Ray foraging algorithm (MRFO); Metaheuristic optimization.

PubMed Disclaimer

Conflict of interest statement

The authors declare there are no competing interests.

Figures

Figure 1
Figure 1. General deep learning architecture for image classification.
Figure 2
Figure 2. The proposed hybrid breast cancer recognition framework.
Figure 3
Figure 3. A flowchart summarization of the MRFO steps.
Figure 4
Figure 4. Hyperparameters selection and best combinations graphical summarization.
Figure 5
Figure 5. Confusion matrix related to the two-classes dataset.
Figure 6
Figure 6. Confusion matrix related to the three-classes dataset.
Figure 7
Figure 7. Summarization of the learning and optimization experiments related to the two-classes dataset.
Figure 8
Figure 8. Summarization of the learning and optimization experiments related to the three-classes dataset.
See this image and copyright information in PMC

References

    1. Abualigah L, Elaziz MA, Khasawneh AM, Alshinwan M, Ibrahim RA, Al-qaness MA, Mirjalili S, Sumari P, Gandomi AH. Meta-heuristic optimization algorithms for solving real-world mechanical engineering design problems: a comprehensive survey, applications, comparative analysis, and results. Neural Computing and Applications. 2022;34:4081–4110. doi: 10.1007/s00521-021-06747-4. - DOI
    1. Agarwal N, Sondhi A, Chopra K, Singh G. Transfer learning: survey and classification. In: Tiwari S, Trivedi M, Mishra K, Misra A, Kumar K, Suryani E, editors. Smart innovations in communication and computational sciences. vol. 1168. Springer; Singapore: 2021. (Advances in intelligent systems and computing). - DOI
    1. Al Husaini MAS, Habaebi MH, Gunawan TS, Islam MR, Elsheikh EA, Suliman F. Thermal-based early breast cancer detection using inception V3, inception V4 and modified inception MV4. Neural Computing and Applications. 2022;34(1):333–348. doi: 10.1007/s00521-021-06372-1. - DOI - PMC - PubMed
    1. Alshinwan M, Abualigah L, Shehab M, Elaziz MA, Khasawneh AM, Alabool H, Hamad HA. Dragonfly algorithm: a comprehensive survey of its results, variants, and applications. Multimedia Tools and Applications. 2021;80(10):14979–15016. doi: 10.1007/s11042-020-10255-3. - DOI
    1. Alzubaidi L, Al-Amidie M, Al-Asadi A, Humaidi AJ, Al-Shamma O, Fadhel MA, Zhang J, Santamaría J, Duan Y. Novel transfer learning approach for medical imaging with limited labeled data. Cancer. 2021;13(7):1590. doi: 10.3390/cancers13071590. - DOI - PMC - PubMed

LinkOut - more resources