摘要 本研究提出TCM-DiffRAG框架通过整合知识图谱与思维链技术显著提升大语言模型在中医诊疗场景中的表现。实验表明该方法在三个测试集上均取得突破性成果尤其在个性化辨证任务中准确率从0.038提升至0.356。这为传统中医与人工智能深度融合开辟了新路径。阅读原文或https://t.zsxq.com/VWVUe获取原文pdf一、研究背景与挑战自2022年底以来大语言模型LLMs在医疗领域展现出巨大潜力。然而尽管通用LLMs在医疗自然语言处理任务中表现出色但在专业性和准确性方面仍远未达到专家水平有时甚至会产生误导性内容。中医诊疗面临独特挑战。不同于现代医学关注疾病本身中医的核心在于辨证论治——通过分析症状和体征来确定机体的根本性失调。中医辨证理论包括八纲、脏腑、经络、气血或三焦等多种体系不同流派如经方派、土派、温病派在具体诊疗过程中存在显著差异形成同病异治和异病同治的局面。现有的检索增强生成RAG技术在中医临床场景中遭遇诸多困境推理复杂性高中医临床问题涉及大量潜在推理传统嵌入模型只能匹配表面相似文本无法识别潜在逻辑结构知识库局限RAG知识库多源自教科书理论性强但与实际临床实践存在差距个性化不足难以反映不同中医流派的诊疗思维差异二、TCM-DiffRAG创新方法论2.1 双层次通用知识图谱构建研究团队收集了580部中医经典著作、名医医案等文献提出宏观-微观知识图谱构建方法。宏观层面利用文档布局模型识别PDF每页元素提取标题及对应段落文本构建类似树状图的知识图谱。节点由书籍标题组成节点关系通过标题的父子结构自动生成。微观层面使用大语言模型从段落文本中提取实体和关系。宏观标题节点作为结构枢纽与微观实体建立双向映射。数学表达为文档集D由章节层次结构ℋ和内容集合P组成D{ℋ,P}通过LLM提取医学逻辑三元组集GbookGbookLLMextract(D)每个三元组tk∈Gbook满足tk(esub,r,eobj)2.2 个性化知识图谱增强与迁移为弥合通用知识与特定临床实践的鸿沟研究通过分析不同流派医生的诊疗案例将通用知识图谱增强并迁移为个性化知识图谱。核心流程包括思维链分解将给定问题和答案输入Qwen2.5-72B-instruct模型生成多跳推理链并分解为结构化三元组三元组匹配与溯源生成的三元组与通用知识图谱中的实体对齐定位中医经典中的原文依据问题与文档关联为给定问题从文档集中检索k个最相关文本片段推理思维过程生成以对齐的三元组和相关经典文本为上下文驱动模型生成完整推理过程个性化图谱生成从推理文本中提取新实体和关系与原始通用知识图谱整合个性化知识图谱的优势在于通过分析医生实际诊疗案例的问答分解推理链显式捕获其个性化推理逻辑如流派偏好、辨证侧重避免通用知识图谱的同质化缺陷。同时在通用知识图谱约束下引入个性化知识时严格遵循中医理论体系的核心权威性。2.3 TCM-DiffRAG架构设计TCM-DiffRAG的核心创新在于通过思维链推理模型将临床问题分解为多跳三元组路径序列并基于个性化知识图谱进行语义对齐和证据生成。关键步骤步骤1思维链模型训练使用包含推理过程的问答对数据集C和分解推理过程得到的三元组Gstyle构建监督数据集DSFT通过领域监督微调获得专门化模型LLMcot。研究选择Qwen2.5-7B-instruct模型进行全参数微调使用8张A800 80G GPU基于LLaMA Factory框架批次大小为每GPU 2个学习率1e-4。步骤2多跳检索与知识增强思维链分解对输入问题qi使用LLMcot解析为多跳推理路径个性化知识召回多跳推理的三元组与个性化知识图谱进行语义相似度匹配条款检索通过映射从三元组到推理文本条款召回相关文本片段步骤3可溯源诊疗决策大语言模型基于召回的个性化知识图谱及其关联经典文本生成增强响应。得益于图的深度图遍历能力支持多跳推理以及隐式关联性和可扩展性召回内容既保证了结构化知识的广度又确保了可溯源推理的深度。三、实验设计与数据集3.1 数据集构成研究将语料数据集划分为四类中医书籍、TCM-MCQ、TCM-SD和经方-SD。语料库片段数平均令牌数平均三元组数中医书籍433,9503308TCM-MCQ训练集21,66010312TCM-MCQ测试集600117/TCM-SD训练集43,08540916TCM-SD测试集5,486416/经方-SD训练集20,04919416经方-SD测试集5,012194/三个评估基准代表不同难度等级TCM-MCQ语料库聚焦测试通用中医知识掌握源自中医医学考试题库要求从五个选项中选择唯一正确答案难度最低TCM-SD语料库源自徐州中医院真实病历要求从148个候选证候中确定唯一正确答案RAG难度显著增加经方-SD语料库来自浙江中医药大学第二、第三附属医院门诊病例要求从42个候选证候中选择唯一正确答案基于中医经方学派具有鲜明流派特征挑战最大3.2 评估框架研究采用RAGAS评估框架使用OpenAI的gpt-3.5-turbo-16k作为基础LLM阿里云text-embedding-v3作为嵌入模型召回文档数设置为k20。四、实验结果与性能分析4.1 通用知识图谱有效性评估实验对比了四种方法的性能表现方法准确率答案相似度上下文精确度上下文召回率上下文实体召回率无RAG0.4030.786\\\固定字符分割0.5400.8560.6210.8290.173宏观知识图谱分割0.6400.8630.8080.8480.188微观知识图谱分割0.6270.8710.7820.8360.192宏观-微观知识图谱集成检索0.6870.8850.8460.8870.244实验结果显示宏观-微观知识图谱集成方法在所有评估指标上均处于领先地位验证了所构建知识图谱方法的显著先进性。关键发现固定字符分割容易导致语义碎片化和实体断裂尤其在处理教科书表格时常破坏结构完整性宏观知识图谱分割依赖书籍原始层次结构有效保证语义单元完整性微观知识图谱分割能相对精确匹配实体关系但缺乏提取问题关键实体的显式步骤宏观-微观集成方法结合两者优势宏观层面保留辨证论治的逻辑框架和语义完整性微观三元组精确锁定核心知识实体4.2 思维链模型效能评估使用包含思维过程的数据集C和Gstyle对qwen-2.5-7B-instruct模型进行微调获得LLM-cot-7B。以deepseek-r1为参考模型使用Likert量表0-5分评估三元组质量结果显示微调后的LLM-cot-7B生成的三元组质量显著优于qwen-2.5-7B-instructP 0.05。此外LLM-cot-7B在TCM-SD测试集上的效果0.74显著优于前期研究0.52证实使用思维过程数据微调LLM能取得更优结果。进一步消融实验表明TCM-DiffRAG结合LLM-cot-7B和个性化知识图谱在所有三个测试集上均优于单独使用LLM-cot-7B。4.3 TCM-DiffRAG消融实验评估TCM-MCQ测试集图6当仅依靠模型自身能力时主要基于中文数据集训练的qwen-plus和deepseek-r1显著优于gpt-4o-mini和gemini-2.5-flash-preview。gpt-4o-mini和gemini-2.5-flash-preview在不同RAG方法下均获得显著提升而qwen-plus和deepseek-r1仅在使用TCM-DiffRAG结合LLM-cot-7B和个性化知识图谱时才有改进。这可能因为后两者已具备优秀的通用中医知识能力普通RAG方法引入噪声召回导致负面效果。TCM-SD数据集图7难度更高普通RAG方法对LLM性能提升效果不佳。仅在应用TCM-DiffRAG方法后才观察到显著改进。使用LLM-cot-7B作为思维链生成模型时相比原始模型有明显性能提升。这是因为TCM-SD作为临床实践测试集对模型推理能力要求极高。LLM-cot-7B能将输入查询分解为更细粒度的三元组并进行推理这些互联三元组形成具有临床思维链的知识图谱结构平衡了RAG的信息检索广度与思维链的深度推理能力。经方-SD测试集图8三个测试集中最具挑战性。未使用RAG时四个生成模型仅能达到0.03-0.07的准确率即使使用普通RAG方法也无显著改善。但使用LLM-cot-7B结合个性化知识图谱后准确率提升至0.35-0.38。可能原因除LLM缺乏经方诊疗推理能力外还在于通用知识图谱缺乏经方诊疗数据使用个性化知识图谱后可实现更显著改进。性能提升示例qwen-plus模型在三个测试集上的得分从0.927、0.361、0.038显著提升至0.952、0.788、0.356。非中文LLM的改进更为显著。此外TCM-DiffRAG的表现优于直接监督微调SFT的LLM和其他基准RAG方法。五、讨论与启示5.1 核心创新总结TCM-DiffRAG的核心创新在于引入结构化知识库构建工作流程和基于思维链CoT的训练方法。知识库构建方面充分利用中医书籍的章节层次结构进行文档分割构建通用中医知识图谱。相比传统固定长度字符分割方法这种基于语义单元的预处理方法具有显著优势。临床实践方面为弥合通用知识与特定临床实践的差距考虑到中医诊断思维的鲜明个性化特征基于通用知识图谱进一步构建个性化知识图谱及其配套CoT模型。该CoT模型能自动将输入问题分解为符合特定风格的多跳查询三元组序列并相应检索子图。TCM-DiffRAG结合了传统知识图谱RAG的广泛检索范围优势与基于推理RAG的深度推理优势在三个基准数据集上均取得最优性能。5.2 通用性与可推广性研究结论具有普遍性可扩展至其他非医疗领域TCM-MCQ简单领域知识问答LLM本身在此类任务上已表现优异通用RAG方法改进有限甚至可能产生负面效果TCM-SD领域通用推理问答虽然LLM掌握了通用行业知识但处理复杂推理问题的能力仍显不足。引入CoT模型可显著增强LLM性能经方-SD领域特定推理问答LLM缺乏此类特定私有数据的训练表现不佳。这是大多数企业的典型应用场景需要从内部业务数据进行推理。在此场景下TCM-DiffRAG结合CoT模型和个性化知识图谱可带来显著性能提升5.3 研究局限与未来方向本研究存在一定局限性知识图谱的评估维度需进一步丰富和完善TCM-DiffRAG在小样本数据条件下的性能有待进一步研究不同规模和基础的LLM作为CoT模型的差异化评估仍待探索这些将是未来研究的重要方向。六、结论TCM-DiffRAG展示了将结构化中医知识图谱与基于思维链的推理相结合能够大幅提升个性化诊断任务的性能。通用知识图谱与个性化知识图谱的联合使用实现了通用知识与临床推理的有效对齐。这些结果凸显了推理感知RAG框架在推进大语言模型中医应用方面的潜力。研究构建了一套评估数据集来验证中医思维链模型和不同知识库的性能为传统中医与人工智能的深度融合提供了可行路径也为其他专业领域的知识图谱增强生成提供了宝贵参考。