Prediction of intensive care unit admission in hospitalized patients with community-acquired pneumonia: An eXtreme gradient boosting model

  • Lê Đức Duẩn Bệnh viện Trung ương Quân đội 108
  • Nguyễn Hải Ghi Bệnh viện Trung ương Quân đội 108
  • Nguyễn Thái Cường Bệnh viện Trung ương Quân đội 108
  • Ngô Chí Công Bệnh viện Trung ương Quân đội 108
  • Đậu Xuân Thành Bệnh viện Trung ương Quân đội 108
  • Dương Hữu Bắc Bệnh viện Trung ương Quân đội 108
  • Nguyễn Khắc Hùng Bệnh viện Trung ương Quân đội 108
  • Đỗ Thanh Hòa Bệnh viện Trung ương Quân đội 108

Main Article Content

Keywords

Community-Acquired Pneumonia, ICU admission prediction, XGBoost model.

Abstract

Objective: To evaluate the performance of the machine learning model eXtreme Gradient Boosting (XGBoost) in predicting the need for Intensive Care Unit admission in hospitalized patients with Community-Acquired Pneumonia. Subject and method: A retrospective, descriptive study conducted on Community-Acquired Pneumonia patients ≥ 18 years old in the Emergency Department, 108 Military Central Hospital, from 2021 to 2022. Clinical and paraclinical data were collected within the first 24 hours of admission. The XGBoost model was trained on 70% of the data and tested on the remaining 30%. The final model utilized 6 predictors with the highest Gain scores. Result: A total of the 350 patients studied, 75 (21.4%) required ICU admission. The six most important preditors (in descending order of importance) were: impaired mental status, elevated blood urea, thrombocytopenia, tachypnea, multilobar infiltrates on X-ray, and neutrophilia. The XGBoost model had a higher predictive performance compared to CURB-65 score, with AUCs of 0.876 and 0.826, respectively. Decision Curve Analysis confirmed that the XGBoost model provided maximum and stable net benefit over a wide range of threshold probabilities (20% to 80%). Conclusion: The XGBoost model, using 6 key clinical and paraclinical variables, significantly outperformed the CURB-65 score in predicting the need for ICU admission in hospitalized CAP patients. This simple model is a good tool to stratify the risk among patients with community-acquired pneumonia presenting to the Emergency Department.

Article Details

References

1. Troeger C, Blacker B, Khalil IA et al (2018) Estimates of the global, regional, and national morbidity, mortality, and aetiologies of lower respiratory infections in 195 countries, 1990–2016: a systematic analysis for the Global Burden of Disease Study 2016. The Lancet infectious diseases 18(11): 1191-1210.
2. Viasus D, Cillóniz C, Cardozo CG, Puerta P, Garavito A, Torres A (2018) Early, short and long-term mortality in community-acquired pneumonia. Annals of Research Hospitals: 2.
3. Ilg A, Moskowitz A, Konanki V et al (2019) Performance of the CURB-65 Score in Predicting Critical Care Interventions in Patients Admitted With Community-Acquired Pneumonia. Ann Emerg Med. 74(1): 60-68. doi:10.1016/j.annemergmed.2018.06.017
4. Montomoli J, Romeo L, Moccia S et al (2021) Machine learning using the extreme gradient boosting (XGBoost) algorithm predicts 5-day delta of SOFA score at ICU admission in COVID-19 patients. J Intensive Med 1(2): 110-116. doi:10.1016/j.jointm.2021.09.002
5. Wiens M, Verone-Boyle A, Henscheid N, Podichetty JT, Burton J (2025) A Tutorial and Use Case Example of the eXtreme Gradient Boosting (XGBoost) Artificial Intelligence Algorithm for Drug Development Applications. Clin Transl Sci 18(3):70172. doi:10.1111/cts.70172
6. Xu Z, Guo K, Chu W, Lou J, Chen C (2022) Performance of Machine Learning Algorithms for Predicting Adverse Outcomes in Community-Acquired Pneumonia. Front Bioeng Biotechnol 10: 903426. doi:10.3389/fbioe.2022.903426
7. Lương Ngọc Khuê, Ngô Quý Châu, Nguyễn Viết Nhung, eds (2020) Hướng dẫn chẩn đoán và điều trị viêm phổi mắc phải tại cộng đồng ở người lớn. Bộ Y tế; 2020.
8. Huang D, Gong L, Wei C, Wang X, Liang Z (2024) An explainable machine learning-based model to predict intensive care unit admission among patients with community-acquired pneumonia and connective tissue disease. Respir Res 25(1): 246. doi:10.1186/s12931-024-02874-3
9. Odeyemi YE, Lal A, Barreto EF et al (2024) Early machine learning prediction of hospitalized patients at low risk of respiratory deterioration or mortality in community-acquired pneumonia: Derivation and validation of a multivariable model. Biomol Biomed 24(2): 337-345. doi:10.17305/bb.2023.9754
10. Samaha HMS, Elsaid AR (2015) Platelet count and level of paCO2 are predictors of CAP prognosis. Egyptian Journal of Chest Diseases and Tuberculosis 64(2): 453-458. doi:https://doi.org/10.1016/j.ejcdt.2015.02.002
11. Niederman MS, Mandell LA, Anzueto A et al (2001) Guidelines for the management of adults with community-acquired pneumonia: diagnosis, assessment of severity, antimicrobial therapy, and prevention. American journal of respiratory and critical care medicine 163(7): 1730-1754.