SurvivalLVQ: Interpretable supervised clustering and prediction in survival analysis via Learning Vector Quantization

• Published in: Pattern recognition 2024-09, Vol.153, p.110497, Article 110497
• Main Authors: de Boer, Jasper, Dedja, Klest, Vens, Celine
• Format: Article
• Language: English
• Subjects: Clustering; Interpretable; Learning vector quantization (LVQ); Survival analysis
• ISSN: 0031-3203; 1873-5142
• DOI: 10.1016/j.patcog.2024.110497

Identifying subgroups with similar survival outcomes is a pivotal challenge in survival analysis. Traditional clustering methods often neglect the outcome variable, potentially leading to inaccurate representation of risk profiles. To address this, we present SurvivalLVQ, a novel interpretable method that adapts Learning Vector Quantization (LVQ) to survival analysis. Unlike traditional classification uses of LVQ, SurvivalLVQ groups individuals by survival probabilities and assigns a unique survival curve to each cluster, representing the collective survival behavior within that group. Moreover, it can predict individual survival curves using weighted averages from nearby clusters. When tested on 76 benchmark datasets, it outperformed other clustering methods and showed competitive prediction performance. SurvivalLVQ bridges the gap between clustering techniques and outcome-oriented methods. Its strong clustering performance, coupled with competitive prediction capabilities and with easy to interpret outcomes, make it a promising tool for various applications within survival analysis. •SurvivalLVQ: Learning Vector Quantization is adapted to survival analysis.•SurvivalLVQ groups cases by survival probability and assigns unique survival curves.•SurvivalLVQ demonstrates strong clustering and competitive predictive performance.•SurvivalLVQ bridges the gap between clustering and outcome-oriented methods.