arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2212.00554cs.LGcs.AI
收藏

基于深度联邦学习的ICU死亡风险早期预测

Early prediction of the risk of ICU mortality with Deep Federated Learning

Stockholm University(斯德哥尔摩大学)

Korbinian Randl, Núria Lladós Armengol, Lena Mondrejevski, Ioanna Miliou

AI总结:

本研究评估深度联邦学习在早期预测ICU死亡风险中的能力,比较显示其性能与集中式方法相当且显著优于本地方法,为隐私保护下的跨医院ICU死亡预测提供可行解决方案。

摘要:

重症监护室(ICU)通常收治具有严重死亡风险的患者。近期研究表明,机器学习能够指示患者的死亡风险,并引导医生关注那些护理需求较高的个体。然而,医疗数据通常受隐私法规约束,因此无法轻易共享以构建使用多家医院联合数据的集中式机器学习模型。联邦学习是一种为数据隐私设计的机器学习框架,可用于规避此问题。在本研究中,我们评估了深度联邦学习在早期预测ICU死亡风险方面的能力。我们在AUPRC、F1-score和AUROC方面比较了联邦、集中式和本地机器学习的预测性能。结果表明,联邦学习的表现与集中式方法同样出色,且显著优于本地方法,从而为早期ICU死亡预测提供了可行的解决方案。此外,我们发现当患者病史窗口越接近出院或死亡时,预测性能越高。最后,我们证明使用F1-score作为早停指标可以稳定并提升本方法在当前任务上的性能。

英文摘要:

Intensive Care Units usually carry patients with a serious risk of mortality. Recent research has shown the ability of Machine Learning to indicate the patients' mortality risk and point physicians toward individuals with a heightened need for care. Nevertheless, healthcare data is often subject to privacy regulations and can therefore not be easily shared in order to build Centralized Machine Learning models that use the combined data of multiple hospitals. Federated Learning is a Machine Learning framework designed for data privacy that can be used to circumvent this problem. In this study, we evaluate the ability of deep Federated Learning to predict the risk of Intensive Care Unit mortality at an early stage. We compare the predictive performance of Federated, Centralized, and Local Machine Learning in terms of AUPRC, F1-score, and AUROC. Our results show that Federated Learning performs equally well as the centralized approach and is substantially better than the local approach, thus providing a viable solution for early Intensive Care Unit mortality prediction. In addition, we show that the prediction performance is higher when the patient history window is closer to discharge or death. Finally, we show that using the F1-score as an early stopping metric can stabilize and increase the performance of our approach for the task at hand.

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

↑