Prediction of early death after atrial fibrillation diagnosis using a machine learning approach: A French nationwide cohort study

Aims Atrial fibrillation is associated with important mortality but the usual clinical risk factor based scores only modestly predict mortality. This study aimed to develop machine learning models for the prediction of death occurrence within the year following atrial fibrillation diagnosis and compare predictive ability against usual clinical risk scores. Methods and Results We used a nationwide cohort of 2,435,541 newly diagnosed atrial fibrillation patients seen in French hospitals from 2011 to 2019. Three machine learning models were trained to predict mortality within the first year using a training set (70% of the cohort). The best model was selected to be evaluate and compared with previously published scores on the validation set (30% of the cohort). Discrimination of the best model was evaluated using the C index. Within the first year following atrial fibrillation diagnosis, 342,005 patients (14.4%) died after a period of 83 (SD 98) days (median 37 [10-129]). The best machine learning model selected was a deep neural network with a C index of 0.785 (95% CI, 0.781-0.789) on the validation set. Compared to clinical risk scores, the selected model was superior to the CHA2DS2-VASc and HAS-BLED risk scores and superior to dedicated scores such as Charlson Comorbidity Index and Hospital Frailty Risk Score to predict death within the year following atrial fibrillation diagnosis (C indexes: 0.597; 0.562; 0.643; 0.626 respectively. P<0.0001). Conclusion Machine learning algorithms predict early death after atrial fibrillation diagnosis and may help clinicians to better risk stratify atrial fibrillation patients at high risk of mortality.