可解释机器学习模型预测心力衰竭合并急性肾损伤患者在重症监护室内的短期死亡率：一项回顾性队列研究

摘要

目的 开发一种可解释的机器学习模型，以提高心力衰竭（heart failure, HF）合并急性肾损伤（acute kidney injury, AKI）患者短期死亡率的早期预测准确性。方法 本回顾性队列研究利用了基于电子病历的公开大型数据库重症医学信息数据库MIMIC-Ⅳ（Medical Information Mart for Intensive CareⅣ,版本2.0）。提取了患者入住ICU最初24 h的数据，并将其分为训练集（80%）和验证集（20%）。利用沙普利可加性解释（Shapley additive explanation, SHAP）方法解释极端梯度提升（XGBoost）模型的工作原理，并识别关键的预后因素。使用曲线下面积（area under the curve, AUC）指标评估XGBoost模型的预测能力，并与3种其他机器学习模型进行比较，其解释性通过SHAP方法得到增强。结果 研究包括8 028名合并AKI的HF患者。XGBoost模型的表现优于其他模型，达到了0.93的AUC[95%置信区间（confidence interval, CI）0.78～0.94，准确度=0.89]，而神经网络的表现最差（AUC=0.79,95%CI 0.77～0.82，准确度=0.82）。决策曲线分析显示XGBoost模型在9%至80%的阈值概率内具有更高的净收益。SHAP分析识别了前20个预测因素，年龄以及格拉斯哥昏迷评分被识别为重要的因素，SHAP平均值分别为1.29和1.24。结论 本研究开发的可解释模型提高了预测ICU内HF患者并发AKI的死亡风险的能力。这个模型不仅有助于制定有效的治疗计划，还能优化资源分配。

关键词

急性肾损伤 / 心力衰竭 / 机器学习 / 预测模型 / MIMIC

基金资助

国家自然科学基金（No.82300848）; 四川省科技厅重点研发项目（No.2022YFS0147）资助