# Enhanced Atrial Fibrillation Prediction in ESUS Patients with Hypergraph-based Pre-training

arXiv: https://arxiv.org/abs/2603.13297 · HTML: https://mlsanity.com/paper/2603.13297/

## Key facts

- No citing papers in the map yet (data through 14 Oct 2026).
- Builds on 9 of its 34 references that are in the map.
- First posted on arXiv 2 Mar 2026.

## Builds on (most cited first, 9 of 9)

- [Categorical Reparameterization with Gumbel-Softmax](https://mlsanity.com/paper/1611.01144/) (2016)
- [I'm Me, We're Us, and I'm Us: Tri-directional Contrastive Learning on Hypergraphs](https://mlsanity.com/paper/2206.04739/) (2022)
- [PromptLink: Leveraging Large Language Models for Cross-Source Biomedical Concept Linking](https://mlsanity.com/paper/2405.07500/) (2024)
- [KERAP: A Knowledge-Enhanced Reasoning Approach for Accurate Zero-shot Diagnosis Prediction Using Multi-agent LLMs](https://mlsanity.com/paper/2507.02773/) (2025)
- [HypKG: Hypergraph-based Knowledge Graph Contextualization for Precision Healthcare](https://mlsanity.com/paper/2507.19726/) (2025)
- [TACCO: Task-guided Co-clustering of Clinical Concepts and Patient Visits for Disease Subtyping based on EHR Data](https://mlsanity.com/paper/2406.10061/) (2024)
- [Utilizing Large Language Models for Zero-Shot Medical Ontology Extension from Clinical Notes](https://mlsanity.com/paper/2511.16548/) (2025)
- [CHGNN: A Semi-Supervised Contrastive Hypergraph Learning Network](https://mlsanity.com/paper/2303.06213/) (2023)
- [Towards Automatic Evaluation and Selection of PHI De-identification Models via Multi-Agent Collaboration](https://mlsanity.com/paper/2510.16194/) (2025)

## Abstract (arXiv)

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.