Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces K-SurvMeans, an innovative clustering method tailored for survival data that optimizes cluster centers based on survival outcomes. By maximizing pairwise survival differences, the method ensures that clusters are well-separated from a survival perspective. The use of the Particle Swarm algorithm for optimization, alongside dimensionality reduction techniques, significantly enhances cluster separation and computational efficiency, as demonstrated through experiments on benchmark survival datasets.
Clustering survival data just got a major upgrade—K-SurvMeans achieves superior separation in survival distributions compared to existing methods.
In this work, we propose K-SurvMeans, a novel extension of K-Means for clustering survival data. The method explicitly uses the survival outcome in the clustering process to optimize cluster centers, thereby maximizing pairwise survival differences between clusters. The objective function encourages the clusters to be well-separated from the survival perspective. Since the resulting optimization problem is non-differentiable, we employ the Particle Swarm algorithm for the Optimization process. To further improve flexibility and mitigate the curse of dimensionality, we extend the framework to operate in a learned low-dimensional latent space obtained via a dimensionality reduction. This allows the method to capture better-separated clusters and enhance optimization efficiency by reducing the search space. Experiments on multiple publicly available benchmark survival datasets demonstrate that K-SurvMeans consistently yields clusters with improved separation in survival distributions compared to existing deep learning-based survival clustering methods.