DDXPlus, CPDD
收藏资源简介:
DDXPlus是一个大规模合成的电子健康记录(EHR)数据集,广泛认可用于提供复杂多样的医疗诊断案例。它包括全面的患者数据,如社会人口信息、基础疾病、症状和既往病史。CPDD是一个来自新加坡陈笃生医院的私人EHR数据集,用于慢性疼痛管理。
DDXPlus is a large-scale synthetic electronic health record (EHR) dataset widely recognized for providing complex and diverse medical diagnostic cases. It encompasses comprehensive patient data including sociodemographic information, underlying diseases, symptoms and past medical history. CPDD is a private EHR dataset from Tan Tock Seng Hospital in Singapore, used for chronic pain management.
MedRAG 数据集概述
数据集简介
MedRAG是一个为医疗领域设计的 Retrieval-Augmented Generation(RAG)模型,通过集成知识图谱(KG)驱动的推理来增强模型的能力。该模型旨在帮助医疗专业人员基于患者的症状生成诊断和治疗建议,提高诊断的准确性并减少误诊的风险。
关键特性
- 知识图谱增强推理:整合了诊断知识图谱以提高RAG模型的推理能力。
- 准确的诊断支持:为复杂或症状相似的疾病提供具体的诊断见解和个性化的治疗建议。
- 后续问题生成:主动生成相关后续问题,以澄清模糊的患者信息并增强决策。
- 在真实世界和公共数据集上评估:在公共DDXPlus数据集和私有的慢性疼痛诊断数据集(CPDD)上展示了优于现有RAG模型的性能。
数据集详情
- DDXPlus数据集:一个大规模的合成电子健康记录(EHR)数据集,包含复杂的、多样化的医疗诊断案例。
- CPDD数据集:新加坡合作伙伴医院——陈笃生医院提供的慢性疼痛管理私人EHR数据集。
使用说明
- 准备仓库和依赖项:克隆仓库到本地机器并安装
requirements.txt中的依赖。 - 修改令牌:在
authentication.py中替换占位符以使用自己的OpenAI和Hugging Face API令牌。 - 运行
main.py脚本:更新路径和令牌后,运行main.py以启动程序。
实验结果
- MedRAG在多个指标和所有数据集上实现了最佳或次佳性能(只有一个例外)。
- 在不同的LLM基础模型上评估KG驱动的推理,结果表明加入KG推理显著提高了所有基础模型的诊断准确性。
引用
bibtex @inproceedings{zhao2025medrag, title={MedRAG: Enhancing Retrieval-augmented Generation with Knowledge Graph-Elicited Reasoning for Healthcare Copilot}, author={Zhao, Xuejiao and Liu, Siyan and Yang, Su-Yin and Miao, Chunyan}, booktitle={THE WEB CONFERENCE 2025} }




