Enhancing breast cancer prediction through stacking ensemble and deep learning integration

被引：0

作者：

Gurcan, Fatih ^{[1
]}

机构：

[1] Department of Management Information Systems, Faculty of Economics and Administrative Sciences, Karadeniz Technical University, Trabzon

来源：

PeerJ Computer Science | 2025年 / 11卷

关键词：

Breast cancer prediction; Deep learning; Empirical study; Ensemble learning; Stacking ensemble model; Stacking integration;

D O I：

10.7717/PEERJ-CS.2461

中图分类号：

学科分类号：

摘要：

Breast cancer is one of the most common types of cancer in women and is recognized as a serious global public health issue. The increasing incidence of breast cancer emphasizes the importance of early detection, which enhances the effectiveness of treatment processes. In addressing this challenge, the importance of machine learning and deep learning technologies is increasingly recognized. The aim of this study is to evaluate the integration of ensemble models and deep learning models using stacking ensemble techniques on the Breast Cancer Wisconsin (Diagnostic) dataset and to enhance breast cancer diagnosis through this methodology. To achieve this, the efficacy of ensemble methods such as Random Forest, XGBoost, LightGBM, ExtraTrees, Hist Gradient Boosting, AdaBoost, Gradient Boosting, and CatBoost in modeling breast cancer diagnosis was comprehensively evaluated. In addition to ensemble methods, deep learning models including convolutional neural network (CNN), recurrent neural network (RNN), gated recurrent unit (GRU), bidirectional long short-term memory (BILSTM), long short-term memory (LSTM) were analyzed as meta predictors. Among these models, CNN stood out for its high accuracy and rapid training time, making it an ideal choice for real-time diagnostic applications. Finally, the study demonstrated how breast cancer prediction was enhanced by integrating a set of base predictors, such as LightGBM, ExtraTrees, and CatBoost, with a deep learning-based meta-predictor, such as CNN, using stacking ensemble methodology. This stacking integration model offers significant potential for healthcare decision support systems with high accuracy, F1 score, and receiver operating characteristic area under the curve (ROC AUC), along with reduced training times. The results from this research offer important insights for enhancing decision-making strategies in the diagnosis and management of breast cancer. Copyright 2025 Gurcan Distributed under Creative Commons CC-BY 4.0

引用

共 38 条

[1]

Abbas S, Jalil Z, Javed AR, Batool I, Khan MZ, Noorwali A, Gadekallu TR, Akbar A., BCD-WERT: a novel approach for breast cancer detection using whale optimization based efficient features and extremely randomized tree algorithm, PeerJ Computer Science, 7, (2021)

[2]

Abreu PH, Santos MS, Abreu MH, Andrade B, Silva DC., Predicting breast cancer recurrence using machine learning techniques: a systematic review, ACM Computing Surveys, 49, (2016)

[3]

Ak MF., A comparative analysis of breast cancer detection and diagnosis using data visualization and machine learning applications, Healthcare, 8, (2020)

[4]

Al-Azzam N, Shatnawi I., Comparing supervised and semi-supervised Machine Learning Models on Diagnosing Breast Cancer, Annals of Medicine and Surgery, 62, pp. 53-64, (2021)

[5]

Aurelien G., Hands-on machine learning with Scikit-Learn, Keras and TensorFlow: concepts, tools, and techniques to build intelligent systems, (2019)

[6]

Boeri C, Chiappa C, Galli F, De Berardinis V, Bardelli L, Carcano G, Rovera F., Machine learning techniques in breast cancer prognosis prediction: a primary evaluation, Cancer Medicine, 9, pp. 3234-3243, (2020)

[7]

Chugh G, Kumar S, Singh N., Survey on machine learning and deep learning applications in breast cancer diagnosis, Cognitive Computation, 13, pp. 1451-1470, (2021)

[8]

Din NM Ud, Dar RA, Rasool M, Assad A., Breast cancer detection using deep learning: datasets, methods, and challenges ahead, Computers in Biology and Medicine, 149, (2022)

[9]

Ebrahim M, Sedky AAH, Mesbah S., Accuracy assessment of machine learning algorithms used to predict breast cancer, Data, 8, (2023)

[10]

Fatima N, Liu L, Hong S, Ahmed H., Prediction of breast cancer, comparative review of machine learning techniques, and their analysis, IEEE Access, 8, pp. 150360-150376, (2020)

← 1 2 3 4 →