Original scientific paper
https://doi.org/10.1080/00051144.2023.2226946
GBDTMO: as new option for early-stage breast cancer detection and classification using machine learning
A. S. Vibith
; Department of Computer Science and Engineering, Sri Venkateswara College of Engineering, Sriperumbudur, Tamilnadu, India
M. C. Jobin Christ
; Department of Biomedical Engineering, Rajalakshmi Engineering College, Thandalam, Tamilnadu, India
Abstract
Breast cancer is the second leading cause of disease death in women, after lung and bronchus cancer. According to measurements, mammography misses breast cancer in 10% to 15% of cases for women aged 50 to 69 years. In the current study, we used the Wisconsin breast cancer dataset to develop a two-stage model for breast cancer diagnosis. The main goal of this study effort is to effectively carry out feature selection and classification tasks. Gradient Boosting Decision Tree-based Mayfly Optimisation (GBDTMO), an innovative and efficient breast cancer diagnostic machine learning system, is provided. In the second stage, we employ a Mayfly search to determine which subset of traits is the best. Two more well-known datasets on breast cancer, the ICCR and the Cancer Corpus, were also compared for classification accuracy. The accuracy of the suggested GBDTMO model was higher than that of the existing GBDT and Practical Federated Gradient Boosting Decision Tree (PFGBDT), which had accuracy values of 93.25% and 94.25%, respectively. Similarly, the recall, F-measure, and ROC area values were 98.52%, 97.52%, and 96.32%, respectively. Furthermore, it demonstrated a lower RMSE of 0.98 than the existing GBDT and PFGBDT.
Keywords
Breast cancer; machine learning; accuracy; classification; ensemble; Mayfly search
Hrčak ID:
315943
URI
Publication date:
27.6.2023.
Visits: 35 *