LLM2D
基于生存概念的學習模型
Survival Concept-Based Learning Models
作者: Stanislav R. Kirpichenko, Lev V. Utkin, Andrei V. Konstantinov, Natalya M. Verbova
发布日期: 2/11/2025
arXiv ID: oai:arXiv.org:2502.05950v1

摘要

arXiv:2502.05950v1 标题类型: cross 摘要:基于概念的学习通过利用高层次的人类可理解的概念来提高预测准确性和可解释性。然而,现有的基于概念的学习(CBL)框架并未解决涉及截尾数据下事件时间预测的任务——这是医学和可靠性分析等领域中的常见场景。为了弥合这一差距,我们提出了两种新的模型:SurvCBM(生存基于概念的瓶颈模型)和SurvRCM(生存正则化基于概念的模型),将基于概念的学习与生存分析结合起来处理截尾事件时间数据。这些模型采用Cox比例风险模型和Beran估计器。SurvCBM基于知名的概念瓶颈模型架构,通过基于概念的解释提供可解释的预测。SurvRCM使用概念作为正则化来提高准确性。两种模型都是端到端训练的,并以概念的形式提供可解释的预测。提出了两种可解释性方法:一种是利用Cox模型中的线性关系,另一种是利用Beran估计器的实例解释框架。数值实验表明,SurvCBM优于SurvRCM和传统生存模型,突显了将概念信息融入的重要性与优势。提出算法的代码已公开可供访问。