Case-Base Neural Networks: survival analysis with time-varying, higher-order interactions

01/16/2023
by   Jesse Islam, et al.
0

Neural network-based survival methods can model data-driven covariate interactions. While these methods have led to better predictive performance than regression-based approaches, they cannot model both time-varying interactions and complex baseline hazards. To address this, we propose Case-Base Neural Networks (CBNN) as a new approach that combines the case-base sampling framework with flexible architectures. Our method naturally accounts for censoring and does not require method specific hyperparameters. Using a novel sampling scheme and data augmentation, we incorporate time directly into a feed-forward neural network. CBNN predicts the probability of an event occurring at a given moment and estimates the hazard function. We compare the performance of CBNN to survival methods based on regression and neural networks in two simulations and two real data applications. We report two time-dependent metrics for each model. In the simulations and real data applications, CBNN provides a more consistent predictive performance across time and outperforms the competing neural network approaches. For a simple simulation with an exponential hazard model, CBNN outperforms the other neural network methods. For a complex simulation, which highlights the ability of CBNN to model both a complex baseline hazard and time-varying interactions, CBNN outperforms all competitors. The first real data application shows CBNN outperforming all neural network competitors, while a second real data application shows competitive performance. We highlight the benefit of combining case-base sampling with deep learning to provide a simple and flexible modeling framework for data-driven, time-varying interaction modeling of survival outcomes. An R package is available at https://github.com/Jesse-Islam/cbnn.

READ FULL TEXT
research
09/22/2020

casebase: An Alternative Framework For Survival Analysis and Comparison of Event Rates

In epidemiological studies of time-to-event data, a quantity of interest...
research
11/11/2020

Semi-Structured Deep Piecewise Exponential Models

We propose a versatile framework for survival analysis that combines adv...
research
05/19/2023

The Deep Promotion Time Cure Model

We propose a novel method for predicting time-to-event in the presence o...
research
07/16/2023

Towards Flexible Time-to-event Modeling: Optimizing Neural Networks via Rank Regression

Time-to-event analysis, also known as survival analysis, aims to predict...
research
07/26/2020

DeepHazard: neural network for time-varying risks

Prognostic models in survival analysis are aimed at understanding the re...
research
08/06/2020

Zeroing Neural Networks, an Introduction to, a Survey of, and Predictive Computations for Time-varying Matrix Problems

This paper is designed to increase knowledge and understanding of time-v...
research
06/27/2020

A General Machine Learning Framework for Survival Analysis

The modeling of time-to-event data, also known as survival analysis, req...

Please sign up or login with your details

Forgot password? Click here to reset