让AI诊断更可信:用外部医学指南显式增强多模态模型决策
RAD: Towards Trustworthy Retrieval-Augmented Multi-modal Clinical Diagnosis
- 通过检索与精炼疾病知识,显式注入临床指南到诊断流程
- 在4个不同解剖部位数据集上达到顶尖性能,聚焦异常区域更精准
- 首次提出多模态诊断可解释性评估标准,适合医疗AI可信研究者
临床诊断需高度专业性和严格遵循指南。当前AI医疗研究多依赖知识图谱或文本预训练,隐式编码知识,忽视下游任务所需的特定知识。为此,我们提出检索增强诊断(RAD)框架,直接在下游任务中显式注入外部知识。RAD通过三大机制实现:从多源医学资料中检索并精炼以疾病为中心的知识;采用指南增强的对比损失,约束多模态特征与指南知识的潜在距离;双变压器解码器以指南为查询引导跨模态融合,对齐从指南获取到特征提取再到决策的临床诊断流程。此外,针对多模态诊断模型缺乏可解释性量化评估的问题,我们引入一套从图像与文本角度评估可解释性的标准。在四个不同解剖部位的数据集上广泛验证显示,RAD具备良好泛化能力,达到当前最优性能。同时,模型能更精确聚焦于异常区域和关键指标,确保基于证据、值得信赖的诊断。代码已开源:https://github.com/tdlhl/RAD。
原文摘要 · Abstract (English)
Clinical diagnosis is a highly specialized discipline requiring both domain expertise and strict adherence to rigorous guidelines. While current AI-driven medical research predominantly focuses on knowledge graphs or natural text pretraining paradigms to incorporate medical knowledge, these approaches primarily rely on implicitly encoded knowledge within model parameters, neglecting task-specific knowledge required by diverse downstream tasks. To address this limitation, we propose Retrieval-Augmented Diagnosis (RAD), a novel framework that explicitly injects external knowledge into multimodal models directly on downstream tasks. Specifically, RAD operates through three key mechanisms: retrieval and refinement of disease-centered knowledge from multiple medical sources, a guideline-enhanced contrastive loss that constrains the latent distance between multi-modal features and guideline knowledge, and the dual transformer decoder that employs guidelines as queries to steer cross-modal fusion, aligning the models with clinical diagnostic workflows from guideline acquisition to feature extraction and decision-making. Moreover, recognizing the lack of quantitative evaluation of interpretability for multimodal diagnostic models, we introduce a set of criteria to assess the interpretability from both image and text perspectives. Extensive evaluations across four datasets with different anatomies demonstrate RAD's generalizability, achieving state-of-the-art performance. Furthermore, RAD enables the model to concentrate more precisely on abnormal regions and critical indicators, ensuring evidence-based, trustworthy diagnosis. Our code is available at https://github.com/tdlhl/RAD.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。