Development of a Machine Learning Model for Predicting 28-Day Mortality of Septic Patients With Atrial Fibrillation

ABSTRACT: Introduction: Septic patients with atrial fibrillation (AF) are common in the intensive care unit accompanied by high mortality. The early prediction of prognosis of these patients is critical for clinical intervention. This study aimed to develop a model by using machine learning (ML) algorithms to predict the risk of 28-day mortality in septic patients with AF. Methods: In this retrospective cohort study, we extracted septic patients with AF from the Medical Information Mart for Intensive Care III (MIMIC-III) and IV database. Afterward, only MIMIC-IV cohort was randomly divided into training or internal validation set. External validation set was mainly extracted from MIMIC-III database. Propensity score matching was used to reduce the imbalance between the external validation and internal validation data sets. The predictive factors for 28-day mortality were determined by using multivariate logistic regression. Then, we constructed models by using ML algorithms. Multiple metrics were used for evaluation of performance of the models, including the area under the receiver operating characteristic curve, sensitivity, specificity, recall, and accuracy. Results: A total of 5,317 septic patients with AF were enrolled, with 3,845 in the training set, 960 in the internal testing set, and 512 in the external testing set, respectively. Then, we established four prediction models by using ML algorithms. AdaBoost showed moderate performance and had a higher accuracy than the other three models. Compared with other severity scores, the AdaBoost obtained more net benefit. Conclusion: We established the first ML model for predicting the 28-day mortality of septic patients with AF. Compared with conventional scoring systems, the AdaBoost model performed moderately. The model established will have the potential to improve the level of clinical practice.