arXiv Machine Learning

Graph-Enhanced Disease Trajectory Discovery and Temporal Endpoint Forecasting from Longitudinal Electronic Health Records

The study introduces a graph‑enhanced framework for discovering disease trajectories and forecasting temporal endpoints using longitudinal EHR data from MIMIC‑IV v3.1. It constructs patient‑specific motifs from consecutive admissions, applies a time‑aware Transformer to select up to 24 motifs per endpoint, and reports modest improvements in micro‑AUPRC and event‑time MAE, with larger gains for events occurring after 730 days. The authors also explore branching‑motif conditioning and GNN‑based edge repair, finding that motifs mainly provide long‑range timing support rather than superior endpoint discrimination or survival prediction.

arXiv AI
3d ago

GraphBEAT: Graph-based Bucketized Event Aggregation Over Time for Early Risk Stratification of Gastrointestinal Cancers

GraphBEAT is a temporal graph representation learning framework that bucketizes clinical events from longitudinal electronic health records into time‑ordered graphs, enabling early risk stratification for gastrointestinal cancers. It uses a time‑decay message‑passing mechanism and multi‑head attention pooling to generate patient‑level representations, and was trained on multi‑site diagnosis codes and laboratory tests to predict pancreatic, liver, and intestinal cancer at 12, 24, and 36‑month lead times. Across all experiments, GraphBEAT outperformed baseline graph neural networks and sequence models while requiring far fewer floating‑point operations and parameters.

By Leo Y. Li-Han, Chris Varghese, Elizabeth B. Habermann, Mark J. Truty, William R. G. Perry, Cornelius A. Thiels, Hojjat Salehinejad
arXiv AI
Sep 10

PGP-Clinical-TimeKAN: Prior-Guided Joint Probabilistic Forecasting of Clinical Trajectories

PGP-Clinical-TimeKAN is a trajectory-first framework for joint probabilistic forecasting of multivariate physiological data, combining missingness-aware temporal encoders, a soft organ-system prior, patient-specific relations, nonlinear Kolmogorov‑Arnold messages, and a low‑rank multivariate Student‑t head. Evaluated on a MIMIC‑IV cohort of 6,882 patients, it achieves the second‑lowest normalized MAE and the lowest RMSE among 13 models, while providing calibrated probabilistic forecasts with empirical coverage at 50%, 80%, and 95% intervals. Ablation studies show that relational structure is critical for performance, and increasing covariance rank improves likelihood but not point accuracy. whyItMatters":"The model demonstrates that joint trajectory forecasting can yield highly accurate, calibrated predictions of physiological trajectories, offering a potentially inspectable intermediate task for clinical deterioration prediction."

By Weizhi Nie, Rihao Chang, Weijie Wang, Yuting Su
arXiv Machine Learning
Jul 21

Differentiable latent structure discovery for interpretable forecasting in clinical time series

arXiv:2604. 27967v2 Announce Type: replace Abstract: Background: We introduce StructGP, a continuous-time multi-task Gaussian process that couples process convolutions with differentiable structure learning to uncover a sparse, ordered directed acyclic graph (DAG) of inter-variable dependencies while preserving principled uncertainty.

By Ivan Lerner, Jean Feydy, Alexandre Kalimouttou, Anita Burgun, Francis Bach
arXiv Machine Learning
Aug 4

EHR2Path: Comprehensive Pathway-Level Modeling of Longitudinal Patient Trajectories from Multimodal Electronic Health Records

arXiv:2506. 04831v3 Announce Type: replace Abstract: Forecasting how a patient's condition is likely to evolve, including possible deterioration, recovery, treatment needs, and care transitions, could support more proactive and personalized care, but requires modeling heterogeneous and longitudinal electronic health record (EHR) data.

By Chantal Pellegrini, Ege \"Ozsoy, David Bani-Harouni, Matthias Keicher, Nassir Navab
arXiv Machine Learning
Jun 8

One Loss to Rule Them All: Marked Time-to-Event for Structured EHR Foundation Models

arXiv:2602. 00541v2 Announce Type: replace Abstract: Clinical events captured in Electronic Health Records (EHR) are irregularly sampled and may consist of a mixture of discrete events and numerical measurements, such as laboratory values or treatment dosages.

By Zilin Jing, Vincent Jeanselme, Yuta Kobayashi, Simon A. Lee, Chao Pang, Aparajita Kashyap, Yanwei Li, Xinzhuo Jiang, Shalmali Joshi
arXiv AI
Sep 1

INTERVenE: Temporal-Abstraction-Interval Based Transformers for Short-Horizon Medical Event Prediction

INTERVenE introduces Transformer models that use a knowledge‑based temporal abstraction (KBTA) token stream of named clinical concepts instead of raw measurements, enabling per‑token attributions to resolve directly to clinical concepts. Two variants are offered: an auto‑regressive decoder that generates future abstraction trajectories with step‑wise risk readouts, and a bidirectional encoder that jointly predicts risk and time‑to‑event in a single pass. On 57,078 MIMIC‑IV admissions, the encoder variant outperforms neural baselines with a support‑weighted AUPRC of 0.672 and AUROC of 0.901, while the decoder provides complementary token‑level risk trajectories.

By Shahar Oded, Yuval Shahar
arXiv AI
Jul 22

Trajectory-Aware Clinical Risk Prediction via Severity-Grounded Knowledge Graphs and Retrieval-Augmented Generation

arXiv:2607. 18270v1 Announce Type: new Abstract: While Electronic Health Records (EHRs) offer a wealth of clinical data, effectively augmenting a patient's records with heterogeneous external knowledge to predict the patient's clinical risk remains a significant challenge.

By Kyunghoon Jeon, Youmin Ko, Woohwan Jung, Hyunjoon Kim