vix.ing · top · new · best · stats · spec

Improving Machine Learning Based Sepsis Diagnosis Using Heart Rate Variability

2024/08/01 by Sai A. Balaji, Balaji, Sai, Christopher Sun +3 · 1 citation
Engineering · Medicine · #FOS: Computer and information sciences #Heart Rate Variability and Autonomic Control #Machine Learning (cs.LG) #Non-Invasive Vital Sign Monitoring

paper · pdf · doi:10.48550/arxiv.2408.02683

openalex publication_date 2024/08/01 · openalex created_date 2025/01/17 · openalex updated_date 2026/07/28

Abstract

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.

Cited by

Related