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

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Shumilov, Arseniy, Zhu, Yueting, Ashrafi, Negin, Abdollahi, Armin, Placencia, Greg, Alaei, Kamiar, Pishgar, Maryam
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866915086107410432
author Shumilov, Arseniy
Zhu, Yueting
Ashrafi, Negin
Abdollahi, Armin
Placencia, Greg
Alaei, Kamiar
Pishgar, Maryam
author_facet Shumilov, Arseniy
Zhu, Yueting
Ashrafi, Negin
Abdollahi, Armin
Placencia, Greg
Alaei, Kamiar
Pishgar, Maryam
contents Sepsis is a severe condition responsible for many deaths in the United States and worldwide, making accurate prediction of outcomes crucial for timely and effective treatment. Previous studies employing machine learning faced limitations in feature selection and model interpretability, reducing their clinical applicability. This research aimed to develop an interpretable and accurate machine learning model to predict in-hospital sepsis mortality, addressing these gaps. Using ICU patient records from the MIMIC-III database, we extracted relevant data through a combination of literature review, clinical input refinement, and Random Forest-based feature selection, identifying the top 35 features. Data preprocessing included cleaning, imputation, standardization, and applying the Synthetic Minority Over-sampling Technique (SMOTE) to address class imbalance, resulting in a dataset of 4,683 patients with 17,429 admissions. Five models-Random Forest, Gradient Boosting, Logistic Regression, Support Vector Machine, and K-Nearest Neighbor-were developed and evaluated. The Random Forest model demonstrated the best performance, achieving an accuracy of 0.90, AUROC of 0.97, precision of 0.93, recall of 0.91, and F1-score of 0.92. These findings underscore the potential of data-driven machine learning approaches to improve critical care, offering clinicians a powerful tool for predicting in-hospital sepsis mortality and enhancing patient outcomes.
format Preprint
id arxiv_https___arxiv_org_abs_2408_01612
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Data-Driven Machine Learning Approaches for Predicting In-Hospital Sepsis Mortality
Shumilov, Arseniy
Zhu, Yueting
Ashrafi, Negin
Abdollahi, Armin
Placencia, Greg
Alaei, Kamiar
Pishgar, Maryam
Machine Learning
Sepsis is a severe condition responsible for many deaths in the United States and worldwide, making accurate prediction of outcomes crucial for timely and effective treatment. Previous studies employing machine learning faced limitations in feature selection and model interpretability, reducing their clinical applicability. This research aimed to develop an interpretable and accurate machine learning model to predict in-hospital sepsis mortality, addressing these gaps. Using ICU patient records from the MIMIC-III database, we extracted relevant data through a combination of literature review, clinical input refinement, and Random Forest-based feature selection, identifying the top 35 features. Data preprocessing included cleaning, imputation, standardization, and applying the Synthetic Minority Over-sampling Technique (SMOTE) to address class imbalance, resulting in a dataset of 4,683 patients with 17,429 admissions. Five models-Random Forest, Gradient Boosting, Logistic Regression, Support Vector Machine, and K-Nearest Neighbor-were developed and evaluated. The Random Forest model demonstrated the best performance, achieving an accuracy of 0.90, AUROC of 0.97, precision of 0.93, recall of 0.91, and F1-score of 0.92. These findings underscore the potential of data-driven machine learning approaches to improve critical care, offering clinicians a powerful tool for predicting in-hospital sepsis mortality and enhancing patient outcomes.
title Data-Driven Machine Learning Approaches for Predicting In-Hospital Sepsis Mortality
topic Machine Learning
url https://arxiv.org/abs/2408.01612