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.
@article{arxiv.1811.12323,
title = {A Deep Latent-Variable Model Application to Select Treatment Intensity in Survival Analysis},
author = {Cédric Beaulac and Jeffrey S. Rosenthal and David Hodgson},
journal= {arXiv preprint arXiv:1811.12323},
year = {2018}
}
Comments
Machine Learning for Health (ML4H) Workshop at NeurIPS 2018 arXiv:1811.07216