A Comparative Study for Survival Prediction of NKI Data Using Statistical and Machine-Learning Approaches

Authors

• Ahmed M. Gad Prof, The British University in EgyptFollow
• Niveen ElZayat Dr, Cairo UniversityFollow
• Amal Sakr Dr, Cairo UniversityFollow
• Mahmoud Rashwan Prof, Cairo UniversityFollow
• Mariam Hegazy Ms, Cairo UniversityFollow

Document Type

Article

Publication Date

2024

Abstract

The main aim of the study is to compare both statistical and machine learning algorithms in survival analysis prediction, taking into account the dimensionality of the dataset. Performance criteria like the concordance-censored index have been employed in the context of 5-fold and 10-fold cross-validation. The dataset utilized was the Netherlands Cancer Institute (NKI). Two machine-learning algorithms (eXtreme Gradient Boosting and Fast Survival Support Vector Machine) and one statistical technique (Cox proportional hazard model) were employed. The results showed through the application that, as features utilized increase, XGBoost surpasses other algorithms. However, as the number of selected features decreases drastically, the statistical analysis and the Fast Survival Support Vector Machine are both better at survival prediction.

Recommended Citation

N. I. Elzayat, A. M. Gad, M. Hegazy, A. Sakr and M. Rashwan, "A Comparative Study for Survival Prediction of NKI Data Using Statistical and Machine-Learning Approaches," 2024 6th Novel Intelligent and Leading Emerging Sciences Conference (NILES), Giza, Egypt, 2024, pp. 269-273, doi: 10.1109/NILES63360.2024.10753226.