提升医疗视觉语言模型的准确性,防止误诊
MMed-RAG: Versatile Multimodal RAG System for Medical Vision Language Models
- 设计领域感知检索与自适应上下文选择,增强多模态信息融合
- 在5个医疗数据集上使事实准确率平均提升43.8%
- 适合需要高可信度诊断辅助的临床研究与AI系统开发者
人工智能在医疗领域展现出巨大潜力,尤其在疾病诊断与治疗规划方面。近年来,医疗大视觉语言模型(Med-LVLMs)为交互式诊断工具带来新可能,但常出现事实幻觉,导致误诊。微调和检索增强生成(RAG)是缓解此问题的方法,但高质量数据有限且训练与部署数据分布差异限制了微调应用。尽管RAG轻量高效,现有方法在不同医疗领域泛化能力不足,且易引发模态间及模型与真实情况间的错位。本文提出通用多模态RAG系统MMed-RAG,通过领域感知检索机制、自适应检索上下文选择方法以及可证明的基于RAG的偏好微调策略,显著提升检索过程的通用性与可靠性,改善引入检索内容后的对齐效果。在涉及放射科、眼科、病理科的五个医疗数据集上,针对医学视觉问答与报告生成任务的实验表明,MMed-RAG使Med-LVLM的事实准确率平均提升43.8%。代码与数据已开源。
原文摘要 · Abstract (English)
Artificial Intelligence (AI) has demonstrated significant potential in healthcare, particularly in disease diagnosis and treatment planning. Recent progress in Medical Large Vision-Language Models (Med-LVLMs) has opened up new possibilities for interactive diagnostic tools. However, these models often suffer from factual hallucination, which can lead to incorrect diagnoses. Fine-tuning and retrieval-augmented generation (RAG) have emerged as methods to address these issues. However, the amount of high-quality data and distribution shifts between training data and deployment data limit the application of fine-tuning methods. Although RAG is lightweight and effective, existing RAG-based approaches are not sufficiently general to different medical domains and can potentially cause misalignment issues, both between modalities and between the model and the ground truth. In this paper, we propose a versatile multimodal RAG system, MMed-RAG, designed to enhance the factuality of Med-LVLMs. Our approach introduces a domain-aware retrieval mechanism, an adaptive retrieved contexts selection method, and a provable RAG-based preference fine-tuning strategy. These innovations make the RAG process sufficiently general and reliable, significantly improving alignment when introducing retrieved contexts. Experimental results across five medical datasets (involving radiology, ophthalmology, pathology) on medical VQA and report generation demonstrate that MMed-RAG can achieve an average improvement of 43.8% in the factual accuracy of Med-LVLMs. Our data and code are available in https://github.com/richard-peng-xia/MMed-RAG.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。