Survival regression aims to predict the time when an event of interest will
take place, typically a death or a failure. A fully parametric method [18] is
proposed to estimate the survival function as a mixture of individual
parametric distributions in the presence of censoring. In this paper, We
present a novel method to predict the survival time by better clustering the
survival data and combine primitive distributions. We propose two variants of
variational auto-encoder (VAE), discrete and continuous, to generate the latent
variables for clustering input covariates. The model is trained end to end by
jointly optimizing the VAE loss and regression loss. Thorough experiments on
dataset SUPPORT and FLCHAIN show that our method can effectively improve the
clustering result and reach competitive scores with previous methods. We
demonstrate the superior result of our model prediction in the long-term. Our
code is available at this https URL

通过更好地聚类生存数据和组合原始分布，本文提出了两种变体的变分自编码器（VAE），离散和连续，来生成聚类输入协变量的潜在变量。我们的模型通过同时优化 VAE 损失和回归损失进行端到端训练，实验证明了我们的方法在聚类结果和长期预测上具有竞争力，特别是在期限较长的情况下。