arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2311.13015cs.LGcs.CY

面向重症监护患者的快速且可解释的死亡风险评分

Fast and Interpretable Mortality Risk Scores for Critical Care Patients

Department of Computer Science, Duke University(杜克大学计算机科学系);Microsoft Research(微软研究院);Cornell University(康奈尔大学)

Chloe Qinyu Zhu, Muhang Tian, Lesia Semenova, Jiachang Liu, Jack Xu, Joseph Scarpa, Cynthia Rudin

AI总结:

本文提出 GroupFasterRisk 算法,利用硬/软稀疏、分组稀疏和单调性约束生成多种可解释死亡风险评分,在 MIMIC III 与 eICU 上达到黑盒模型精度且显著更稀疏。

摘要:

重症监护病房(ICU)患者的死亡预测通常依赖黑盒模型(医院无法接受其使用)或人工调校的可解释模型(可能导致性能损失)。我们旨在通过基于现代可解释机器学习技术设计与黑盒模型同样准确的可解释死亡风险评分,弥合这两类模型之间的差距。我们开发了一种名为 GroupFasterRisk 的新算法,它具有若干重要优势:同时使用硬直接稀疏正则化和软直接稀疏正则化;引入分组稀疏性以获得更具内聚性的模型;支持单调性约束以纳入领域知识;并生成许多同样优良的模型,使领域专家能够从中选择。在评估方面,我们利用了现有最大的公开 ICU 监测数据集(MIMIC III 和 eICU)。GroupFasterRisk 生成的模型优于 OASIS 和 SAPS II 评分,并且在仅使用至多三分之一参数的情况下表现与 APACHE IV/IVa 相似。对于患有脓毒症/败血症、急性心肌梗死、心力衰竭和急性肾衰竭的患者,GroupFasterRisk 模型优于 OASIS 和 SOFA。最后,与基于 OASIS 变量相比,不同的死亡预测机器学习方法在使用 GroupFasterRisk 选择的变量时表现更好。GroupFasterRisk 的模型表现优于医院当前使用的风险评分,并与黑盒机器学习模型相当,同时稀疏程度高出数个数量级。由于 GroupFasterRisk 可生成多种风险评分,它带来了设计灵活性——这是实际模型创建的关键推动因素。GroupFasterRisk 是一种快速、易用且灵活的流程,能够学习一组多样化的稀疏风险评分以用于死亡预测。

英文摘要:

Prediction of mortality in intensive care unit (ICU) patients typically relies on black box models (that are unacceptable for use in hospitals) or hand-tuned interpretable models (that might lead to the loss in performance). We aim to bridge the gap between these two categories by building on modern interpretable ML techniques to design interpretable mortality risk scores that are as accurate as black boxes. We developed a new algorithm, GroupFasterRisk, which has several important benefits: it uses both hard and soft direct sparsity regularization, it incorporates group sparsity to allow more cohesive models, it allows for monotonicity constraint to include domain knowledge, and it produces many equally-good models, which allows domain experts to choose among them. For evaluation, we leveraged the largest existing public ICU monitoring datasets (MIMIC III and eICU). Models produced by GroupFasterRisk outperformed OASIS and SAPS II scores and performed similarly to APACHE IV/IVa while using at most a third of the parameters. For patients with sepsis/septicemia, acute myocardial infarction, heart failure, and acute kidney failure, GroupFasterRisk models outperformed OASIS and SOFA. Finally, different mortality prediction ML approaches performed better based on variables selected by GroupFasterRisk as compared to OASIS variables. GroupFasterRisk's models performed better than risk scores currently used in hospitals, and on par with black box ML models, while being orders of magnitude sparser. Because GroupFasterRisk produces a variety of risk scores, it allows design flexibility - the key enabler of practical model creation. GroupFasterRisk is a fast, accessible, and flexible procedure that allows learning a diverse set of sparse risk scores for mortality prediction.

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

↑