The impact of evaluation strategy on sepsis prediction model performance metrics in intensive care data
Do, D.-K.; Rockenschaub, P.; Boie, S.; Kumpf, O.; Volk, H.-D.; Balzer, F.; von Dincklage, F.; Lichtner, G.
Show abstract
BackgroundThe prediction of the onset of sepsis, a life-threatening condition resulting from a dysregulated response to an infection, is one of the most common prediction tasks in intensive care-related machine learning research. To assess the performance of such models, different evaluation strategies (fixed horizon, peak score and continuous evaluation) are commonly employed, but there is no clear consensus on which approach should be used in order to provide clinically meaningful performance evaluation. ObjectiveTo assess different evaluation approaches of sepsis prediction models trained on a public intensive care dataset applied to German intensive care data. MethodsIn this retrospective, observational cohort study, we assessed the efficacy of machine learning models, pre-trained on the MIMIC-IV dataset, when applied to BerlinICU, a multi-site German intensive care dataset. To understand the real-world impact of implementing these models, we examined the performance variability across various evaluation strategies. ResultsThe BerlinICU dataset includes 40,132 intensive care admissions spanning 10 years (2012-2021). Using the latest Sepsis-3 definition, we identified 4,134 septic admissions (prevalence 10.3%). Application of a temporal convolution network model to BerlinICU yielded an area under the receiver operating characteristic curve (AUROC) of 0.67 (95% CI: 0.66-0.68) for continuous evaluation with a 6-hour prediction horizon, compared to 0.84 (95% CI: 0.83-0.85) on the test set of MIMIC-IV. On BerlinICU, peak score evaluation showed a similar AUROC compared to continuous evaluation, while fixed horizon evaluation showed a reduced AUROC of 0.61 (95% CI: 0.60-0.62). Onset matching had minimal impact on performance estimates using continuous evaluation or fixed horizon evaluation but increased estimates for peak score evaluation. Performance metrics improved with shorter prediction horizons across all strategies. ConclusionOur results demonstrate that the choice of evaluation strategy has a significant impact on the performance metrics of intensive care prediction models. The same model applied to the same dataset yields markedly different performance metrics depending on the evaluation approach. Therefore, careful selection of the evaluation approach is essential to ensure that the interpretation of performance metrics aligns with clinical intentions and enables meaningful comparisons between studies. In our view, the continuous evaluation approach best reflects the continual monitoring of patients that is performed in real-world clinical practice. In contrast, fixed horizon and peak score evaluation approaches may produce skewed results when not properly matching the length of stay distributions between sepsis cases and control cases. Especially for peak score evaluation, longer visits tend to produce higher maximum scores because sampling from more values increases the likelihood of capturing higher values purely by chance.
Matching journals
The top 5 journals account for 50% of the predicted probability mass.
Similar papers in this journal
- Development and Prospective Implementation of a Large Language Model based System for Early Sepsis Prediction 95%
- A comprehensive ML-based Respiratory Monitoring System for Physiological Monitoring & Resource Planning in the ICU 94%
- GLUCOSE: A Distributional Reinforcement Learning Model for Optimal Glucose Control After Cardiac Surgery 93%
Similar papers in this journal
- A comparison of machine learning models versus clinical evaluation for mortality prediction in patients with sepsis 97%
- Development of a Risk Prediction Model for Sepsis-Related Delirium Based on Multiple Machine Learning Approaches and an Online Calculator 94%
- Development and validation of a cellular host response test as an early diagnostic for sepsis 94%
Similar papers in this journal
- SWIFT: A Deep Learning Approach to Prediction of Hypoxemic Events in Critically-Ill Patients Using SpO 2 Waveform Prediction 93%
- Contrasting factors associated with COVID-19-related ICU admission and death outcomes in hospitalised patients by means of Shapley values 93%
- Predicting the causative pathogen among children with pneumonia using a causal Bayesian network 92%
Similar papers in this journal
- Predicting bloodstream infection outcome using machine learning 97%
- Evaluation of Domain Generalization and Adaptation on Improving Model Robustness to Temporal Dataset Shift in Clinical Medicine 94%
- Using explainable machine learning to identify patients at risk of reattendance at discharge from emergency departments 94%
Similar papers in this journal
- Generalizability Challenges of Mortality Risk Prediction Models: A Retrospective Analysis on a Multi-center Database 94%
- Identification of predictive patient characteristics for assessing the probability of COVID-19 in-hospital mortality 93%
- Identification of physiological adverse events using continuous vital signs monitoring during paediatric critical care transport: a novel data-driven approach 92%
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.