Abstract:Background and Objectives: Sepsis is one of the common causes of death in intensive care units. A reliable prognostic model based on patients' data acquired at the intensive care unit (ICU) would enable clinicians to make treatment decisions to improve clinical outcomes for septic patients. This study aims to develop a machine-learning framework for building such prognostic tools by exploring the class-imbalanced longitudinal data of a group of septic patients.Methods: A feature-represented input dataset is devised in the form of concatenated triples to increase the data size relative to the dimension of the feature space. Each concatenated triplet consists of a patient's static data, the k-day consecutively collected longitudinal data, and the clinical outcome (k=2,3,4,5). The structured input data are then used to train classifiers in combination with appropriate feature engineering techniques. The trained classifiers are tested on a new set of septic patients to ensure their clinical efficacy. We implement the modeling approach using five classifiers: K nearest neighbors, Logistic Regression, Support Vector Machine, Random Forest (RF), and Extreme Gradient Boosting (XGBoost) coupled with a set of feature engineering techniques. AUROC and a new metric, $\gamma$, made up of the F1 score on the external validation set, are used to assess the efficacy of the models.Results: Five prognostic models are built on the feature-represented input dataset accounting for 10 selected dynamic features from the patient medical data. Our research shows that the XGBoost (AUROC=0.777, F1 score=0.694) and RF (AUROC=0.769, F1 score=0.647) model combined with the ensemble under-sampling strategy outperform all other models in the external validation or testing. For example, the improvement in AUROC and overfitting are (6.66\%, 54.96\%) and (0.52\%, 77.72\%) for the RF and XGBoost model with the sampling strategy compared to the same models without using the sampling strategy, respectively. This indicates that the machine-learning framework can greatly improve the accuracy and generalizability of standard classifiers. Conclusion:A new modeling framework is devised to develop prognostic tools for treatment outcomes of septic patients using small, class-imbalanced, and high-dimensional datasets. It enables standard classifiers to use small datasets to achieve relatively high predictability by engineering new structured datasets encoded with temporal features, sampling strategies, and dimension reduction techniques, providing clinically useful prognostic models and setting an example for applying machine learning methods to small data problems in medicine.

Machine learning-based prognostic model for 30-day mortality prediction in Sepsis-3

Prediction of 30-Day Mortality for ICU Patients with Sepsis-3

Predicting 30-days mortality for MIMIC-III patients with sepsis-3: a machine learning approach using XGboost

Prediction of Sepsis Mortality in ICU Patients Using Machine Learning Methods

Machine-Learning-Enabled Prognostic Models for Sepsis

Using Machine Learning Algorithms to predict sepsis and its stages in ICU patients

Using Machine Learning to Develop and Validate an In-Hospital Mortality Prediction Model for Patients with Suspected Sepsis

Machine learning approach for the prediction of 30-day mortality in patients with sepsis-associated encephalopathy

Early Prediction of Mortality in Critical Care Setting in Sepsis Patients Using Structured Features and Unstructured Clinical Notes

Using machine learning methods to predict in-hospital mortality of sepsis patients in the ICU

Data-Driven Machine Learning Approaches for Predicting In-Hospital Sepsis Mortality

Enhancing predictions with a stacking ensemble model for ICU mortality risk in patients with sepsis-associated encephalopathy

Predicting Mortality in Sepsis-Associated Acute Respiratory Distress Syndrome: A Machine Learning Approach Using the MIMIC-III Database

Early Prediction of Mortality, Severity, and Length of Stay in the Intensive Care Unit of Sepsis Patients Based on Sepsis 3.0 by Machine Learning Models

Machine learning models to predict 30-day mortality for critical patients with myocardial infarction: a retrospective analysis from MIMIC-IV database

A Machine Learning-Based Prediction of Hospital Mortality in Patients With Postoperative Sepsis

Machine Learning Models for Early Prediction of Sepsis on Large Healthcare Datasets

A Machine Learning Predictive Model of In-Hospital Mortality in Patients with Sepsis Complicated by Anemia: a Retrospective Study Based on the MIMIC-III Database

A Machine Learning Model for Early Prediction and Detection of Sepsis in Intensive Care Unit Patients

The prediction power of machine learning on estimating the sepsis mortality at intensive care unit