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

基于 Kalman Filter 的院内死亡预测模型性能随时间监测框架

A Kalman Filter Based Framework for Monitoring the Performance of In-Hospital Mortality Prediction Models Over Time

University of Minnesota(明尼苏达大学);Abbott Northwestern Hospital(雅培西北医院);Allina Health(阿利纳健康医疗系统)

Jiacheng Liu, Lisa Kirkland, Jaideep Srivastava

AI总结:

本研究提出一种基于 Kalman filter 的框架,通过调整样本量和类别分布的外推方差,解决真实场景下院内死亡预测模型性能指标在不同时间段不可比的问题,并在合成数据与 COVID-19 患者数据上验证了其有效性。

摘要:

与研究人员可以确定所需正负样本最小数量的临床试验,或验证集大小和类别分布静态且已知的机器学习研究不同,在真实场景中,对入院患者的大小和分布几乎无法控制。因此,在不同时间段测量时,受试者工作特征曲线下面积(AUCROC)和精确率-召回率曲线下面积(AUCPR)等评估指标可能无法直接比较。因此,在本研究中,针对长时间运行的二分类器,我们提出根据样本量和类别分布调整这些性能指标,以便在两个时间段之间进行公平比较。注意,样本数量和类别分布(即正样本比例)是影响 AUCROC 方差的两个稳健性因素。为了更好地估计性能指标的均值并理解性能随时间的变化,我们提出了一种基于 Kalman filter 的框架,该框架具有针对不同时间段的总样本数和正样本数进行调整的外推方差。该方法的有效性首先在合成数据集上得到验证,随后回顾性应用于 2021 年和 2022 年 COVID-19 患者提前 2 天的院内死亡预测模型。此外,我们得出结论,我们的预测模型未受到疾病演变、治疗改进和医院运营计划变化的显著影响。

英文摘要:

Unlike in a clinical trial, where researchers get to determine the least number of positive and negative samples required, or in a machine learning study where the size and the class distribution of the validation set is static and known, in a real-world scenario, there is little control over the size and distribution of incoming patients. As a result, when measured during different time periods, evaluation metrics like Area under the Receiver Operating Curve (AUCROC) and Area Under the Precision-Recall Curve(AUCPR) may not be directly comparable. Therefore, in this study, for binary classifiers running in a long time period, we proposed to adjust these performance metrics for sample size and class distribution, so that a fair comparison can be made between two time periods. Note that the number of samples and the class distribution, namely the ratio of positive samples, are two robustness factors which affect the variance of AUCROC. To better estimate the mean of performance metrics and understand the change of performance over time, we propose a Kalman filter based framework with extrapolated variance adjusted for the total number of samples and the number of positive samples during different time periods. The efficacy of this method is demonstrated first on a synthetic dataset and then retrospectively applied to a 2-days ahead in-hospital mortality prediction model for COVID-19 patients during 2021 and 2022. Further, we conclude that our prediction model is not significantly affected by the evolution of the disease, improved treatments and changes in hospital operational plans.

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

↑