Critical Evidence Graphs (CEG)
收藏资源简介:
MedCEG团队构建的Critical Evidence Graphs (CEG)数据集由1万例复杂临床案例组成,旨在通过结构化证据图增强医疗语言模型的推理可验证性。该数据集整合了MedQA、MedCase和JAMA Challenge的原始数据,通过大模型筛选高难度案例并生成问答-推理三元组,再经多模型协同提取语义关系构建证据图,最终提炼关键推理子图(CEG)。其核心内容涵盖临床实体、因果关系及最小必要推理路径,支持医疗AI生成符合临床逻辑的透明化决策过程。该资源应用于医疗推理强化学习,解决传统方法中推理过程缺乏临床有效性的问题,推动可靠医疗AI的发展。
The Critical Evidence Graphs (CEG) dataset, developed by the MedCEG Team, consists of 10,000 complex clinical cases, with the goal of enhancing the reasoning verifiability of medical language models via structured evidence graphs. This dataset integrates raw data sourced from MedQA, MedCase, and the JAMA Challenge. First, large language models (LLMs) are utilized to screen high-difficulty clinical cases and generate question-answer-reasoning triples; subsequently, evidence graphs are constructed by extracting semantic relationships through multi-model collaboration, and finally key reasoning subgraphs (CEG) are distilled. The core content of this dataset covers clinical entities, causal relationships, and minimal necessary reasoning paths, which supports medical AI in generating transparent decision-making processes that conform to clinical logic. This resource is applied to medical reasoning reinforcement learning, addressing the problem that the reasoning processes of traditional methods lack clinical validity, and promoting the development of reliable medical AI.
MedCEG数据集概述
数据集基本信息
- 数据集名称:MedCEG (Reinforcing Verifiable Medical Reasoning with Critical Evidence Graph)
- 核心目标:通过**关键证据图(Critical Evidence Graph, CEG)**增强医学语言模型,监督推理过程,确保可验证且逻辑的医学推断。
- 访问地址:https://huggingface.co/LinjieMu/MedCEG
- 相关论文:http://arxiv.org/abs/2512.13510
- 许可协议:Apache 2.0
数据内容与结构
- 数据文件:
DataExample.jsonl包含构建的训练数据样本。 - 数据字段说明:
question:输入的临床问题。answer:标准答案。reasoning_content:与CEG对齐的重写“思考”过程。graph/triplets:完整的证据图(Evidence Graph, EG)。graph/core_reasoning_subgraph:用于奖励计算的关键证据图(CEG)。
方法框架与流程
- 模型架构与流程:
- 冷启动(Cold-Start):将结构化证据图转化为自然语言,教授模型逻辑依赖关系,预热推理能力。
- 图引导的强化学习(Graph-guided Reinforcement Learning):利用**关键证据图(CEG)**提供密集的、面向过程的奖励,引导模型走向准确的诊断路径。
- 训练工具:使用VeRL(Volcengine Reinforcement Learning)进行RLHF/RLAIF阶段训练。
- 自定义过程奖励函数:位于
code/reward目录,用于评估推理过程质量。
推理与使用
- 推理脚本:
Inference.py演示如何使用MedCEG生成回答。 - 模型加载:通过HuggingFace Transformers库加载模型与分词器。
- 输入格式:用户问题后附加后缀
Put your final answer in oxed{}。。 - 生成配置:使用贪婪解码(
do_sample=False),最大新生成令牌数为8196。
实验性能
- 主要结果:在多个医学基准测试中达到最先进的性能,在准确性和推理质量上均有显著提升。
- 推理过程质量评估:从五个维度评估推理过程:逻辑连贯性、事实准确性、证据忠实度、可解释性与清晰度、信息利用度。MedCEG在生成临床合理的推理方面显著优于基线模型。
文件结构
. ├── ReadMe.md └── code ├── DataExample.jsonl # 示例数据结构 ├── evaluation │ └── ProcessEvaluation.py # 评估推理过程精确度 ├── reward # VeRL的核心奖励逻辑 │ ├── graph_extract.py # 通过LLM从推理文本中提取三元组 │ ├── GraphReward.py # 计算基于图的奖励(节点、结构、链) │ ├── GraphMCQ.py # 多项选择题的奖励入口点 │ ├── GraphOpenendQuestion.py # 开放式问题的奖励入口点 │ └── TripletsRecall.py # 计算三元组召回的实用工具 └── server └── EmbeddingServer.py # 用于BGE嵌入的FastAPI服务器

- 1MedCEG: Reinforcing Verifiable Medical Reasoning with Critical Evidence Graph上海交通大学 · 2025年



