arXiv AI By Yuen Chen, Yulun Wu, Samuel Sharpe, Igor Melnyk, Nam H. Nguyen, Furong Huang, C. Bayan Bruss, Rizal Fathony

Bridging the Divide: End-to-End Sequence-Graph Learning

Read the original on arXiv AI →

The paper "Bridging the Divide: End-to-End Sequence-Graph Learning" introduces BRIDGE, an architecture that jointly learns from sequential and relational data by coupling a sequence model with a graph module under a single objective. It adds a token-level cross-attention layer, TOKENXATTN, to enable fine-grained message passing between events in neighboring sequences. Experiments on relationship prediction and fraud detection show that BRIDGE outperforms static graph models, temporal graph methods, and sequence-only baselines on ranking and classification metrics.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv AI.

arXiv AI
Sep 7

REFINE: LLM Refinement over Budgeted Text-Attributed Graphs for Personalized Medical Concept Representation

REFINE is a framework that refines medical concept representations by creating patient‑specific temporal graphs from a global text‑attributed knowledge graph. It uses a reinforcement learning policy to allocate a personalized graph expansion budget for each observed code, then processes the resulting graph with a heterogeneous GNN and a frozen LLM that refines representations via graph‑aware soft prompts. Experiments on MIMIC‑III and MIMIC‑IV demonstrate that REFINE consistently improves various EHR prediction backbones, surpasses strong baselines, and shows robust gains across ablation studies, KG selection, and data insufficiency scenarios.

By Mohsen Nayebi Kerdabadi, Arya Hadizadeh Moghaddam, Dongjie Wang, Zijun Yao