Prediction of mortality risk in patients with severe community-acquired pneumonia in the intensive care unit using machine learning

Abstract The aim of this study was to develop and validate a machine learning-based mortality risk prediction model for patients with severe community-acquired pneumonia (SCAP) in the intensive care unit (ICU). We collected data from two centers as the development and external validation cohorts. Va...

Full description

Saved in:
Bibliographic Details
Main Authors: Jingjing Pan, Tao Guo, Haobo Kong, Wei Bu, Min Shao, Zhi Geng
Format: Article
Language:English
Published: Nature Portfolio 2025-01-01
Series:Scientific Reports
Subjects:
Online Access:https://doi.org/10.1038/s41598-025-85951-x
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1841544712312324096
author Jingjing Pan
Tao Guo
Haobo Kong
Wei Bu
Min Shao
Zhi Geng
author_facet Jingjing Pan
Tao Guo
Haobo Kong
Wei Bu
Min Shao
Zhi Geng
author_sort Jingjing Pan
collection DOAJ
description Abstract The aim of this study was to develop and validate a machine learning-based mortality risk prediction model for patients with severe community-acquired pneumonia (SCAP) in the intensive care unit (ICU). We collected data from two centers as the development and external validation cohorts. Variables were screened using the Recursive Feature Elimination method. Five machine learning algorithms were used to build predictive models. Models were evaluated through nested cross-validation to select the best one. The model was interpreted using Shapley Additive Explanations. We selected the optimal model to generate the web calculator. A total of 23 predictive features were selected. The Light Gradient Boosting Machine (LightGBM) model had an area under the receiver operating characteristic curve (AUC) of 0.842 (95% CI: 0.757–0.927), with an external 5-fold cross-validation average AUC of 0.842 ± 0.038, which was superior to the other models. External validation results also demonstrated good performance by the LightGBM model with an AUC of 0.856 (95% CI: 0.792–0.921). Based on this, we generated a web calculator by combining five high importance predictive factors. The LightGBM model was confirmed to be efficient and stable in predicting the mortality risk of patients with SCAP admitted to the ICU. The web calculator based on the LightGBM model can provide clinicians with a prognostic evaluation tool.
format Article
id doaj-art-fddd1951a0f940f49cf3c434c1c09dda
institution Kabale University
issn 2045-2322
language English
publishDate 2025-01-01
publisher Nature Portfolio
record_format Article
series Scientific Reports
spelling doaj-art-fddd1951a0f940f49cf3c434c1c09dda2025-01-12T12:20:54ZengNature PortfolioScientific Reports2045-23222025-01-0115111510.1038/s41598-025-85951-xPrediction of mortality risk in patients with severe community-acquired pneumonia in the intensive care unit using machine learningJingjing Pan0Tao Guo1Haobo Kong2Wei Bu3Min Shao4Zhi Geng5Department of Pulmonary and Critical Care Medicine, Anhui Chest HospitalCenter for Biomedical Imaging, University of Science and Technology of ChinaDepartment of Pulmonary and Critical Care Medicine, Anhui Chest HospitalDepartment of Critical Care Medicine, The First Affiliated Hospital of Anhui Medical UniversityDepartment of Critical Care Medicine, The First Affiliated Hospital of Anhui Medical UniversityDepartment of Neurology, The First Affiliated Hospital of Anhui Medical UniversityAbstract The aim of this study was to develop and validate a machine learning-based mortality risk prediction model for patients with severe community-acquired pneumonia (SCAP) in the intensive care unit (ICU). We collected data from two centers as the development and external validation cohorts. Variables were screened using the Recursive Feature Elimination method. Five machine learning algorithms were used to build predictive models. Models were evaluated through nested cross-validation to select the best one. The model was interpreted using Shapley Additive Explanations. We selected the optimal model to generate the web calculator. A total of 23 predictive features were selected. The Light Gradient Boosting Machine (LightGBM) model had an area under the receiver operating characteristic curve (AUC) of 0.842 (95% CI: 0.757–0.927), with an external 5-fold cross-validation average AUC of 0.842 ± 0.038, which was superior to the other models. External validation results also demonstrated good performance by the LightGBM model with an AUC of 0.856 (95% CI: 0.792–0.921). Based on this, we generated a web calculator by combining five high importance predictive factors. The LightGBM model was confirmed to be efficient and stable in predicting the mortality risk of patients with SCAP admitted to the ICU. The web calculator based on the LightGBM model can provide clinicians with a prognostic evaluation tool.https://doi.org/10.1038/s41598-025-85951-xMachine learningSevere community-acquired pneumoniaMortality risk predictionIntensive care unit
spellingShingle Jingjing Pan
Tao Guo
Haobo Kong
Wei Bu
Min Shao
Zhi Geng
Prediction of mortality risk in patients with severe community-acquired pneumonia in the intensive care unit using machine learning
Scientific Reports
Machine learning
Severe community-acquired pneumonia
Mortality risk prediction
Intensive care unit
title Prediction of mortality risk in patients with severe community-acquired pneumonia in the intensive care unit using machine learning
title_full Prediction of mortality risk in patients with severe community-acquired pneumonia in the intensive care unit using machine learning
title_fullStr Prediction of mortality risk in patients with severe community-acquired pneumonia in the intensive care unit using machine learning
title_full_unstemmed Prediction of mortality risk in patients with severe community-acquired pneumonia in the intensive care unit using machine learning
title_short Prediction of mortality risk in patients with severe community-acquired pneumonia in the intensive care unit using machine learning
title_sort prediction of mortality risk in patients with severe community acquired pneumonia in the intensive care unit using machine learning
topic Machine learning
Severe community-acquired pneumonia
Mortality risk prediction
Intensive care unit
url https://doi.org/10.1038/s41598-025-85951-x
work_keys_str_mv AT jingjingpan predictionofmortalityriskinpatientswithseverecommunityacquiredpneumoniaintheintensivecareunitusingmachinelearning
AT taoguo predictionofmortalityriskinpatientswithseverecommunityacquiredpneumoniaintheintensivecareunitusingmachinelearning
AT haobokong predictionofmortalityriskinpatientswithseverecommunityacquiredpneumoniaintheintensivecareunitusingmachinelearning
AT weibu predictionofmortalityriskinpatientswithseverecommunityacquiredpneumoniaintheintensivecareunitusingmachinelearning
AT minshao predictionofmortalityriskinpatientswithseverecommunityacquiredpneumoniaintheintensivecareunitusingmachinelearning
AT zhigeng predictionofmortalityriskinpatientswithseverecommunityacquiredpneumoniaintheintensivecareunitusingmachinelearning