Semi-Supervised Variational Autoencoder for Survival Prediction
arXiv:1910.04488 · doi:10.1007/978-3-030-46643-5_12
Abstract
In this paper we propose a semi-supervised variational autoencoder for classification of overall survival groups from tumor segmentation masks. The model can use the output of any tumor segmentation algorithm, removing all assumptions on the scanning platform and the specific type of pulse sequences used, thereby increasing its generalization properties. Due to its semi-supervised nature, the method can learn to classify survival time by using a relatively small number of labeled subjects. We validate our model on the publicly available dataset from the Multimodal Brain Tumor Segmentation Challenge (BraTS) 2019.
arXiv commentPublished in the pre-conference proceeding of "2019 International MICCAI BraTS Challenge"
References in corpus (7)
- Adam: A Method for Stochastic Optimization
- Auto-Encoding Variational Bayes
- Categorical Reparameterization with Gumbel-Softmax
- Semi-Supervised Learning with Deep Generative Models
- The Concrete Distribution: A Continuous Relaxation of Discrete Random Variables
- Automatic Brain Tumor Segmentation using Cascaded Anisotropic Convolutional Neural Networks
- Self-Normalizing Neural Networks