1. Reverse survival model (RSM): a pipeline for explaining predictions of deep survival models.
- Author
-
Rezaei, Mohammad R., Fard, Reza Saadati, Pourjafari, Ebrahim, Ziaei, Navid, Sameizadeh, Amir, Shafiee, Mohammad, Alavinia, Mohammad, Abolghasemian, Mansour, and Sajadi, Nick
- Subjects
SURVIVAL analysis (Biometry) ,ARTIFICIAL neural networks ,SURVIVAL rate ,PROBABILITY density function ,INTENSIVE care units ,VENTILATION - Abstract
The aim of survival analysis in healthcare is to estimate the probability of occurrence of an event, such as a patient's death in an intensive care unit (ICU). Recent developments in deep neural networks (DNNs) for survival analysis show the superiority of these models in comparison with other well-known models in survival analysis applications. Ensuring the reliability and explainability of deep survival models deployed in healthcare is a necessity. Since DNN models often behave like a black box, their predictions might not be easily trusted by clinicians, especially when predictions are contrary to a physician's opinion. A deep survival model that explains and justifies its decision-making process could potentially gain the trust of clinicians. In this research, we propose the reverse survival model (RSM) framework that provides detailed insights into the decision-making process of survival models. For each patient of interest, RSM can extract similar patients from a dataset and rank them based on the most relevant features that deep survival models rely on for their predictions. RSM acts as an add-on to a deep survival model and offers three functionalities: 1) Finding the most relevant clinical measurements for the probability density functions (PDFs) of events. 2) Categorizing patients into disjoint clusters based on the similarity of their survival PDFs. 3) Ranking similar patients based on the similarity of survival outcomes and relevant clinical measurements. The explainability of deep survival models is rarely addressed in the literature. Therefore, the RSM pipeline is a unique approach to explaining the predictions of deep survival models. We validated the RSM pipeline by testing it on a synthetic dataset and MIMIC-IV, a dataset of intensive care unit (ICU) clinical observations. Our experiments showed that given a deep survival model and a patient of interest, RSM can successfully detect similar patient records from historical data and rank them based on the similarities between their survival PDFs and the most relevant patient observations. [ABSTRACT FROM AUTHOR]
- Published
- 2023
- Full Text
- View/download PDF