Enhanced Atrial Fibrillation Prediction in ESUS Patients with Hypergraph-based Pre-training
Fuente:
arXiv
Enregistré dans:
| Auteurs principaux: | , , , , , |
|---|---|
| Format: | Preprint |
| Publié: |
2026
|
| Sujets: | |
| Accès en ligne: | |
| Tags: |
Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
|
| _version_ | 1866912969138372608 |
|---|---|
| author | Xie, Yuzhang Wu, Yuhua Wang, Ruiyu Nahab, Fadi Hu, Xiao Yang, Carl |
| author_facet | Xie, Yuzhang Wu, Yuhua Wang, Ruiyu Nahab, Fadi Hu, Xiao Yang, Carl |
| contents | Atrial fibrillation (AF) is a major complication following embolic stroke of undetermined source (ESUS), elevating the risk of recurrent stroke and mortality. Early identification is clinically important, yet existing tools face limitations in accuracy, scalability, and cost. Machine learning (ML) offers promise but is hindered by small ESUS cohorts and high-dimensional medical features. To address these challenges, we introduce supervised and unsupervised hypergraph-based pre-training strategies to improve AF prediction in ESUS patients. We first pre-train hypergraph-based patient embedding models on a large stroke cohort (7,780 patients) to capture salient features and higher-order interactions. The resulting embeddings are transferred to a smaller ESUS cohort (510 patients), reducing feature dimensionality while preserving clinically meaningful information, enabling effective prediction with lightweight models. Experiments show that both pre-training approaches outperform traditional models trained on raw data, improving accuracy and robustness. This framework offers a scalable and efficient solution for AF risk prediction after stroke. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2603_13297 |
| institution | arXiv |
| publishDate | 2026 |
| record_format | arxiv |
| spellingShingle | Enhanced Atrial Fibrillation Prediction in ESUS Patients with Hypergraph-based Pre-training Xie, Yuzhang Wu, Yuhua Wang, Ruiyu Nahab, Fadi Hu, Xiao Yang, Carl Machine Learning Artificial Intelligence Atrial fibrillation (AF) is a major complication following embolic stroke of undetermined source (ESUS), elevating the risk of recurrent stroke and mortality. Early identification is clinically important, yet existing tools face limitations in accuracy, scalability, and cost. Machine learning (ML) offers promise but is hindered by small ESUS cohorts and high-dimensional medical features. To address these challenges, we introduce supervised and unsupervised hypergraph-based pre-training strategies to improve AF prediction in ESUS patients. We first pre-train hypergraph-based patient embedding models on a large stroke cohort (7,780 patients) to capture salient features and higher-order interactions. The resulting embeddings are transferred to a smaller ESUS cohort (510 patients), reducing feature dimensionality while preserving clinically meaningful information, enabling effective prediction with lightweight models. Experiments show that both pre-training approaches outperform traditional models trained on raw data, improving accuracy and robustness. This framework offers a scalable and efficient solution for AF risk prediction after stroke. |
| title | Enhanced Atrial Fibrillation Prediction in ESUS Patients with Hypergraph-based Pre-training |
| topic | Machine Learning Artificial Intelligence |
| url | https://arxiv.org/abs/2603.13297 |