PeerJ Computer Science (May 2024)

Enhancing early breast cancer diagnosis through automated microcalcification detection using an optimized ensemble deep learning framework

  • Jing Ru Teoh,
  • Khairunnisa Hasikin,
  • Khin Wee Lai,
  • Xiang Wu,
  • Chong Li

DOI
https://doi.org/10.7717/peerj-cs.2082
Journal volume & issue
Vol. 10
p. e2082

Abstract

Read online Read online

Background Breast cancer remains a pressing global health concern, necessitating accurate diagnostics for effective interventions. Deep learning models (AlexNet, ResNet-50, VGG16, GoogLeNet) show remarkable microcalcification identification (>90%). However, distinct architectures and methodologies pose challenges. We propose an ensemble model, merging unique perspectives, enhancing precision, and understanding critical factors for breast cancer intervention. Evaluation favors GoogleNet and ResNet-50, driving their selection for combined functionalities, ensuring improved precision, and dependability in microcalcification detection in clinical settings. Methods This study presents a comprehensive mammogram preprocessing framework using an optimized deep learning ensemble approach. The proposed framework begins with artifact removal using Otsu Segmentation and morphological operation. Subsequent steps include image resizing, adaptive median filtering, and deep convolutional neural network (D-CNN) development via transfer learning with ResNet-50 model. Hyperparameters are optimized, and ensemble optimization (AlexNet, GoogLeNet, VGG16, ResNet-50) are constructed to identify the localized area of microcalcification. Rigorous evaluation protocol validates the efficacy of individual models, culminating in the ensemble model demonstrating superior predictive accuracy. Results Based on our analysis, the proposed ensemble model exhibited exceptional performance in the classification of microcalcifications. This was evidenced by the model’s average confidence score, which indicated a high degree of dependability and certainty in differentiating these critical characteristics. The proposed model demonstrated a noteworthy average confidence level of 0.9305 in the classification of microcalcification, outperforming alternative models and providing substantial insights into the dependability of the model. The average confidence of the ensemble model in classifying normal cases was 0.8859, which strengthened the model’s consistent and dependable predictions. In addition, the ensemble models attained remarkably high performances in terms of accuracy, precision, recall, F1-score, and area under the curve (AUC). Conclusion The proposed model’s thorough dataset integration and focus on average confidence ratings within classes improve clinical diagnosis accuracy and effectiveness for breast cancer. This study introduces a novel methodology that takes advantage of an ensemble model and rigorous evaluation standards to substantially improve the accuracy and dependability of breast cancer diagnostics, specifically in the detection of microcalcifications.

Keywords