Predicting Survivability of Cancer Patients with Metastatic Patterns Using Explainable AI
Nalela, P.; Rao, D.; Rao, P.
Show abstract
Cancer remains a leading global health challenge and a major cause of mortality. This study leverages machine learning (ML) to predict the survivability of cancer patients with metastatic patterns using the comprehensive MSK-MET dataset, which includes genomic and clinical data from 25,775 patients across 27 cancer types. We evaluated five ML models-XGBoost, Nave Bayes, Decision Tree, Logistic Regression, and Random Fores using hyperparameter tuning and grid search. XGBoost emerged as the best performer with an area under the curve (AUC) of 0.82. To enhance model interpretability, SHapley Additive exPlanations (SHAP) were applied, revealing key predictors such as metastatic site count, tumor mutation burden, fraction of genome altered, and organ-specific metastases. Further survival analysis using Kaplan-Meier curves, Cox Proportional Hazards models, and XGBoost Survival Analysis identified significant predictors of patient outcomes, offering actionable insights for clinicians. These findings could aid in personalized prognosis and treatment planning, ultimately improving patient care.
Matching journals
The top 6 journals account for 50% of the predicted probability mass.
Similar papers in this journal
- On the predictability of postoperative complications for cancer patients: a Portuguese cohort study 96%
- Combining symbolic regression with the Cox proportional hazards model improves prediction of heart failure deaths 94%
- OASIS+: leveraging machine learning to improve the prognostic accuracy of OASIS severity score for predicting in-hospital mortality 94%
Similar papers in this journal
- Survival Prediction Landscape: An In-Depth Systematic Literature Review on Activities, Methods, Tools, Diseases, and Databases 95%
- Predicting the disease outcome in COVID-19 positive patients through Machine Learning: a retrospective cohort study with Brazilian data 93%
- Predicting Ward Transfer Mortality with Machine Learning 92%
Similar papers in this journal
- Novel ratio-metric features enable the identification of new driver genes across cancer types 94%
- Classification models for Invasive Ductal Carcinoma Progression, based on gene expression data-trained supervised machine learning 93%
- MultiSurv: Long-term cancer survival prediction using multimodal deep learning 93%
Similar papers in this journal
- Development of an absolute assignment predictor for triple-negative breast cancer subtyping using machine learning approaches 95%
- Deep Multimodal Graph-Based Network for Survival Prediction from Highly Multiplexed Images and Patient Variables 93%
- AI-MET: A Deep Learning-based Clinical Decision Support System for Distinguishing Multisystem Inflammatory Syndrome in Children from Endemic Typhus 92%
Similar papers in this journal
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.