An interpretable machine learning model for predicting in-hospital mortality in ICU patients with ventilator-associated pneumonia.
<h4>Background</h4>Ventilator-associated pneumonia (VAP) is a common nosocomial infection in ICU, significantly associated with poor outcomes. However, there is currently a lack of reliable and interpretable tools for assessing the risk of in-hospital mortality in VAP patients. This stud...
Saved in:
| Main Authors: | , , , , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
Public Library of Science (PLoS)
2025-01-01
|
| Series: | PLoS ONE |
| Online Access: | https://doi.org/10.1371/journal.pone.0316526 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1841533186111176704 |
|---|---|
| author | Junying Wei Heshan Cao Mingling Peng Yinzhou Zhang Sibei Li Wuhua Ma Yuhui Li |
| author_facet | Junying Wei Heshan Cao Mingling Peng Yinzhou Zhang Sibei Li Wuhua Ma Yuhui Li |
| author_sort | Junying Wei |
| collection | DOAJ |
| description | <h4>Background</h4>Ventilator-associated pneumonia (VAP) is a common nosocomial infection in ICU, significantly associated with poor outcomes. However, there is currently a lack of reliable and interpretable tools for assessing the risk of in-hospital mortality in VAP patients. This study aims to develop an interpretable machine learning (ML) prediction model to enhance the assessment of in-hospital mortality risk in VAP patients.<h4>Methods</h4>This study extracted VAP patient data from versions 2.2 and 3.1 of the MIMIC-IV database, using version 2.2 for model training and validation, and version 3.1 for external testing. Feature selection was conducted using the Boruta algorithm, and 14 ML models were constructed. The optimal model was identified based on the area under the receiver operating characteristic curve (AUROC), accuracy, sensitivity, and specificity across both validation and test cohorts. SHapley Additive exPlanations (SHAP) analysis was applied for global and local interpretability.<h4>Results</h4>A total of 1,894 VAP patients were included, with 12 features ultimately selected for model construction: 24-hour urine output, blood urea nitrogen, age, diastolic blood pressure, platelet count, anion gap, body temperature, bicarbonate level, sodium level, body mass index, and whether combined with congestive heart failure and cerebrovascular disease. The random forest (RF) model showed the best performance, achieving an AUC of 0.780 in internal validation and 0.724 in external testing, outperforming other ML models and common clinical scoring systems.<h4>Conclusion</h4>The RF model demonstrated robust and reliable performance in predicting in-hospital mortality risk for VAP patients. The developed online tool can assist clinicians in efficiently assessing VAP in-hospital mortality risk, supporting clinical decision-making. |
| format | Article |
| id | doaj-art-ac94c732a5db4d8c9cab5815225d31d5 |
| institution | Kabale University |
| issn | 1932-6203 |
| language | English |
| publishDate | 2025-01-01 |
| publisher | Public Library of Science (PLoS) |
| record_format | Article |
| series | PLoS ONE |
| spelling | doaj-art-ac94c732a5db4d8c9cab5815225d31d52025-01-17T05:31:40ZengPublic Library of Science (PLoS)PLoS ONE1932-62032025-01-01201e031652610.1371/journal.pone.0316526An interpretable machine learning model for predicting in-hospital mortality in ICU patients with ventilator-associated pneumonia.Junying WeiHeshan CaoMingling PengYinzhou ZhangSibei LiWuhua MaYuhui Li<h4>Background</h4>Ventilator-associated pneumonia (VAP) is a common nosocomial infection in ICU, significantly associated with poor outcomes. However, there is currently a lack of reliable and interpretable tools for assessing the risk of in-hospital mortality in VAP patients. This study aims to develop an interpretable machine learning (ML) prediction model to enhance the assessment of in-hospital mortality risk in VAP patients.<h4>Methods</h4>This study extracted VAP patient data from versions 2.2 and 3.1 of the MIMIC-IV database, using version 2.2 for model training and validation, and version 3.1 for external testing. Feature selection was conducted using the Boruta algorithm, and 14 ML models were constructed. The optimal model was identified based on the area under the receiver operating characteristic curve (AUROC), accuracy, sensitivity, and specificity across both validation and test cohorts. SHapley Additive exPlanations (SHAP) analysis was applied for global and local interpretability.<h4>Results</h4>A total of 1,894 VAP patients were included, with 12 features ultimately selected for model construction: 24-hour urine output, blood urea nitrogen, age, diastolic blood pressure, platelet count, anion gap, body temperature, bicarbonate level, sodium level, body mass index, and whether combined with congestive heart failure and cerebrovascular disease. The random forest (RF) model showed the best performance, achieving an AUC of 0.780 in internal validation and 0.724 in external testing, outperforming other ML models and common clinical scoring systems.<h4>Conclusion</h4>The RF model demonstrated robust and reliable performance in predicting in-hospital mortality risk for VAP patients. The developed online tool can assist clinicians in efficiently assessing VAP in-hospital mortality risk, supporting clinical decision-making.https://doi.org/10.1371/journal.pone.0316526 |
| spellingShingle | Junying Wei Heshan Cao Mingling Peng Yinzhou Zhang Sibei Li Wuhua Ma Yuhui Li An interpretable machine learning model for predicting in-hospital mortality in ICU patients with ventilator-associated pneumonia. PLoS ONE |
| title | An interpretable machine learning model for predicting in-hospital mortality in ICU patients with ventilator-associated pneumonia. |
| title_full | An interpretable machine learning model for predicting in-hospital mortality in ICU patients with ventilator-associated pneumonia. |
| title_fullStr | An interpretable machine learning model for predicting in-hospital mortality in ICU patients with ventilator-associated pneumonia. |
| title_full_unstemmed | An interpretable machine learning model for predicting in-hospital mortality in ICU patients with ventilator-associated pneumonia. |
| title_short | An interpretable machine learning model for predicting in-hospital mortality in ICU patients with ventilator-associated pneumonia. |
| title_sort | interpretable machine learning model for predicting in hospital mortality in icu patients with ventilator associated pneumonia |
| url | https://doi.org/10.1371/journal.pone.0316526 |
| work_keys_str_mv | AT junyingwei aninterpretablemachinelearningmodelforpredictinginhospitalmortalityinicupatientswithventilatorassociatedpneumonia AT heshancao aninterpretablemachinelearningmodelforpredictinginhospitalmortalityinicupatientswithventilatorassociatedpneumonia AT minglingpeng aninterpretablemachinelearningmodelforpredictinginhospitalmortalityinicupatientswithventilatorassociatedpneumonia AT yinzhouzhang aninterpretablemachinelearningmodelforpredictinginhospitalmortalityinicupatientswithventilatorassociatedpneumonia AT sibeili aninterpretablemachinelearningmodelforpredictinginhospitalmortalityinicupatientswithventilatorassociatedpneumonia AT wuhuama aninterpretablemachinelearningmodelforpredictinginhospitalmortalityinicupatientswithventilatorassociatedpneumonia AT yuhuili aninterpretablemachinelearningmodelforpredictinginhospitalmortalityinicupatientswithventilatorassociatedpneumonia AT junyingwei interpretablemachinelearningmodelforpredictinginhospitalmortalityinicupatientswithventilatorassociatedpneumonia AT heshancao interpretablemachinelearningmodelforpredictinginhospitalmortalityinicupatientswithventilatorassociatedpneumonia AT minglingpeng interpretablemachinelearningmodelforpredictinginhospitalmortalityinicupatientswithventilatorassociatedpneumonia AT yinzhouzhang interpretablemachinelearningmodelforpredictinginhospitalmortalityinicupatientswithventilatorassociatedpneumonia AT sibeili interpretablemachinelearningmodelforpredictinginhospitalmortalityinicupatientswithventilatorassociatedpneumonia AT wuhuama interpretablemachinelearningmodelforpredictinginhospitalmortalityinicupatientswithventilatorassociatedpneumonia AT yuhuili interpretablemachinelearningmodelforpredictinginhospitalmortalityinicupatientswithventilatorassociatedpneumonia |