Enhancing Breast Cancer Diagnosis with Ensemble Learning: Leveraging Convolutional Neural Networks and Pretrained Models through Averaged Predictions

Elia Setiana; Reni Nursyanti; Nur Alamsyah; Nayla Nurul Azkiya

doi:10.65780/bima.v1i4.26

Authors

Elia Setiana Universitas Informatika dan Bisnis Indonesia Author
Reni Nursyanti Universitas Informatika dan Bisnis Indonesia Author
Nur Alamsyah Universitas Informatika dan Bisnis Indonesia Author
Nayla Nurul Azkiya Universitas Informatika dan Bisnis Indonesia Author

DOI:

https://doi.org/10.65780/bima.v1i4.26

Keywords:

breast cancer detection, deep learning, ensemble learning, convolutional neural networks, pre-trained models

Abstract

Breast cancer continues to be a serious health issue at the global level, and early detection can significantly improve patient outcomes. This research uses imaging techniques to examine the design of an improved classification model in breast cancer detection. This project uses deep learning approaches through Convolutional Neural Networks (CNN) and ensemble learning models to potentially improve classification accuracy. To further enhance performance while controlling for class imbalance and overfitting, we leverage several models, such as ResNet18 and VGG16, with data augmentation and pre-trained models. Our methods included standard preprocessing of medical images, splitting datasets into training, testing and validation sets, and training each model with the Adam optimizer. Performance measurement included accuracy, precision, recall, and F1 score metrics. Overall, prototypes recently created displayed clear advantages based on finding results achieved through an ensemble method, which demonstrated improved model stability and reduced significant misclassification errors, and model accuracy reached 0.96. This research is crucial while developing strong deep-learning models to aid in breast cancer detection, ultimately allowing us to set a base for developing better diagnostic inference systems in medical-based applications. These systems may help improve early detection and overall patient care.

Downloads

Download data is not yet available.

References

[1] L. Solorzano, S. Robertson, B. Acs, J. Hartman, and M. Rantalainen, ‘Ensemble-based deep learning improves detection of invasive breast cancer in routine histopathology images’, Heliyon, vol. 10, no. 12, p. e32892, Jun. 2024, doi: 10.1016/j.heliyon.2024.e32892. DOI: https://doi.org/10.1016/j.heliyon.2024.e32892

[2] Z. Wang, J. Xie, and J. Zhang, ‘A Robust Enhanced Ensemble Learning Method for Breast Cancer Data Diagnosis on Imbalanced Data’, IEEE Access, vol. 12, pp. 189776–189788, 2024, doi: 10.1109/ACCESS.2024.3516376. DOI: https://doi.org/10.1109/ACCESS.2024.3516376

[3] D. Shah, M. A. U. Khan, M. Abrar, and M. Tahir, ‘Optimizing Breast Cancer Detection With an Ensemble Deep Learning Approach’, International Journal of Intelligent Systems, vol. 2024, no. 1, p. 5564649, 2024, doi: 10.1155/2024/5564649. DOI: https://doi.org/10.1155/2024/5564649

[4] B. Zolfaghari, L. Mirsadeghi, K. Bibak, and K. Kavousi, ‘Cancer Prognosis and Diagnosis Methods Based on Ensemble Learning’, ACM Comput. Surv., vol. 55, no. 12, p. 262:1-262:34, Mar. 2023, doi: 10.1145/3580218. DOI: https://doi.org/10.1145/3580218

[5] S. Majumdar, P. Pramanik, and R. Sarkar, ‘Gamma function based ensemble of CNN models for breast cancer detection in histopathology images’, Expert Systems with Applications, vol. 213, p. 119022, Mar. 2023, doi: 10.1016/j.eswa.2022.119022. DOI: https://doi.org/10.1016/j.eswa.2022.119022

