Abstract:BackgroundBreast cancer is one of the most common diseases in women worldwide. Many studies have been conducted to predict the survival indicators, however most of these analyses were predominantly performed using basic statistical methods. As an alternative, this study used machine learning techniques to build models for detecting and visualising significant prognostic indicators of breast cancer survival rate.MethodsA large hospital-based breast cancer dataset retrieved from the University Malaya Medical Centre, Kuala Lumpur, Malaysia (n = 8066) with diagnosis information between 1993 and 2016 was used in this study. The dataset contained 23 predictor variables and one dependent variable, which referred to the survival status of the patients (alive or dead). In determining the significant prognostic factors of breast cancer survival rate, prediction models were built using decision tree, random forest, neural networks, extreme boost, logistic regression, and support vector machine. Next, the dataset was clustered based on the receptor status of breast cancer patients identified via immunohistochemistry to perform advanced modelling using random forest. Subsequently, the important variables were ranked via variable selection methods in random forest. Finally, decision trees were built and validation was performed using survival analysis.ResultsIn terms of both model accuracy and calibration measure, all algorithms produced close outcomes, with the lowest obtained from decision tree (accuracy = 79.8%) and the highest from random forest (accuracy = 82.7%). The important variables identified in this study were cancer stage classification, tumour size, number of total axillary lymph nodes removed, number of positive lymph nodes, types of primary treatment, and methods of diagnosis.ConclusionInterestingly the various machine learning algorithms used in this study yielded close accuracy hence these methods could be used as alternative predictive tools in the breast cancer survival studies, particularly in the Asian region. The important prognostic factors influencing survival rate of breast cancer identified in this study, which were validated by survival curves, are useful and could be translated into decision support tools in the medical domain.

A hybrid machine learning model for timely prediction of breast cancer

Breast Cancer Prediction Using Deep Learning and Machine Learning Techniques

Early Breast Cancer Prediction using Machine Learning and Deep Learning Techniques

Evaluation of Machine Learning Algorithms for the Prognosis of Breast Cancer from the Surveillance, Epidemiology, and End Results Database

Hybrid ensemble deep learning model for advancing breast cancer detection and classification in clinical applications

Breast Cancer Diagnosis and Prognosis Using Triple Hybrid Deep Learning Approach

Artificial intelligence based machine learning algorithm for prediction of cancer in female anatomy

Breast Cancer Prediction and Classification Using Supervised Learning Techniques

Performance assessment of hybrid machine learning approaches for breast cancer and recurrence prediction

Prediction of Breast Cancer using Machine Learning Approaches

Predicting factors for survival of breast cancer patients using machine learning techniques

Feature Selection-based Machine Learning Comparative Analysis for Predicting Breast Cancer

Breast Cancer Prediction Based on Machine Learning

Machine Learning Techniques for Breast Cancer Prediction

A Hybrid Deep Learning Framework with Decision-Level Fusion for Breast Cancer Survival Prediction

Machine learning models in breast cancer survival prediction

Supervised Machine Learning for Breast Cancer Risk Factors Analysis and Survival Prediction

Breast Cancer Prediction Based on Multiple Machine Learning Algorithms

Machine Learning Approaches to Predict Breast Cancer: Bangladesh Perspective

Machine Learning Based Comparative Analysis for Breast Cancer Prediction

Machine Learning Algorithms For Breast Cancer Prediction And Diagnosis