Construction and validation of a machine learning-based nomogram model for predicting pneumonia risk in patients with
Yi-Chao Wang1, Qian He1, Yue-Jing Wu1
1Affiliated Mental Health Center & Hangzhou Seventh People's Hospital, Zhejiang University School of Medicine, Hangzhou, Zhejiang, China.
Objective:
Catatonia was often complicated by pneumonia, and the development of severe pneumonia after admission posed significant challenges to its treatment. This study aimed to develop a Nomogram Model based on pre-admission characteristics of patients with catatonia to predict the risk of pneumonia after admission.
Methods:
This retrospective observational study reviewed catatonia patients hospitalized at Hangzhou Seventh People's Hospital from September 2019 to November 2024. Data included demographic characteristics, medical history, maintenance medications, and pre-admission clinical presentations. Patients were divided into catatonia with and without pneumonia groups. The LASSO Algorithm was used for feature selection, and seven machine learning models: Decision Tree(DT), Logistic Regression(LR), Naive Bayes(NB), Random Forest(RF), K Nearest Neighbors(KNN), Gradient Boosting Machine(GBM), Support Vector Machine(SVM) were trained. Model performance was evaluated using AUC, Accuracy, Sensitivity, Specificity, Positive Predictive Value, Negative Predictive Value, F1 Score, Cohen's Kappa, and Brier Score, and Brier score. The best-performing model was selected for multivariable analysis to determine the variables included in the final Nomogram Model. The Nomogram Model was further validated through ROC Curves, Calibration Curves, Decision Curve Analysis (DCA), and Bootstrapping to ensure discrimination, calibration, and clinical applicability.
Results:
Among 156 patients, 79 had no pneumonia, and 77 had pneumonia. LASSO Algorithm identified 15 non-zero coefficient variables (LASSO 1-SEλ=0.076). The GBM showed the best performance (AUC = 0.954, 95% CI: 0.924-0.983, vs other models by DeLong's test: P < 0.05). Five key variables: Age, Clozapine, Diaphoresis, Intake Refusal, and Waxy Flexibility were used to construct the Nomogram Model. Validation showed good discrimination (AUC = 0.803, 95% CI: 0.735-0.870), calibration, and clinical applicability. Internal validation (Bootstrapping, n=500) confirmed model stability (AUC = 0.814, 95% CI: 0.743-0.878; Hosmer-Lemeshow P = 0.525).
Conclusion:
This study developed a Nomogram Model based on five key factors, demonstrating significant clinical value in predicting the risk of pneumonia in hospitalized patients with catatonia.