IMR Press / CEOG / Volume 50 / Issue 12 / DOI: 10.31083/j.ceog5012271
Open Access Original Research
A Novel Artificial Intelligence Techniques for Women Breast Cancer Classification Using Ultrasound Images
Show Less
1 Department of Information and Communication Engineering, Tianjin University, 300072 Tianjin, China
2 Department of Information Technology and Decision Sciences, University of Energy and Natural Resources, 00233 Sunyani, Ghana
3 Department of Computer Science and Engineering, University of New South Wales, Sydney, NSW 2052, Australia
4 International Divisions, Ajeenkya D. Y. Patil University, 412105 Pune, India
5 Research Division, Swiss School of Business and Management, 1213 Geneva, Switzerland
*Correspondence: afrifastephen@tju.edu.cn (Stephen Afrifa); v.varadarajan@unsw.edu.au (Vijayakumar Varadarajan)
Clin. Exp. Obstet. Gynecol. 2023, 50(12), 271; https://doi.org/10.31083/j.ceog5012271
Submitted: 5 June 2023 | Revised: 18 August 2023 | Accepted: 25 August 2023 | Published: 26 December 2023
Copyright: © 2023 The Author(s). Published by IMR Press.
This is an open access article under the CC BY 4.0 license.
Abstract

Background: Females benefit from ultrasound screening and diagnosis of breast cancer, and artificial intelligence has enabled the automatic identification of medical conditions on medical imaging. Methods: This study aimed to develop machine learning (ML) and deep learning (DL) models for the detection and classification of breast cancer in a breast ultrasound image (BUSI) and United States (US) ultrasound images datasets and to compare the models’ performance to previous studies. The ultrasound scans were collected from women between the ages of 25 and 75. The dataset contains 780 images with a resolution of 500 × 500 pixels. There were 133 normal images with no cancerous masses, 437 images with cancerous masses, and 210 images with benign masses among the 780 cancerous images in the BUSI dataset whiles the US ultrasound images includes 123 and 109 ultrasound images of malignant and benign breast tumors. Two traditional ML models, random forest (RF) and K-Nearest Neighbor (KNN), as well as a deep learning (DL) model using convolutional neural networks (CNN), were trained to classify breast masses as benign, malignant, or normal. Results: The CNN obtained an accuracy of 96.10%, the RF an accuracy of 61.46%, and the KNN an accuracy of 64.39% with the BUSI dataset. Standard evaluation measures were employed to assess the performance for benignancy, malignancy, and normality classification. Furthermore, the models’ area under the curve-receiver operating characteristics (AUC-ROC) are 0.99 by the CNN, 0.85 by the RF, and 0.65 by the KNN. Conclusions: The study’s findings revealed that DL surpasses conventional ML when it comes to training image datasets; hence, DL is suggested for breast cancer detection and classification. Furthermore, the resilience of the models used in this study overcomes data imbalance by allowing them to train both binary and multiclass datasets.

Keywords
artificial intelligence
breast cancer
convolutional neural network
deep learning
K-Nearest Neighbor
machine learning
random forest
ultrasound
Figures
Fig. 1.
Share
Back to top