arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2103.11254cs.LGstat.AP
收藏

通过基于树的机器学习模型预测的SHAP解释了解心力衰竭患者EHR的临床特征

Understanding Heart-Failure Patients EHR Clinical Features via SHAP Interpretation of Tree-Based Machine Learning Model Predictions

University of Pittsburgh(匹兹堡大学);Beijing Information Science & Technology University(北京信息科技大学);University of Pittsburgh Medical Center(匹兹堡大学医学中心);Astrata Solutions(阿斯特拉塔解决方案公司)

Shuyu Lu, Ruoyu Chen, Wei Wei, Xinghua Lu

摘要:

心力衰竭(HF)是死亡的主要原因。准确监测HF进展和调整治疗对于改善患者结局至关重要。经验丰富的心脏病专家可以根据患者电子健康记录(EHR)中的症状、体征和实验室结果的组合做出准确的HF分期诊断,而无需直接测量心脏功能。我们研究了机器学习模型,更具体地说是XGBoost模型,是否可以基于EHR准确地预测患者阶段,我们进一步应用SHapley加法解释(SHAP)框架来识别信息特征及其解释。我们的研究结果表明,基于EHR的结构化数据,我们的模型可以预测患者的射血分数(EF)评分,具有中等的准确性。SHAP分析确定了信息特征并揭示了HF的潜在临床亚型。我们的研究结果为如何设计计算系统提供了见解,以通过持续挖掘患者的EHR数据来准确监测HF患者的疾病进展。

英文摘要:

Heart failure (HF) is a major cause of mortality. Accurately monitoring HF progress and adjust therapies are critical for improving patient outcomes. An experienced cardiologist can make accurate HF stage diagnoses based on combination of symptoms, signs, and lab results from the electronic health records (EHR) of a patient, without directly measuring heart function. We examined whether machine learning models, more specifically the XGBoost model, can accurately predict patient stage based on EHR, and we further applied the SHapley Additive exPlanations (SHAP) framework to identify informative features and their interpretations. Our results indicate that based on structured data from EHR, our models could predict patients' ejection fraction (EF) scores with moderate accuracy. SHAP analyses identified informative features and revealed potential clinical subtypes of HF. Our findings provide insights on how to design computing systems to accurately monitor disease progression of HF patients through continuously mining patients' EHR data.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