Predicting atrial fibrillation in patients with acute respiratory failure using machine learning: application of the MIMIC-III and MIMIC-IV datasets

TL;DR

This study uses machine learning to predict atrial fibrillation risk in patients with acute respiratory failure, using data from MIMIC-III and MIMIC-IV databases.

Contribution

The study introduces the first predictive model for atrial fibrillation in patients with acute respiratory failure.

Findings

Age was identified as the most important predictor variable for atrial fibrillation.

XGBoost and Random Forest models achieved AUC scores of 0.816 and 0.822, respectively, showing strong predictive performance.

Abstract

Acute respiratory failure (ARF) and atrial fibrillation (AF) are common diseases. This study established a predictive model for the risk of atrial fibrillation in patients with ARF, aiming to provide tools for clinical application.

This study examined the data of 21,594 patients in the MIMIC-IV database, including factors such as age, vital signs, and laboratory results on the first day of admission. Six feature selection techniques and six machine learning algorithms were used to construct the prediction model, and then the prediction model was verified using the MIMIC-III database. Evaluate the performance of the model through the comparison of results.

A total of 59 predictor variables were identified, among which age was the most important factor. These variables are used to establish predictive models. The verification results show that the XGBoost model (AUC: 0.816) and the Random Forest (RF) model (AUC: 0.822) have the best performance. This study presents the first predictive model for atrial fibrillation in patients with acute respiratory failure.

Both the XGBoost and RF models demonstrated outstanding performance. These findings will make significant contributions to the diagnosis of clinical complications and the resolution of public health issues.

Figures