A customised down-sampling machine learning approach for sepsis prediction

Qinhao Wu, Fei Ye, Qianqian Gu, Feng Shao, Xi Long, Zhuozhao Zhan, Junjie Zhang, Jun He, Yangzhou Zhang, Quan Xiao

Research output: Contribution to journalArticleAcademicpeer-review

1 Citation (Scopus)
34 Downloads (Pure)

Abstract

Objective
Sepsis is a life-threatening condition in the ICU and requires treatment in time. Despite the accuracy of existing sepsis prediction models, insufficient focus on reducing alarms could worsen alarm fatigue and desensitisation in ICUs, potentially compromising patient safety. In this retrospective study, we aim to develop an accurate, robust, and readily deployable method in ICUs, only based on the vital signs and laboratory tests.

Methods
Our method consists of a customised down-sampling process and a specific dynamic sliding window and XGBoost to offer sepsis prediction. The down-sampling process was applied to the retrospective data for training the XGBoost model. During the testing stage, the dynamic sliding window and the trained XGBoost were used to predict sepsis on the retrospective datasets, PhysioNet and FHC.

Results
With the filtered data from PhysioNet, our method achieved 80.74% accuracy (77.90% sensitivity and 84.42% specificity) and 83.95% (84.82% sensitivity and 82.00% specificity) on the test set of PhysioNet-A and PhysioNet-B, respectively. The AUC score was 0.89 for both datasets. On the FHC dataset, our method achieved 92.38% accuracy (88.37% sensitivity and 95.16% specificity) and 0.98 AUC score on the test set of FHC.

Conclusion
Our results indicate that the down-sampling process and the dynamic sliding window with XGBoost brought robust and accurate performance to give sepsis prediction under various hospital settings. The localisation and robustness of our method can assist in sepsis diagnosis in different ICU settings.
Original languageEnglish
Article number105365
Number of pages11
JournalInternational Journal of Medical Informatics
Volume184
Early online date6 Feb 2024
DOIs
Publication statusPublished - Apr 2024

Keywords

  • intensive care unit
  • sepsis prediction
  • machine learning
  • alarm reduction

Fingerprint

Dive into the research topics of 'A customised down-sampling machine learning approach for sepsis prediction'. Together they form a unique fingerprint.

Cite this