Survival prediction plays a central role for healthcare providers and clinical researchers. Accurate risk stratification enables early intervention and improved patient management. Most existing deep survival models learn one common feature representation for all patients, which may hide important differences between patient subgroups. In contrast, a Mixture-of-Experts (MoE) framework allows different parts of the model to focus on different patient patterns, leading to more individualized representations.
Thus, we propose a mixture-of-experts enhanced adaptive deep clustering survival framework (AdaCSM) for modeling heterogeneous survival patterns. We introduce a routing-based expert mechanism that enables conditional specialization within a parametric survival modeling framework. The proposed architecture allocates patients to specialized risk predictors dynamically while preserving the patient survival and subtype clustering objectives.
We compare our method with state-of-the-art survival and deep clustering models on multiple real-world longitudinal clinical cohorts spanning diverse disease domains. The proposed method demonstrates improved predictive performance and leads to interpretable results in survival analysis.
Blogger's Review: This study significantly enhances the personalization and interpretability of survival prediction models by introducing a mixture-of-experts mechanism. AdaCSM shows superior performance, especially in handling heterogeneous patient populations, providing a fresh perspective for clinical applications worth further exploration.