arXiv · 2508.20500
Structure-aware Hypergraph Transformer for Diagnosis Prediction in Electronic Health Records
Abstract
Electronic Health Records (EHR) systematically organize patient health data through standardized medical codes, serving as a comprehensive and invaluable source for predictive modeling. Graph neural networks (GNNs) have demonstrated effectiveness in modeling interactions between medical codes within EHR. However, existing GNN-based methods are inadequate due to: a) their reliance on pairwise relations fails to capture the inherent higher-order dependencies in clinical data, and b) the localized message-passing scheme limits representation power. To address these issues, this paper proposes a novel Structure-aware HyperGraph Transformer (SHGT) framework following three-fold ideas: a) employing a hypergraph structural encoder to capture higher-order interactions among medical codes, b) integrating the Transformer architecture to reason over the entire hypergraph, and c) designing a tailored loss function incorporating hypergraph reconstruction to preserve the hypergraph's original structure. Experiments on real-world EHR datasets demonstrate that the proposed SHGT outperforms existing state-of-the-art models on diagnosis prediction.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Haiyan Wang, Ye Yuan. 2025-08-28. Structure-aware Hypergraph Transformer for Diagnosis Prediction in Electronic Health Records. https://arxiv.org/abs/2508.20500
Cite the original work for its findings. Save a collection to share your selection of sources.