arXiv AI

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.

arXiv AI
Aug 25

Clinical Graph-JEPA: Predictive Patient-State Knowledge Graphs for Cognitive Decision Support

The paper introduces Clinical Graph-JEPA, a framework for building and refining predictive patient-state knowledge graphs from clinical records. It combines multi-agent relation proposal, ontology-aware normalization, deterministic evidence scoring, and JEPA-based latent refinement to construct evidence-scored graphs from MIMIC-IV data and recover missing clinical relations. Experiments show that injecting discharge-note representations into note-grounded entities boosts leave-one-out MRR by 31% relative improvement.

By Kushagra Yadav, Nalin Prabhath, Amit Lamba, Goeun Han, Yining Mao
arXiv AI
Jun 3

ChatHealthAI: Aligning Electronic Health Record Representations with Large Language Models for Grounded Clinical Reasoning

arXiv:2606. 02802v1 Announce Type: new Abstract: Large language models (LLMs) exhibit strong natural-language reasoning abilities for clinical decision support, but struggle to effectively model structured longitudinal electronic health records (EHRs).

By Bo-Hong Wang, Baicheng Peng, Ruilin Wang, Jun Bai, Ziyang Song, Yue Li
arXiv AI
Sep 3

Import What You Need: Learning When and How to Augment EHR Graphs with External Knowledge

The paper introduces ReTA, a reinforcement‑learning framework that dynamically decides how to augment electronic health record (EHR) graphs with external knowledge graphs (KGs) on a per‑visit basis. ReTA offers three actions—Soft Import, Hard Import, and Skip—allowing it to enrich node features, graft compact KG subgraphs, or leave the graph unchanged depending on the patient’s evolving state and a budget constraint. Experiments on MIMIC‑III and MIMIC‑IV demonstrate that ReTA consistently outperforms strong baselines, transfers across datasets and KGs, and provides interpretable augmentation patterns while maintaining efficiency.

By Chen Chen, Mohsen Nayebi Kerdabadi, Dongjie Wang, Mei Liu, Zijun Yao
arXiv Machine Learning
4d ago

Reinforcement Learning over Patient Trajectories for Clinical Reasoning in EHR Foundation Models

The paper proposes a reinforcement learning (RL) fine‑tuning framework for electronic health record (EHR) foundation models, treating them as generative policies over patient trajectories. By framing clinical prediction tasks such as hospital readmission as event‑conditioned, time‑windowed reasoning problems and designing time‑aware, rollout‑sensitive rewards, the authors show that RL fine‑tuning consistently outperforms pre‑trained backbones and strong baselines. The approach enables smaller models to surpass larger pre‑trained models in data‑limited settings, induces positive transfer across tasks, and produces trajectories with stronger structural and semantic alignment to ground truth, improving downstream utility.

By Yuxin Xiao, Sheng Zhang, Chandan Singh, Tristan Naumann, Hoifung Poon, Jianfeng Gao, Xiaodong Liu
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