[6] E. K. Jadoon, F. G. Khan, S. Shah, A. Khan, and M. ElAffendi, ‘Deep Learning-Based Multi-Modal Ensemble Classification Approach for Human Breast Cancer Prognosis’, IEEE Access, vol. 11, pp. 85760–85769, 2023, doi: 10.1109/ACCESS.2023.3304242. DOI: https://doi.org/10.1109/ACCESS.2023.3304242

[7] S. Sharmin, T. Ahammad, Md. A. Talukder, and P. Ghose, ‘A Hybrid Dependable Deep Feature Extraction and Ensemble-Based Machine Learning Approach for Breast Cancer Detection’, IEEE Access, vol. 11, pp. 87694–87708, 2023, doi: 10.1109/access.2023.3304628. DOI: https://doi.org/10.1109/ACCESS.2023.3304628

[8] F. Azour and A. Boukerche, ‘An Efficient Transfer and Ensemble Learning Based Computer Aided Breast Abnormality Diagnosis System’, IEEE Access, vol. 11, pp. 21199–21209, 2023, doi: 10.1109/ACCESS.2022.3192857. DOI: https://doi.org/10.1109/ACCESS.2022.3192857

[9] J. Ahmad et al., ‘Deep learning empowered breast cancer diagnosis: Advancements in detection and classification’, PLOS ONE, vol. 19, no. 7, p. e0304757, Jul. 2024, doi: 10.1371/journal.pone.0304757. DOI: https://doi.org/10.1371/journal.pone.0304757

[10] S. Zarif, H. Abdulkader, I. Elaraby, A. Alharbi, W. S. Elkilani, and P. Pławiak, ‘Using hybrid pre-trained models for breast cancer detection’, PLOS ONE, vol. 19, no. 1, p. e0296912, Jan. 2024, doi: 10.1371/journal.pone.0296912. DOI: https://doi.org/10.1371/journal.pone.0296912

[11] U. Naseem et al., ‘An Automatic Detection of Breast Cancer Diagnosis and Prognosis Based on Machine Learning Using Ensemble of Classifiers’, IEEE Access, vol. 10, pp. 78242–78252, 2022, doi: 10.1109/ACCESS.2022.3174599. DOI: https://doi.org/10.1109/ACCESS.2022.3174599

[12] Z. Hameed, S. Zahia, B. Garcia-Zapirain, J. Javier Aguirre, and A. María Vanegas, ‘Breast Cancer Histopathology Image Classification Using an Ensemble of Deep Learning Models’, Sensors, vol. 20, no. 16, p. 4373, Aug. 2020, doi: 10.3390/s20164373. DOI: https://doi.org/10.3390/s20164373

[13] A. A. Balasubramanian et al., ‘Ensemble Deep Learning-Based Image Classification for Breast Cancer Subtype and Invasiveness Diagnosis from Whole Slide Image Histopathology’, Cancers, vol. 16, no. 12, p. 2222, Jan. 2024, doi: 10.3390/cancers16122222. DOI: https://doi.org/10.3390/cancers16122222

[14] M. I. Mahmud, M. Mamun, and A. Abdelgawad, ‘A Deep Analysis of Transfer Learning Based Breast Cancer Detection Using Histopathology Images’, 2023, arXiv. doi: 10.48550/ARXIV.2304.05022. DOI: https://doi.org/10.1109/SPIN57001.2023.10117110

[15] R. M. Munshi, L. Cascone, N. Alturki, O. Saidani, A. Alshardan, and M. Umer, ‘A novel approach for breast cancer detection using optimized ensemble learning framework and XAI’, Image and Vision Computing, vol. 142, p. 104910, Feb. 2024, doi: 10.1016/j.imavis.2024.104910. DOI: https://doi.org/10.1016/j.imavis.2024.104910

[16] A. Alotaibi et al., ‘ViT-DeiT: An Ensemble Model for Breast Cancer Histopathological Images Classification’, 2022, arXiv. doi: 10.48550/ARXIV.2211.00749.