Prediction of Postoperative Pulmonary Edema Risk Using Machine Learning

Abstract

Postoperative pulmonary edema (PPE) is a well-known postoperative complication. We hypothesized that a machine learning model could predict PPE risk using pre- and intraoperative data, thereby improving postoperative management. This retrospective study analyzed the medical records of patients aged > 18 years who underwent surgery between January 2011 and November 2021 at five South Korean hospitals. Data from four hospitals (n = 221,908) were used as the training dataset, whereas data from the remaining hospital (n = 34,991) were used as the test dataset. The machine learning algorithms used were extreme gradient boosting, light-gradient boosting machine, multilayer perceptron, logistic regression, and balanced random forest (BRF). The prediction abilities of the machine learning models were assessed using the area under the receiver operating characteristic curve, feature importance, and average precisions of precision-recall curve, precision, recall, f1 score, and accuracy. PPE occurred in 3584 (1.6%) and 1896 (5.4%) patients in the training and test sets, respectively. The BRF model exhibited the best performance (area under the receiver operating characteristic curve: 0.91, 95% confidence interval: 0.84–0.98). However, its precision and f1 score metrics were not good. The five major features included arterial line monitoring, American Society of Anesthesiologists physical status, urine output, age, and Foley catheter status. Machine learning models (e.g., BRF) could predict PPE risk and improve clinical decision-making, thereby enhancing postoperative management.