Machine learning for outcome prediction in patients with non-valvular atrial fibrillation from the GLORIA-AF registry

Abstract

Abstract Clinical risk scores that predict outcomes in patients with atrial fibrillation (AF) have modest predictive value. Machine learning (ML) may achieve greater results when predicting adverse outcomes in patients with recently diagnosed AF. Several ML models were tested and compared with current clinical risk scores on a cohort of 26,183 patients (mean age 70.13 (standard deviation 10.13); 44.8% female) with non-valvular AF. Inputted into the ML models were 23 demographic variables alongside comorbidities and current treatments. For one-year stroke prediction, ML achieved an area under the curve (AUC) of 0.653 (95% confidence interval 0.576–0.730), compared to the CHADS2 and CHA2DS2-VASc scores performance of 0.587 (95% CI 0.559–0.615) and 0.535 (95% CI 0.521–0.550), respectively. Using ML for one-year major bleed prediction increased the AUC from 0.537 (95% CI 0.518–0.557) generated by the HAS-BLED score to 0.677 (95% CI 0.619–0.724). ML was able to predict one-year and three-year all-cause mortality with an AUC of 0.734 (95% CI 0.696–0.771) and 0.742 (95% CI 0.718–0.766). In this study a significant improvement in performance was observed when transitioning from clinical risk scores to machine learning-based approaches across all applications tested. Obtaining precise prediction tools is desirable for increased interventions to reduce event rates. Trial Registry https://www.clinicaltrials.gov ; Unique identifier: NCT01468701, NCT01671007, NCT01937377.