Comparing Machine Learning Approaches for Predicting CFD-Derived Stroke Risk Indicators in Atrial Fibrillation Patients
Journal:
bioRxiv
Published Date:
Jun 8, 2026
Abstract
Non-valvular atrial fibrillation (AF) is associated with a five-fold increased risk of stroke, mainly due to impaired contractility of the left atrium (LA) leading to blood stasis and subsequent thrombus formation within the left atrial appendage (LAA). Current AF stroke risk stratification schemes, such as the CHA2DS2-VASc/ CHA2DS2-VA score, use comorbidities and do not capture mechanistic factors like blood flow dynamics and hypercoagulability. To address this, we developed a multiphase computational fluid dynamics (CFD) model of the LA, incorporating patient-specific geometries; modelling of the coagulation cascade; and non-Newtonian blood behaviour within the LAA. Using 84 simulation cases generated via Latin Hypercube Sampling of physiological blood parameters and 21 patient-derived LA anatomies, we trained surrogate machine learning models, including Ridge regression, XGBoost, Gaussian Process Emulators (GPEs), and deep learning networks, to predict CFD outputs such as blood viscosity in and fibrin concentrations in the LAA. Deep learning achieved R{superscript 2} values up to 0.90, with the accuracy increasing when both physiological parameters and the raw CT image were included. Other models showed uneven performance with R2 values below 0.7, highlighting the role of nonlinearities between parameters. The study presents a novel CFD model that captures the transition from blood stasis to clot formation, representing the full thrombotic continuum underlying stroke risk in AF, and a deep learning approach to enable efficient prediction of mechanistic outputs of clinical value for stroke risk stratification in AF patients.
