arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.02683cs.LG
收藏

利用心率变异性改进基于机器学习的脓毒症诊断

Improving Machine Learning Based Sepsis Diagnosis Using Heart Rate Variability

The Woodlands College Park High School(伍德兰兹学院公园高中);Stanford University(斯坦福大学);University of California, Berkeley(加州大学伯克利分校)

Sai Balaji, Christopher Sun, Anaiy Somalwar

AI总结:

本文利用心率变异性特征,结合特征工程、XGBoost、Random Forest、集成模型与神经网络进行脓毒症预测,并通过 LIME 增强模型可解释性。

摘要:

早期且准确地诊断脓毒症对于改善患者预后至关重要。本研究旨在利用心率变异性(HRV)特征开发一种有效的脓毒症检测预测模型。研究通过包括统计自助法(bootstrapping)和 Boruta 算法在内的特征工程方法识别关键 HRV 特征,随后采用不同的超参数设置训练 XGBoost 和 Random Forest 分类器。此外,研究构建了集成模型,以汇总高召回率分类器和高精度分类器的预测概率,从而提升模型性能。最后,研究基于 HRV 特征训练了一个神经网络模型,其 F1 分数达到 0.805,精确率为 0.851,召回率为 0.763。研究通过可解释性分析,将表现最佳的机器学习模型与该神经网络进行比较;其中实施了 Local Interpretable Model-agnostic Explanations(局部可解释模型无关解释,LIME),以依据特定特征的数值范围和阈值确定决策标准。本研究不仅突出了 HRV 在自动化脓毒症诊断中的功效,还提高了黑箱输出的透明度,从而最大化临床适用性。

英文摘要:

The early and accurate diagnosis of sepsis is critical for enhancing patient outcomes. This study aims to use heart rate variability (HRV) features to develop an effective predictive model for sepsis detection. Critical HRV features are identified through feature engineering methods, including statistical bootstrapping and the Boruta algorithm, after which XGBoost and Random Forest classifiers are trained with differential hyperparameter settings. In addition, ensemble models are constructed to pool the prediction probabilities of high-recall and high-precision classifiers and improve model performance. Finally, a neural network model is trained on the HRV features, achieving an F1 score of 0.805, a precision of 0.851, and a recall of 0.763. The best-performing machine learning model is compared to this neural network through an interpretability analysis, where Local Interpretable Model-agnostic Explanations are implemented to determine decision-making criterion based on numerical ranges and thresholds for specific features. This study not only highlights the efficacy of HRV in automated sepsis diagnosis but also increases the transparency of black box outputs, maximizing clinical applicability.

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

↑