Responsible AI for Sepsis Prediction: Bridging the Gap Between Machine Learning Performance and Clinical Trust

Read full text
Authors: Thiago Q. Oliveira, Leandro A. Carvalho, Flávio R. C. Sousa, João B. F. Filho, Khalil F. Oliveira, Daniel A. B. Tavares
Cited by: 1

Abstract

Background: Sepsis remains a leading cause of mortality in Intensive Care Units (ICUs) worldwide. Machine learning models for clinical prediction must be accurate, fair, transparent, and reliable to ensure that physicians feel confident in their decision-making process. Methods: We used the MIMIC-IV, version 3.1, database to evaluate several machine learning architectures, including Logistic Regression, XGBoost, LightGBM, LSTM (Long Short-Term Memory) networks and Transformer models. We predicted three main clinical targets: hospital mortality, length of stay, and septic shock onset. Model interpretability was assessed using Shapley Additive Explanations (SHAP). Results: The XGBoost model demonstrated superior performance in prediction tasks, particularly for hospital mortality (AUROC 0.874), outperforming traditional LSTM networks, transformers and linear baselines. Importance analysis of the variables confirmed the clinical relevance of the model. Conclusions: While XGBoost and ensemble algorithms demonstrate superior predictive power for sepsis prognosis, their clinical adoption necessitates robust explainability mechanisms to gain the doctors trust.

Fields of Study & Subjects

Sepsis Diagnosis and Treatment Machine Learning in Healthcare Artificial Intelligence in Healthcare and Education

Publication details

Journal / source
Preprints.org
Publisher
MDPI AG
Publication date
30 January 2026
Document type
Preprint