Predicting Colorectal Cancer Recurrence and Patient Survival Using Supervised Machine Learning Approach

Background: South Africa (SA) has the highest incidence of colorectal cancer (CRC) in Sub-Saharan Africa (SSA). However, there is limited research on CRC recurrence and survival in SA. CRC recurrence and overall survival are highly variable across studies. Accurate prediction of patients at risk can enhance clinical expectations and decisions within the South African CRC patients population. We explored the feasibility of integrating statistical and machine learning (ML) algorithms to achieve higher predictive performance and interpretability in findings. Methods: We selected and compared six algorithms:- logistic regression (LR), naïve Bayes (NB), C5.0, random forest (RF), support vector machine (SVM) and artificial neural network (ANN). Commonly selected features based on OneR and information gain, within 10-fold cross-validation, were used for model development. The validity and stability of the predictive models were further assessed using simulated datasets. Results: The six algorithms achieved high discriminative accuracies (AUC-ROC). ANN achieved the highest AUC-ROC for recurrence (87.0%) and survival (82.0%), and other models showed comparable performance with ANN. We observed no statistical difference in the performance of the models. Features including radiological stage and patient's age, histology, and race are risk factors of CRC recurrence and patient survival, respectively. Conclusions: Based on other studies and what is known in the field, we have affirmed important predictive factors for recurrence and survival using rigorous procedures. Outcomes of this study can be generalised to CRC patient population elsewhere in SA and other SSA countries with similar patient profiles

Cancer · Colorectal cancer · Discriminative model · Interpretability · Logistic regression · Machine learning · Naive Bayes classifier · Population · Random forest · Receiver operating characteristic · Support vector machine · Colorectal Cancer Screening and Detection · Computer Science · Global Cancer Incidence and Screening · Medicine · Radiomics and Machine Learning in Medical Imaging · Artificial Intelligence · Internal Medicine

• Artificial intelligence-based prediction for cancer-related outcomes in Africa

  Open Access•John Adeoye, Abdul-Warith O Akinshipo et al.•Journal of Global Health•2022

  Incluir item

• Regression Shrinkage and Selection via The Lasso

  Open Access•Robert Tibshirani•Journal of the Royal Statistical…•2011

  Incluir item

• Assessing the Performance of Prediction Models

  Ewout W Steyerberg, Andrew J Vickers et al.•Epidemiology•2010

  Incluir item

• MissForest—non-parametric missing value imputation for mixed-type data

  Open Access•Daniel J Stekhoven, Peter Bühlmann•Bioinformatics•2012

  Incluir item

• Random Forests

  Open Access•Leo Breiman•Machine Learning•2001

  Incluir item

• What You See May Not Be What You Get

  Michael A Babyak•Psychosomatic Medicine•2004

  Incluir item

• What You See May Not Be What You Get

  Open Access•Michael A Babyak•Psychosomatic Medicine•2004

  Incluir item