arXiv AI

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

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.

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
arXiv Machine Learning
Sep 21

HERMES: Contrast-Aware Knowledge Graph Reasoning from Clinical Notes for Patient Outcome Prediction

HERMES is a graph-based framework that uses only clinical text to predict patient outcomes. It builds personalized Knowledge Graphs from clinical notes via Large‑Language‑Model‑guided extraction and Contrastive Logic Modeling, capturing temporal dynamics and treatment changes. A Graph Attention Network then synthesizes patient representations, and experiments on MIMIC‑III and MIMIC‑IV show HERMES outperforms text‑only baselines for in‑hospital mortality and 30‑day readmission prediction.

By Gia-Bach Nguyen, Hoang-Ha Nguyen, Tuan-Cuong Vuong, Trang Mai Xuan, Duy Quoc Ngo, Tien-Cuong Nguyen, Huan Vu, Thien Van Luong
arXiv AI
5d ago

Bridging Semantics and Structure: A Theoretically Grounded Framework for Link Prediction on TAGs

Bridge-LP is a new framework for link prediction on text‑attributed graphs that combines Graph Neural Networks and Language Models. It introduces three design components: jumping connections for robust language model training, compressed embeddings to encode neighborhood information, and a cache embedding strategy to reduce computational cost. Experiments on nine datasets show that Bridge‑LP outperforms 40 existing models, achieving up to a 43.06% improvement in Mean Reciprocal Rank.

By Chen Shao, Ruirui Wang, Tobias Kafer, Danai Koutra