用专家分工机制提升临床生存预测的分组精度与可解释性
Expert-Driven Survival Machines: Improving Stratification and Interpretability in Multiple Clinical Cohorts

- 采用路由机制让不同专家专注特定患者群体,实现个性化风险预测
- 在多个真实临床队列上显著优于现有模型,提升生存预测准确性
- 结果可解释性强,适合需要精准分层的医疗决策场景
生存预测在医疗和临床研究中至关重要,准确的风险分层有助于早期干预和优化患者管理。现有深度生存模型通常为所有患者学习统一特征表示,可能掩盖患者亚群间的差异。为此,本文提出一种基于混合专家(MoE)的自适应深度聚类生存框架(AdaCSM),通过路由机制实现参数化生存建模中的条件专业化。该架构动态分配患者至特定风险预测器,同时保持生存预测与亚型聚类目标。我们在涵盖多种疾病领域的多个真实纵向临床队列上,对比了该方法与当前最优的生存模型及深度聚类模型。实验表明,所提方法在预测性能上表现更优,并生成可解释的生存分析结果。
原文摘要 · Abstract (English)
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. Therefore, in this work, we propose a mixture-of-experts enhanced adaptive deep clustering survival framework (AdaCSM) for modeling such 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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。