Current browse context:
cs.LG
Change to browse by:
References & Citations
Computer Science > Machine Learning
Title: A Deep Variational Approach to Clustering Survival Data
(Submitted on 10 Jun 2021 (v1), last revised 10 Mar 2022 (this version, v3))
Abstract: 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 advances in stochastic gradient variational inference. In contrast to previous work, our proposed method employs a deep generative model to uncover the underlying distribution of both the explanatory variables and censored survival times. We compare our model to the related work on clustering and mixture models for survival data in comprehensive experiments on a wide range of synthetic, semi-synthetic, and real-world datasets, including medical imaging data. Our method performs better at identifying clusters and is competitive at predicting survival times. Relying on novel generative assumptions, the proposed model offers a holistic perspective on clustering survival data and holds a promise of discovering subpopulations whose survival is regulated by different generative mechanisms.
Submission history
From: Ricards Marcinkevics [view email][v1] Thu, 10 Jun 2021 14:10:25 GMT (9214kb,D)
[v2] Tue, 5 Oct 2021 14:03:33 GMT (41399kb,D)
[v3] Thu, 10 Mar 2022 14:35:53 GMT (46159kb,D)
Link back to: arXiv, form interface, contact.