arXiv:2502.05950cs.LGcs.AI2025-02

将概念学习引入生存分析,提升预测准确性和可解释性。

Survival Concept-Based Learning Models

  • 结合概念瓶颈与正则化,用人类可懂概念建模生存事件时间。
  • 在真实医疗数据上,模型优于传统生存分析方法,尤其在处理删失数据时。
  • 适合需要可解释性的医学、可靠性等领域研究者使用。

基于概念的学习通过利用高层次、人类可理解的概念,提升了预测准确性和可解释性。然而,现有的概念学习框架未涉及生存分析任务,该任务需在存在删失数据的情况下预测事件发生时间,常见于医学和可靠性分析领域。为填补这一空白,我们提出两种新模型:SurvCBM(生存概念瓶颈模型)和SurvRCM(生存正则化概念模型),将概念学习与生存分析相结合,以处理删失事件时间数据。两模型分别采用Cox比例风险模型和Beran估计器。SurvCBM基于知名的概念瓶颈模型架构,通过概念解释实现可解释预测;SurvRCM则以概念作为正则化手段,提升准确性。两者均端到端训练,并提供基于概念的可解释预测。提出了两种可解释性方法:一种利用Cox模型中的线性关系,另一种基于实例的解释框架结合Beran估计器。数值实验表明,SurvCBM优于SurvRCM及传统生存模型,凸显了引入概念信息的重要性。所提算法代码已公开。

原文摘要 · Abstract (English)

Concept-based learning enhances prediction accuracy and interpretability by leveraging high-level, human-understandable concepts. However, existing CBL frameworks do not address survival analysis tasks, which involve predicting event times in the presence of censored data -- a common scenario in fields like medicine and reliability analysis. To bridge this gap, we propose two novel models: SurvCBM (Survival Concept-based Bottleneck Model) and SurvRCM (Survival Regularized Concept-based Model), which integrate concept-based learning with survival analysis to handle censored event time data. The models employ the Cox proportional hazards model and the Beran estimator. SurvCBM is based on the architecture of the well-known concept bottleneck model, offering interpretable predictions through concept-based explanations. SurvRCM uses concepts as regularization to enhance accuracy. Both models are trained end-to-end and provide interpretable predictions in terms of concepts. Two interpretability approaches are proposed: one leveraging the linear relationship in the Cox model and another using an instance-based explanation framework with the Beran estimator. Numerical experiments demonstrate that SurvCBM outperforms SurvRCM and traditional survival models, underscoring the importance and advantages of incorporating concept information. The code for the proposed algorithms is publicly available.

生存分析可解释性概念学习医疗AI

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