Interpretable Survival Analysis for Heart Failure Risk Prediction
- URL: http://arxiv.org/abs/2310.15472v1
- Date: Tue, 24 Oct 2023 02:56:05 GMT
- Title: Interpretable Survival Analysis for Heart Failure Risk Prediction
- Authors: Mike Van Ness, Tomas Bosschieter, Natasha Din, Andrew Ambrosy,
Alexander Sandhu, Madeleine Udell
- Abstract summary: We propose a novel survival analysis pipeline that is both interpretable and competitive with state-of-the-art survival models.
Our pipeline achieves state-of-the-art performance and provides interesting and novel insights about risk factors for heart failure.
- Score: 50.64739292687567
- License: http://creativecommons.org/licenses/by/4.0/
- Abstract: Survival analysis, or time-to-event analysis, is an important and widespread
problem in healthcare research. Medical research has traditionally relied on
Cox models for survival analysis, due to their simplicity and interpretability.
Cox models assume a log-linear hazard function as well as proportional hazards
over time, and can perform poorly when these assumptions fail. Newer survival
models based on machine learning avoid these assumptions and offer improved
accuracy, yet sometimes at the expense of model interpretability, which is
vital for clinical use. We propose a novel survival analysis pipeline that is
both interpretable and competitive with state-of-the-art survival models.
Specifically, we use an improved version of survival stacking to transform a
survival analysis problem to a classification problem, ControlBurn to perform
feature selection, and Explainable Boosting Machines to generate interpretable
predictions. To evaluate our pipeline, we predict risk of heart failure using a
large-scale EHR database. Our pipeline achieves state-of-the-art performance
and provides interesting and novel insights about risk factors for heart
failure.
Related papers
- HACSurv: A Hierarchical Copula-based Approach for Survival Analysis with Dependent Competing Risks [51.95824566163554]
HACSurv is a survival analysis method that learns structures and cause-specific survival functions from data with competing risks.
By capturing the dependencies between risks and censoring, HACSurv achieves better survival predictions.
arXiv Detail & Related papers (2024-10-19T18:52:18Z) - SurvCORN: Survival Analysis with Conditional Ordinal Ranking Neural Network [4.772480981435387]
We propose SurvCORN, a novel method utilizing conditional ordinal ranking networks to predict survival curves directly.
We also introduce SurvMAE, a metric designed to evaluate the accuracy of model predictions in estimating time-to-event outcomes.
arXiv Detail & Related papers (2024-09-30T03:01:25Z) - Interpretable Prediction and Feature Selection for Survival Analysis [18.987678432106563]
We present DyS (pronounced dice''), a new survival analysis model that achieves both strong discrimination and interpretability.
DyS is a feature-sparse Generalized Additive Model, combining feature selection and interpretable prediction into one model.
arXiv Detail & Related papers (2024-04-23T02:36:54Z) - MedDiffusion: Boosting Health Risk Prediction via Diffusion-based Data
Augmentation [58.93221876843639]
This paper introduces a novel, end-to-end diffusion-based risk prediction model, named MedDiffusion.
It enhances risk prediction performance by creating synthetic patient data during training to enlarge sample space.
It discerns hidden relationships between patient visits using a step-wise attention mechanism, enabling the model to automatically retain the most vital information for generating high-quality data.
arXiv Detail & Related papers (2023-10-04T01:36:30Z) - Contrastive Learning of Temporal Distinctiveness for Survival Analysis
in Electronic Health Records [10.192973297290136]
We propose a novel Ontology-aware Temporality-based Contrastive Survival (OTCSurv) analysis framework.
OTCSurv uses survival durations from both censored and observed data to define temporal distinctiveness.
We conduct experiments using a large EHR dataset to forecast the risk of hospitalized patients who are in danger of developing acute kidney injury (AKI)
arXiv Detail & Related papers (2023-08-24T22:36:22Z) - SurvivalGAN: Generating Time-to-Event Data for Survival Analysis [121.84429525403694]
Imbalances in censoring and time horizons cause generative models to experience three new failure modes specific to survival analysis.
We propose SurvivalGAN, a generative model that handles survival data by addressing the imbalance in the censoring and event horizons.
We evaluate this method via extensive experiments on medical datasets.
arXiv Detail & Related papers (2023-02-24T17:03:51Z) - Reverse Survival Model (RSM): A Pipeline for Explaining Predictions of
Deep Survival Models [0.0]
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.
arXiv Detail & Related papers (2022-10-27T03:39:01Z) - Dynamic prediction of time to event with survival curves [3.9169188005935927]
We apply our recently developed counterfactual dynamic survival model (CDSM) to static and longitudinal observational data.
We prove that the inflection point of its estimated individual survival curves provides reliable prediction of the patient failure time.
arXiv Detail & Related papers (2021-01-26T12:17:27Z) - A General Framework for Survival Analysis and Multi-State Modelling [70.31153478610229]
We use neural ordinary differential equations as a flexible and general method for estimating multi-state survival models.
We show that our model exhibits state-of-the-art performance on popular survival data sets and demonstrate its efficacy in a multi-state setting.
arXiv Detail & Related papers (2020-06-08T19:24:54Z) - Survival Cluster Analysis [93.50540270973927]
There is an unmet need in survival analysis for identifying subpopulations with distinct risk profiles.
An approach that addresses this need is likely to improve characterization of individual outcomes.
arXiv Detail & Related papers (2020-02-29T22:41:21Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.