A Deep Latent-Variable Model Application to Select Treatment Intensity in Survival Analysis
This work addresses treatment decision-making in medical contexts, but it is incremental as it applies an existing VAE framework to a specific domain.
The authors tackled the problem of selecting treatment intensity in survival analysis by adapting a Variational AutoEncoder (VAE) framework to handle small medical datasets with missing values, resulting in a model that outperforms a classic survival analysis method.
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 on small data sets with missing values. In our model, the true health status appears as a set of latent variables that affects the observed covariates and the survival chances. We show that this flexible model allows insightful decision-making using a predicted distribution and outperforms a classic survival analysis model.