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

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.

Proteomic analysis of circulating extracellular vesicles identifies potential markers of breast cancer progression, recurrence, and response

“…To determine whether we can increase the test set prediction accuracy, we tested several other classification models ( 20 ) and compared the results to the classification by our seven-protein signature. Of the 276 proteins examined in the main cohort RPPA, 267 were also examined in the test set RPPA.…”

A Novel Comprehensive Clinical Stratification Model to Refine Prognosis of Glioblastoma Patients Undergoing Surgical Resection

“…In a clinical setting, the need for classification tools based on the prognostic stratification of GBM cases undergoing surgical protocols is of increasing importance. Numerous attempts have been developed to classify GBM patients, which include combination models of clinical, molecular and radiomic variables used in daily clinical practice [34][35][36][37][38].…”

Improving the Performance of Lung Cancer Prediction Using Machine Learning Techniques on Big Data

“…Ganggayah et al [5] used various classifiers on breast cancer data having 8066 record with 23 predictor and concluded that random forest classifier gives 82% better accuracy.…”