DKG-LLM:一个通过动态知识图谱与大语言模型集成实现医疗诊断和个性化治疗推荐的框架

论文信息

  • 标题 (Title):DKG-LLM: A Framework for Medical Diagnosis and Personalized Treatment Recommendations via Dynamic Knowledge Graph and Large Language Model Integration

  • 作者 (Authors):Ali Sarabadani, Maryam Abdollahi shamami, Hamidreza Sadeghsalehi, Borhan Asadi, Saba Hesarakis

  • 原文链接 (URL)https://arxiv.org/abs/2508.06186

结构化摘要 (Structured Abstract)

  • 背景/目标 (Background/Objective):大型语言模型(LLMs)在处理非结构化文本方面表现出色,但在关键的医疗应用中常因缺乏透明的推理能力而导致输出不一致 。另一方面,传统的静态知识图谱(KGs)难以扩展和实时整合新知识 。本研究旨在提出 DKG-LLM 框架,通过将动态知识图谱(DKG)与 LLM(Grok 3)相结合,为医疗诊断和个性化治疗推荐提供一个可靠且可演进的解决方案。

  • 方法 (Methods):该框架的核心是一种名为**自适应语义融合算法(ASFA)**的定制算法 。ASFA 利用概率模型、贝叶斯推理和图优化技术,从临床报告和 PubMed 文章等异构医疗数据中提取语义信息,并动态更新知识图谱 。该 DKG 初始化包含 15,964 个节点(13种类型)和 127,392 条边(26种关系类型),并且能够高效地处理增量数据(每批次增加约150个新节点和边)。框架采用 Grok 3 模型进行自然语言处理,并结合马尔可夫随机场(MRFs)进行图剪枝以保证可扩展性

  • 结果 (Results):在使用 MIMIC-III 和 PubMed 等真实世界数据集的评估中,DKG-LLM 框架表现出色,诊断准确率达到 84.19%治疗推荐精确率为 89.63%语义覆盖率达到 93.48% 。其图更新效率极高,每批次数据处理时间小于1秒 。在由全科医生、内分泌科医生和心脏病专家进行的定性评估中,该框架也获得了高度认可(平均里氏得分>4.1,Cohen's Kappa系数为0.80)

  • 结论 (Conclusion):DKG-LLM 是一个可靠且具有变革性的工具,它成功地将 DKG 的结构化、可演进知识与 LLM 的强大语言理解能力相结合 。该框架能够有效处理噪声数据和复杂的多症状疾病,并通过基于医生反馈的强化学习机制持续改进,为临床环境提供了一个实用的解决方案


1. 引言 (Introduction)

1.1. 研究背景与核心问题 (Research Background & Problem Statement)

  • 研究背景:随着患者生成的医疗数据和临床报告的快速增长,开发创新的医疗保健系统面临着新的机遇与挑战 。在复杂的临床决策中,准确诊断并给出个性化治疗建议极具挑战性,因为这需要考虑疾病、治疗和患者特定因素之间错综复杂的关系

  • 核心问题 (RQs)

    1. LLM 的局限性:尽管自主的 LLMs 在处理非结构化文本方面表现优异,但它们往往缺乏透明的推理过程,导致在关键医疗应用中输出不一致或不可靠

    2. 静态知识库的瓶颈:现有的方法,特别是基于静态知识图谱(如 UMLS)或规则的系统,在扩展和整合新数据方面能力有限,难以跟上医学知识的快速发展

1.2. 文献综述与研究缺口 (Literature Review & Research Gap)

  • 现有研究:作者梳理了近年来将知识图谱与 LLM 相结合用于医疗领域的研究,例如 DR.KNOWS 、LLMRG 、DynLLM 和 ESCARGOT 。这些工作展示了 KG 能够为 LLM 提供结构化信息,从而提高诊断预测的准确性和可解释性

  • 研究缺口 (Gap):现有研究多集中于利用 KG 辅助 LLM 进行诊断推理或问答,但较少关注知识图谱本身的动态演进能力。当前缺乏一个能够从异构数据源中实时、自动地构建和更新知识图谱,并将其与 LLM 无缝集成以同时支持诊断和个性化治疗推荐的综合性框架。

1.3. 研究目标与核心假设/命题 (Objectives & Hypotheses/Propositions)

  • 研究目标:提出 DKG-LLM 框架,通过集成一个**动态知识图谱(DKG)**和 Grok 3 大语言模型,为医疗领域提供一个全新的、支持知情决策的范式

  • 核心假设/命题:一个能够动态演进的知识图谱,结合 LLM 强大的语义提取能力,并通过一个先进的融合算法(ASFA)进行整合,能够克服 LLM 单独使用时推理不透明和静态知识库更新不及时的双重问题。该框架不仅能提供高精度的诊断,还能根据患者个体情况生成高度个性化的治疗方案。


2. 研究设计与方法 (Methodology)

2.1. 研究范式与方法论 (Research Paradigm & Methodology)

  • 研究范式:本研究为定量研究,通过设计、实现和评估一个新的人工智能系统来验证其在医疗任务中的有效性。

  • 方法论与模型架构:DKG-LLM 框架由三大核心组件构成

    1. 动态知识图谱 (DKG):一个结构化的知识库,其节点(V)代表医疗实体(如疾病、症状、患者画像),边(E)代表语义关系(如“导致”、“治疗”)。与传统 KG 不同,它是动态演进的。

    2. 大型语言模型模块:采用 Grok 3 模型,负责从非结构化的医疗文本(如临床报告、PubMed 文章)中提取语义信息(实体和关系)

    3. 自适应语义融合算法 (ASFA):这是框架的核心创新,是一个集成了概率建模和图优化技术的算法,负责将 Grok 3 的输出与 DKG 进行融合、更新和推理

  • 关键解决方案与优势

    • ASFA 算法:该算法通过五个阶段(数据摄入、语义提取、图更新、推理与推荐、反馈集成)的流水线作业,实现了系统的核心功能

      • 语义提取阶段,它不仅提取实体,还计算置信度分数以处理噪声数据

      • 图更新阶段,它使用**马尔可夫随机场(MRF)**模型对图进行剪枝,去除低概率的节点和边,以保证图的可扩展性和效率

      • 推理与推荐阶段,它分别采用贝叶斯推理进行诊断(计算疾病的后验概率)和约束优化进行治疗推荐(最大化治疗效用函数)

      • 反馈集成阶段,它利用强化学习,根据临床医生的反馈来调整模型参数,实现持续学习和优化

    • 与之前方法的区别和优势

      • 动态性与实时性:与使用 UMLS 等静态知识库的方法相比,DKG-LLM 的知识图谱能够根据最新的医疗数据(如新发表的论文)进行动态更新,更新效率极高(<1秒/批次),保证了知识的时效性

      • 严谨的数学基础:整个框架建立在先进的数学模型之上,包括用于诊断的贝叶斯推理和用于治疗推荐的约束优化,这使得其决策过程比纯 LLM 方法更具鲁棒性和可解释性

      • 可扩展性:通过 MRF 图剪枝机制,系统在不断吸收新知识的同时,能够将图的规模控制在预设的最大值内(987,654条边),确保了计算效率

2.2. 数据来源与样本 (Data Source & Sample)

  • 数据来源

    • 真实世界数据:使用了 MIMIC-III(包含患者临床记录)和 PubMed(包含科研文献),以评估框架从多样化的非结构化文本中提取信息和更新 DKG 的能力

    • 模拟数据:生成了包含复杂病例(如罕见病)和噪声数据(如模糊的临床报告)的合成场景,以测试框架的鲁棒性

  • DKG 规模:DKG 初始化时包含 15,964 个节点(13种类型)和 127,392 条边(26种关系类型),这些数据基于 SNOMED CT 等医学本体构建

2.3. 操作化与测量 (Operationalization & Measurement)

  • 诊断:通过贝叶斯公式计算给定一组症状 S 时,某个疾病 d 的后验概率 P(dS)

  • 治疗推荐:将其建模为一个优化问题,旨在找到能最大化预期效用函数 U(T, d, P) 的治疗方案 T*。该效用函数综合考虑了治疗效果、患者情况和风险

  • 评估指标

    • 定量指标:诊断准确率、治疗推荐精确率、语义覆盖率、图更新效率

    • 定性指标:通过对专科医生进行访谈和问卷调查,使用 5 点里氏量表评估推荐方案的准确性、可靠性和可用性,并使用 Cohen's Kappa 系数衡量评估者之间的一致性


3. 结果与发现 (Results & Findings)

3.1. 主要发现概述 (Overview of Key Findings)

  • 定量性能优越:该框架在所有核心指标上均取得了优异的成绩。诊断准确率达到 84.19%,治疗推荐精确率达到 89.63%,语义覆盖率达到 93.48%

  • 系统高效运行:图更新效率非常高,处理每批次数据(约150个新节点/边)的时间少于1秒,证明了其在临床环境中应用的潜力

  • 临床医生高度认可:由三名专科医生组成的团队对框架的输出进行了评估,给出了很高的评价。在准确性、可靠性和可用性三个维度上,平均里氏得分分别为 4.3、4.1 和 4.2 。评估者之间的一致性很高,Cohen's Kappa 系数达到了 0.80

  • 在复杂场景下表现鲁棒:在处理多症状并发疾病(如糖尿病合并高血压)和噪声数据(如模糊的临床记录)的复杂场景中,框架依然保持了很高的性能,语义提取准确率为 91.5%,图对齐分数为 92.7%

3.2. 关键数据与图表解读 (Interpretation of Key Data & Figures)

  • 表1:综合知识图谱指标 (Comprehensive Knowledge Graph Metrics)

    • 解读:该表将本研究的 DKG 与 DrKG、PrimeKG 等多个知名的生物医学知识图谱进行了比较。虽然 DKG-LLM 的初始节点和边数量较少,但其节点类型(13种)和边类型(26种)的数量非常丰富,这表明其知识表示的语义粒度非常细致,能够捕捉比其他图谱更复杂多样的医学关系

  • 表2和表3:边类型和节点类型 (Edge/Node Types and Description)

    • 解读:这两张表详细展示了 DKG 的“模式”或“本体”。表2列出了26种边类型,如“因果关系”、“治疗关系”、“副作用”和“基因相关”等 。表3列出了13种节点类型,如“疾病”、“症状”、“患者画像”和“生活方式因素”等 。这直观地展示了 DKG 能够构建一个多么全面和精细的医疗知识网络,为复杂的推理提供了坚实的基础。

  • 表4:指标和结果 (Metrics and Result)

    • 解读:这是论文的核心成果汇总表,简洁明了地列出了所有关键的定量和定性评估结果 。从中可以清晰地看到,该框架在诊断、推荐、知识覆盖、效率以及临床医生认可度等多个维度上都取得了成功,全面地证明了其有效性和实用性。


4. 讨论 (Discussion)

4.1. 结果的深度解读 (In-depth Interpretation of Results)

  • DKG-LLM 的高精度和高效率源于其创新的整合方式。动态知识图谱为 LLM 提供了一个结构化、可验证的“外部大脑”,有效地**“锚定”了 LLM 的推理过程,使其不易产生幻觉,并能提供清晰的决策路径。反过来,LLM 的强大语义理解能力使知识图谱能够实时地“学习”**和吸收最新的非结构化医学知识。ASFA 算法是连接这两者的关键桥梁,它通过严谨的数学方法确保了二者融合的准确性和效率。

4.2. 理论贡献 (Theoretical Contributions)

  • 提出全新的医疗AI框架:DKG-LLM 框架本身就是一项重要的理论贡献,它为如何将 LLM 与动态演进的知识结构相结合以解决复杂问题提供了一个新范式。

  • ASFA 算法的创新性:ASFA 算法融合了概率建模、图优化和强化学习等多种技术,为实现异构数据源的动态、自适应融合提供了一套完整的、有数学基础支持的方法论。

  • 对业界的影响:这项研究为开发下一代临床决策支持系统提供了蓝图。它展示了如何构建一个既能利用 LLM 的灵活性,又能保证结构化知识的严谨性和时效性的系统,有望推动更精准、更可靠的 AI 医疗产品的落地。

4.3. 实践启示 (Practical Implications)

  • 临床决策支持:该框架可作为医生的强大助手,尤其是在处理具有多种并发症的复杂病例时,能够提供全面的诊断和个性化治疗建议。

  • 知识管理:医疗机构可以利用该框架自动地从海量的临床文档和最新研究中构建和维护一个动态的、机构专属的知识库。

4.4. 局限性与未来研究 (Limitations & Future Research)

  • 局限性:作者承认,确保数据隐私和针对更大规模的数据集进行优化仍然是挑战

  • 未来研究

    • 数据隐私:计划采用联邦学习来在保护患者隐私的同时训练模型

    • 数据融合:计划整合来自生物传感器的实时数据,以进行更动态的健康监测

    • 应用拓展:计划将框架的应用扩展到新发疾病预测等领域


5. 结论 (Conclusion)

本研究提出的 DKG-LLM 框架,通过其核心的自适应语义融合算法(ASFA),成功地将动态知识图谱与 Grok 3 大语言模型相结合,为医疗诊断和个性化治疗推荐提供了一个变革性的解决方案 。该系统在处理异构数据、动态更新知识库方面表现高效,并在诊断和推荐任务上取得了高准确率,其鲁棒性和实用性得到了真实世界数据和临床专家评估的验证 。总体而言,DKG-LLM 在可扩展性、持续改进能力以及与先进概率模型和图优化的整合方面的创新,使其有巨大潜力彻底改变医疗保健系统,改善患者预后并降低医疗成本

6. 核心参考文献 (Core References)

  1. Gao, Y., et al. (2025). Leveraging medical knowledge graphs into large language models for diagnosis prediction...

    (即 DR.KNOWS,一篇利用 UMLS 知识图谱与 LLM 结合进行诊断预测的相关工作)

  2. Feng, Y., et al. (2025). Knowledge graph-based thought: a knowledge graph-enhanced LLM framework for pan-cancer question answering.

    (一篇利用 KG 增强 LLM 在癌症问答中能力的相关研究)

  3. Zhao, Z., et al. (2024). DynLLM: when large language models meet dynamic graph recommendation.

    (一篇关于 LLM 与动态图结合用于推荐系统的研究,其“动态”特性与本文相关)

  4. Matsumoto, N., et al. (2025). ESCARGOT: an AI agent leveraging large language models, dynamic graph of thoughts, and biomedical knowledge graphs for enhanced reasoning.

    (一篇结合 LLM、动态思维图和生物医学知识图谱进行推理的先进研究)

  5. Yang, H., et al. (2025). Large Language Model-Driven Knowledge Graph Construction in Sepsis Care...

    (一篇利用 LLM 在脓毒症护理领域构建知识图谱的研究,与本文中 KG 的构建和应用场景相关)


–EOF–
转载须以超链接形式标明文章原始出处和作者信息及版权声明.

No comments: