高级检索
当前位置: 首页 > 详情页

基于wgbdt的心衰患者半年内再入院风险预测

WGBDT-based risk prediction for readmission within six months in heart failure patients

文献详情

资源类型:

收录情况: ◇ 北大核心

单位: [1]中国地质大学数学与物理学院 [2]华中科技大学同济医学院附属同济医院儿科学系
出处:
ISSN:

关键词: 心衰 再入院 梯度提升决策树 样本权重 可解释机器学习模型

摘要:
为了解决现有心衰患者再入院风险预测评估模型缺乏可解释性、无法满足临床应用要求的问题,提出了一种基于自适应加权的梯度提升决策树(Weighted Gradient Boosting Decision Trees, WGBDT)的心衰患者半年内再入院风险预测模型.这一模型由基于WGBDT算法的风险预测和基于可解释机器学习(SHapley Additive exPlanation, SHAP)模型的解释性框架构成.其一,WGBDT风险模型通过样本权重更新来完成基分类器的训练.对通过基分类器分类误差率更新的残差样本进行迭代训练的基分类器加权累加,可以获得泛化性和准确率更好的模型;其二,SHAP可解释性框架采用Kernel SHAP和临床医学先验知识相结合的方式,对WGBDT黑箱模型进行解释,完成该模型的可解释性.使用四川省某医院的2008例心衰患者临床数据集对模型进行训练与测试,结果显示:利用该模型获得的结论优于梯度提升决策树(GBDT)、XGBoost、支持向量机、决策树、Adaboost等主流的机器学习算法获得的结论;同时,利用SHAP框架提高了该模型的可解释性,并根据特征的重要性,识别出了影响心衰因素的重要性排序,这为医生制定更加合理的决策提供了科学的参考.

基金:
语种:
第一作者:
第一作者单位: [1]中国地质大学数学与物理学院
通讯作者:
推荐引用方式(GB/T 7714):

资源点击量:426 今日访问量:2 总访问量:410 更新日期:2025-04-01 建议使用谷歌、火狐浏览器 常见问题

版权所有:重庆聚合科技有限公司 渝ICP备12007440号-3 地址:重庆市两江新区泰山大道西段8号坤恩国际商务中心16层(401121)