Semi-Supervised Variational Autoencoder for Survival Prediction
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.
Code (1)
Tasks
Brain Tumor SegmentationPredictionSegmentationSurvival AnalysisSurvival PredictionTumor SegmentationMethods 이 논문이 사용한 방법론
Similar Papers 제목 키워드 기반
Semi-Supervised Variational Inference over Nonlinear Channels
Deep learning methods for communications over unknown nonlinear channels have attracted considerable interest recently. In this paper, we consider semi-supervised learning methods, which are based on variational inferenc…
Meta-LearningVariational InferenceSemi-supervised Variational Autoencoder for Regression: Application on Soft Sensors
We present the development of a semi-supervised regression method using variational autoencoders (VAE), which is customized for use in soft sensing applications. We motivate the use of semi-supervised learning considerin…
regressionUncertainty QuantificationVariational InferenceA Deep Variational Approach to Clustering Survival Data
In this work, we study the problem of clustering survival data $-$ a challenging and so far under-explored task. We introduce a novel semi-supervised probabilistic approach to cluster survival data by leveraging recent a…
ClusteringDeep ClusteringSurvival AnalysisTime-to-Event Prediction+1Semi-Supervised Junction Tree Variational Autoencoder for Molecular Property Prediction
Molecular Representation Learning is essential to solving many drug discovery and computational chemistry problems. It is a challenging problem due to the complex structure of molecules and the vast chemical space. Graph…
Computational chemistryDrug DiscoveryMolecular Property Predictionmolecular representation+2Dr.VAE: Drug Response Variational Autoencoder
We present two deep generative models based on Variational Autoencoders to improve the accuracy of drug response prediction. Our models, Perturbation Variational Autoencoder and its semi-supervised extension, Drug Respon…
Drug Response PredictionGeneral Classification