arXiv:2607.24405cs.LG2026-07

K-SurvMeans通过生存结局优化聚类中心,提升组间生存差异。

K-Survival Means

论文配图:K-Survival Means
图 1 · 摘自论文原文
  • 基于生存结果直接优化聚类中心,强化组间生存差异
  • 在多个公开数据集上优于现有深度学习生存聚类方法
  • 结合降维与粒子群算法,提升聚类分离度与效率

本文提出K-SurvMeans,一种用于生存数据聚类的新方法。该方法在聚类过程中显式利用生存结果优化聚类中心,以最大化组间生存差异。目标函数从生存角度促使聚类充分分离。由于优化问题不可微,采用粒子群算法求解。为进一步提升灵活性并缓解高维困境,方法扩展至通过降维获得的低维隐空间中运行,从而捕捉更清晰分离的聚类,并降低搜索空间提升效率。在多个公开基准生存数据集上的实验表明,相比现有基于深度学习的生存聚类方法,K-SurvMeans始终能产生更具生存差异的聚类结果。

原文摘要 · Abstract (English)

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.

生存分析聚类降维粒子群

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。