Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
GraphBEAT addresses the challenges of modeling irregular, sparse, and heterogeneous longitudinal EHR data for early cancer risk stratification by constructing patient-specific temporal graphs. These graphs encode clinical events within time buckets, utilizing a context-aware, time-decay message-passing mechanism and multi-head attention pooling to generate patient representations. GraphBEAT significantly outperforms existing graph and sequence models in predicting gastrointestinal cancers across various lead times while demonstrating substantial computational efficiency.
A novel temporal graph framework achieves over 250x fewer FLOPs and 20x fewer parameters than Transformers/LSTMs for early gastrointestinal cancer risk stratification from sparse EHRs.
Longitudinal electronic health records (EHRs) contain complementary diagnostic and laboratory information about patient health. However, irregularity, sparsity, heterogeneity, and incompleteness in documentation complicate EHR modeling. To address the problem, we introduce GraphBEAT (Graph-based Bucketized Event Aggregation over Time), a temporal graph representation learning framework that groups clinical events into time buckets and transforms observed features into patient-specific, temporally directed graphs to enable early risk stratification for gastrointestinal cancers. Each node in an EHR graph encodes feature identity, observed value, and temporal position of clinical events. A context-aware, time-decay message-passing mechanism models transitions between successive buckets, followed by multi-head attention pooling to generate patient-level representations. GraphBEAT was trained and evaluated on multi-site longitudinal diagnosis codes and laboratory tests to predict pancreatic, liver, and intestinal cancer at 12-, 24-, and 36-month diagnostic lead times. In all experimental setups, GraphBEAT consistently outperformed baseline graph neural networks (GraphSAGE, GAT, and GCN) and sequence modeling approaches (Transformer and long short-term memory [LSTM]). GraphBEAT also required over 250x fewer floating-point operations per patient and over 20x fewer parameters than the Transformer and LSTM baselines. These results support GraphBEAT as an accurate and computationally efficient approach for cancer risk stratification from sparse, heterogeneous longitudinal health records.