Graph neural networks for the enhanced prediction of new atrial fibrillation episodes after stroke

Rivera-Juzga, CY, Mistry, D ORCID: 0000-0001-8300-109X, Knowles, AT ORCID: 0000-0003-1795-9894, Lip, GYH ORCID: 0000-0002-7566-1626, Ortega-Martorell, S ORCID: 0000-0001-9927-3209 and Olier, I ORCID: 0000-0002-5679-7501 (2025) Graph neural networks for the enhanced prediction of new atrial fibrillation episodes after stroke. Computers in Biology and Medicine, 197 (Pt B). ISSN 0010-4825

Preview

Text Graph neural networks for the enhanced prediction of new atrial fibrillation episodes after stroke.pdf - Published Version Available under License Creative Commons Attribution. Download (4MB) | Preview

Abstract

Traditionally, statistical and machine learning (ML) algorithms have been used to develop risk prediction models for adverse clinical events, such as atrial fibrillation (AF) after stroke. However, these algorithms often fail to encapsulate or exploit possible connections between patients, assuming each patient is fully independent. This study builds a graph of patients interlinked by their medical histories to create a graph-based risk prediction model for AF. We investigate the ability of Graph Convolutional Networks (GCN) and Graph Attention Networks (GAT) to predict AF risk in critically ill stroke patients. We introduce a novel, patient-specific approach for computing similarities between GNN nodes and explore several methods for GNN explainability, including node-specific Shapley value analysis and node relationships based on the attention coefficients of the GAT model. Our findings show that GCN and GAT models, with AUCs of 0.81[0.78–0.84] and 0.84[0.81–0.87], consistently outperform traditional algorithms such as Random Forest, XGBoost, and Logistic Regression, which had the best AUC of 0.78 [0.74–0.82]. This superior performance is observed when our proposed custom similarity metric is used to construct the graph, highlighting the importance of task-specific graph design in enhancing model effectiveness. The attention mechanisms in GAT models likely contributed to this improved performance. This study highlights the strength of GNNs in capturing complex relationships and provides insights into model predictions, demonstrating the generalisability of our methodological approach to other risk prediction models.

Item Type:	Article
Uncontrolled Keywords:	Atrial fibrillation; Clinical decision support; Graph neural networks; Intensive care unit; Machine learning; Risk prediction; Stroke; 08 Information and Computing Sciences; 09 Engineering; 11 Medical and Health Sciences; Biomedical Engineering; 3102 Bioinformatics and computational biology; 4203 Health services and systems; 4601 Applied computing
Subjects:	Q Science > QA Mathematics > QA75 Electronic computers. Computer science
Divisions:	Computer Science and Mathematics
Publisher:	Elsevier
Date of acceptance:	12 September 2025
Date of first compliant Open Access:	24 September 2025
Date Deposited:	24 Sep 2025 11:21
Last Modified:	24 Sep 2025 11:30
DOI or ID number:	10.1016/j.compbiomed.2025.111095
URI:	https://researchonline.ljmu.ac.uk/id/eprint/27199

View Item