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.

Titanic Survival Prediction Based on Machine Learning Algorithms

Survival Prediction and Comparison of the Titanic based on Machine Learning Classifiers

Survival prediction and analysis of Titanic based on logistic regression and KNN

Analysis and Prediction of the Survival of Titanic Passengers Using Machine Learning

Titanic survival prediction based on machine learning.

Comparative analysis for survival prediction from titanic disaster using machine learning

Classification of Titanic Passenger Data and Chances of Surviving the Disaster

Ensemble Approach for Titanic Survival Predictor

Application of Logistic Regression on Passenger Survival Data of the Titanic Liner

Deep Semisupervised Multitask Learning Model and Its Interpretability for Survival Analysis.

Study on the Prediction Model of Accidents and Incidents of Cruise Ship Operation Based on Machine Learning

Comprehensive evaluation of machine learning models for predicting ship energy consumption based on onboard sensor data

Predicting Heart Failure Survival with Machine Learning: Assessing My Risk

Prediction of cardiovascular and cerebrovascular diseases based on machine learning models

Diversity, Quality, and Quantity of Real Ship Data on the Black-Box and Gray-Box Prediction Models of Ship Fuel Consumption

Prediction of Postoperative Survival Time of Hepatocellular Carcinoma Patients Based on Machine Learning

Prediction of US airline passenger satisfaction using machine learning algorithms

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

Ten-year Survival Prediction for Breast Cancer Patients

Machine Learning for Survival Analysis: A Survey

Effective Heart Disease Prediction Using Machine Learning Techniques