| 英文摘要 |
In the intensive care unit (ICU), accurate prediction of clinical outcomes among patients is crucial for improving patient prognosis and the efficiency of medical decision-making. However, diverse and imbalanced ICU data create challenges in the selection and performance improvement of suitable artificial intelligence (AI) models. In this study, the predictive performance of various AI models was compared across surgical, neurological, and medical ICU department data. Additionally, the effects of data balancing strategies on model performance were evaluated to provide practical guidance for clinical applications. The performance of support vector machines, logistic regression models, K-nearest algorithms (KNN), Gaussian Naive Bayes, multilayer perceptron neural networks (MLP), and Extreme Gradient Boosting (XGBoost) was evaluated. XGBoost was identified as the best choice for cross-department predictions, demonstrating the most stable performance across all ICU departments, with an area under the curve (AUC) of 0.82. Gaussian Naive Bayes performed the worst in capturing ICU data features, with an AUC ranging from 0.63 to 0.64. The MLP model performed better with neurological and medical ICU data than with surgical ICU data, with an AUC of 0.76. Logistic regression and KNN models also performed better with neurological and medical ICU data (AUC = 0.74) than with surgical ICU data, highlighting differences in adaptability to specific departmental data between the models. To further improve model performance, the effects of data balancing strategies (original data, oversampling, and undersampling) on predictive accuracy were evaluated with the XGBoost model. Although general overfitting was observed in the original dataset, both oversampling and undersampling strategies considerably improved the model’s predictive ability. Oversampling effectively balanced class distribution and was suitable for general application. Undersampling performed slightly better with medical and neurological ICU data, demonstrating its potential to enhance predictions for imbalanced datasets. The results of this study confirm the stability and superiority of XGBoost in crossdepartmental ICU data predictions, underscore the role of data balancing strategies in improving model performance, and provide practical evidence for AI application in ICU settings. |