arXiv:2609.04415cs.LGcs.AI2026-09

基于患者轨迹动态调整医学图谱信息,提升电子病历预测效果

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

论文配图:REFINE: LLM Refinement over Budgeted Text-Attributed Graphs for Personalized Medical Concept Representation
图 1 · 摘自论文原文
  • 按患者需求动态选择图谱扩展范围,避免信息冗余
  • 融合图神经网络与冻结大模型,精准刻画医疗概念语义
  • 在病历数据少时仍表现稳定,适合临床实际场景

学习丰富的医疗概念表示对电子病历(EHR)预测至关重要。文本属性知识图谱(TKG)通过整合异构医疗关系与文本语义提供了天然基础。然而,现有编码器通常对所有患者统一处理概念,忽略了编码的含义和预测价值依赖于患者特定的临床背景与病程演变。从TKG中学习个性化概念表示面临两大挑战:(1) 如何为每个观测到的编码决定引入多少图谱上下文;(2) 如何将语义信息与患者特定的关系结构对齐。我们提出REFINE,一种面向预算约束的KG感知大模型图谱精炼框架,用于患者个性化医疗概念编码。从全局TKG出发,REFINE构建患者特定的时间图谱。通过序列强化学习策略,为每个观测编码分配个性化图谱扩展预算。生成的患者图谱由异质图神经网络处理以捕捉关系感知的结构依赖,同时冻结的大语言模型利用图感知软提示对概念表示进行语义精炼。在MIMIC-III和MIMIC-IV上的实验表明,REFINE能持续提升多种EHR骨干模型性能,优于强基线,在组件消融、图谱选择及数据不足情况下均表现出稳健增益。

原文摘要 · Abstract (English)

Learning rich medical concept representations is essential for EHR prediction. Text-attributed knowledge graphs (TKGs) provide a natural foundation by organizing heterogeneous medical relations together with textual semantics. However, most existing encoders process concepts uniformly across patients, despite the fact that a code's meaning and predictive value depend on patient-specific clinical context and trajectory. Learning patient-personalized concept representations from TKGs introduces two key challenges: (1) deciding how much KG context to incorporate for each observed code, and (2) aligning semantic information with the patient-specific relational structure. We propose REFINE, a KG-aware budgeted LLM graph refinement framework for patient-personalized medical concept encoding. Starting from a global TKG, REFINE constructs patient-specific temporal graphs. A sequential reinforcement learning policy selects a personalized KG expansion budget for each observed code. The resulting patient graph is processed by a heterogeneous GNN to capture relation-aware structural dependencies, while a frozen LLM uses graph-aware soft prompts to semantically refine concept representations. Experiments on MIMIC-III and MIMIC-IV show that REFINE consistently improves diverse EHR backbones, outperforms strong baselines, and demonstrates robust gains across component ablation, KG selection, and data insufficiency.

医疗表示学习知识图谱大模型应用个性化建模

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。