A General Machine Learning Framework for Survival Analysis
The modeling of time-to-event data, also known as survival analysis, requires specialized methods that can deal with censoring and truncation, time-varying features and effects, and that extend to settings with multiple competing events. However, many machine learning methods for survival analysis only consider the standard setting with right-censored data and proportional hazards assumption. The methods that do provide extensions usually address at most a subset of these challenges and often require specialized software that can not be integrated into standard machine learning workflows directly. In this work, we present a very general machine learning framework for time-to-event analysis that uses a data augmentation strategy to reduce complex survival tasks to standard Poisson regression tasks. This reformulation is based on well developed statistical theory. With the proposed approach, any algorithm that can optimize a Poisson (log-)likelihood, such as gradient boosted trees, deep neural networks, model-based boosting and many more can be used in the context of time-to-event analysis. The proposed technique does not require any assumptions with respect to the distribution of event times or the functional shapes of feature and interaction effects. Based on the proposed framework we develop new methods that are competitive with specialized state of the art approaches in terms of accuracy, and versatility, but with comparatively small investments of programming effort or requirements for specialized methodological know-how.
Code (1)
Tasks
BIG-bench Machine LearningData AugmentationSurvival AnalysisSimilar Papers 제목 키워드 기반
mlr3proba: An R Package for Machine Learning in Survival Analysis
As machine learning has become increasingly popular over the last few decades, so too has the number of machine learning interfaces for implementing these models. Whilst many R libraries exist for machine learning, very …
BenchmarkingBIG-bench Machine LearningSurvival AnalysisEnsemble representation learning: an analysis of fitness and survival for wrapper-based genetic programming methods
Recently we proposed a general, ensemble-based feature engineering wrapper (FEW) that was paired with a number of machine learning methods to solve regression problems. Here, we adapt FEW for supervised classification an…
Feature EngineeringGeneral ClassificationRepresentation LearningReduction Techniques for Survival Analysis
In this work, we discuss what we refer to as reduction techniques for survival analysis, that is, techniques that "reduce" a survival task to a more common regression or classification task, without ignoring the specific…
A Deep Latent-Variable Model Application to Select Treatment Intensity in Survival Analysis
In the following short article we adapt a new and popular machine learning model for inference on medical data sets. Our method is based on the Variational AutoEncoder (VAE) framework that we adapt to survival analysis o…
BIG-bench Machine LearningDecision MakingMissing ValuesSmall Data Image Classification+1Interpretable Prediction and Feature Selection for Survival Analysis
Survival analysis is widely used as a technique to model time-to-event data when some data is censored, particularly in healthcare for predicting future patient risk. In such settings, survival models must be both accura…
feature selectionPredictionSurvival Analysis