Hernandez C, Pachón-García C, Delicado P, Vilaplana V. Interpreting Machine Learning Models for Survival Analysis: A Study of Cutaneous Melanoma Using the SEER Database. In Explainable Artificial Intelligence and Process Mining Applications for Healthcare,. Springer, Cham; 2024.

Abstract

In this study, we train and compare three types of machine learning algorithms for Survival Analysis: Random Survival Forest, DeepSurv and DeepHit, using the SEER database to model cutaneous malignant melanoma. Additionally, we employ SurvLIMEpy library, a Python package designed to provide explainability for survival machine learning models, to analyse feature importance. The results demonstrate that machine learning algorithms outperform the Cox Proportional Hazards Model. Our work underscores the importance of explainability methods for interpreting black-box models and provides insights into important features related to melanoma prognosis.