AI-powered model for predicting mortality risk in VA-ECMO patients: a multicenter cohort study

Veno-arterial extracorporeal membrane oxygenation (VA-ECMO) is a critical life support technology for severely ill patients. Despite its benefits, patients face high costs and significant mortality risks. To improve clinical decision-making, this study aims to develop a non-invasive, efficient artificial intelligence (AI)-enabled model to predict the risk of mortality within 28 days post-weaning from VA-ECMO. A multicenter, retrospective cohort study was conducted across five hospitals in China, including all the patients who received VA-ECMO support between January 2020 and January 2024. Based on the innovatively selected 25 easily obtainable patient examination features as potentially relevant, this study involved developing ten predictive models using both classical and advanced machine learning techniques. The model's performance is evaluated using various statistical metrics and the optimal predictive model are identified. Feature correlations are analyzed using Pearson correlation coefficients, and SHapley Additive exPlanations (SHAP) are employed to interpret feature importance. Decision curve analysis is used to evaluate the clinical utility of the predictive models. The study included 225 patients, with 66 patients from one hospital forming the training cohort. Three validation cohorts were used: internal validation with 16 patients from the training hospital and external validation with 30 and 60 patients from the other 4 hospitals. The random forest model emerged as the best predictor of 28-day mortality, achieving an AUROC of 1.00 in the training cohort and 1.00, 0.97, and 0.93 in the three validation cohorts, respectively. Despite the limited training data, the developed model, eCMoML, demonstrated high accuracy, generalizability and reliability. The model will be available online for immediate use by clinicians. The eCMoML model, validated in a multicenter cohort study, offers a rapid, stable, and accurate tool for predicting 28-day mortality post-VA-ECMO weaning. It has the potential to significantly enhance clinical decision-making, helping doctors better assess patient prognosis, optimize treatment plans, and improve survival rates.