Saved in:
Bibliographic Details
Main Authors: Buginga, Gabriel, Silva, Edmundo de Souza e
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2405.15934
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909210348879872
author Buginga, Gabriel
Silva, Edmundo de Souza e
author_facet Buginga, Gabriel
Silva, Edmundo de Souza e
contents Survival analysis aims to predict the timing of future events across various fields, from medical outcomes to customer churn. However, the integration of clustering into survival analysis, particularly for precision medicine, remains underexplored. This study introduces SurvMixClust, a novel algorithm for survival analysis that integrates clustering with survival function prediction within a unified framework. SurvMixClust learns latent representations for clustering while also predicting individual survival functions using a mixture of non-parametric experts. Our evaluations on five public datasets show that SurvMixClust creates balanced clusters with distinct survival curves, outperforms clustering baselines, and competes with non-clustering survival models in predictive accuracy, as measured by the time-dependent c-index and log-rank metrics.
format Preprint
id arxiv_https___arxiv_org_abs_2405_15934
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Clustering Survival Data using a Mixture of Non-parametric Experts
Buginga, Gabriel
Silva, Edmundo de Souza e
Machine Learning
Survival analysis aims to predict the timing of future events across various fields, from medical outcomes to customer churn. However, the integration of clustering into survival analysis, particularly for precision medicine, remains underexplored. This study introduces SurvMixClust, a novel algorithm for survival analysis that integrates clustering with survival function prediction within a unified framework. SurvMixClust learns latent representations for clustering while also predicting individual survival functions using a mixture of non-parametric experts. Our evaluations on five public datasets show that SurvMixClust creates balanced clusters with distinct survival curves, outperforms clustering baselines, and competes with non-clustering survival models in predictive accuracy, as measured by the time-dependent c-index and log-rank metrics.
title Clustering Survival Data using a Mixture of Non-parametric Experts
topic Machine Learning
url https://arxiv.org/abs/2405.15934