Towards an Accurate Breast Cancer Classification Model based on Ensemble Learning

Breast cancer (BC) is considered the most common cancer among women and the major reason for the increased death rate. This condition begins in breast cells and may spread to the rest of the body tissues. The early detection and prediction of BC can help in saving a patient’s life. In the last decades, machine learning (ML) has played a significant role in the development of models that can be used to detect and predict various diseases at an early stage, which can greatly increase the survival rate of patients. The importance of ML Classification is attributed to its capability to learn from previous datasets, detects patterns that are difficult to comprehend in massive datasets, predicts a categorical variable within a predefined example and provide accurate results within a short amount of time. Feature selection (FS) method was used to reduce the data dimensionality and choose the optimal feature set. In this paper, we proposed a stacking ensemble model that can differentiate between malignant and benign BC cells. A total of 25 different experiments have been conducted using several classifiers, including logistic regression (LR), decision tree (DT), linear discriminant analysis (LDA), K-nearest neighbor (KNN), naive Bayes (NB), and support vector machine (SVM). In addition to several ensembles, the classifiers included random forest (RF), bagging, AdaBoost, voting, and stacking. The results indicate that our ensemble model outperformed other state-of-the-art models in terms of accuracy (98.6%), precision (89.7%), recall, and F1 score (93.33%). The result shows that the ensemble methods with FS have a high improvement of classification accuracy rather than a single method in detecting BC accurately.

@article{Hesham2022,
  title     = {Towards an Accurate Breast Cancer Classification Model based on Ensemble Learning},
  journal   = {International Journal of Advanced Computer Science and Applications},
  volume    = {13},
  number    = {12},
  year      = {2022},
  publisher = {The Science and Information Organization},
  author    = {Aya Hesham and Nora El-Rashidy and Amira Rezk and Noha A. Hikal},
  doi       = {10.14569/IJACSA.2022.0131272},
  url       = {https://doi.org/10.14569/IJACSA.2022.0131272}
}