arXiv:2411.12915cs.CV2024-11CVPR被引 75

用医学专家模型增强视觉语言模型,提升医疗诊断准确性

VILA-M3: Enhancing Vision-Language Models with Medical Expert Knowledge

  • 引入第四阶段专用指令微调,融合领域专家模型知识
  • 在医疗任务上比现有最优模型平均提升9%,特定任务提升6%
  • 适合医疗AI研发、医学影像分析等专业场景使用

通用视觉语言模型(VLMs)在计算机视觉领域取得显著进展,但在医疗等专业领域仍显不足,因缺乏专业知识。传统模型依赖互联网记忆知识,难以胜任医疗所需的高精度要求。当前VLMs通常分三阶段训练:视觉预训练、视觉-语言预训练和指令微调(IFT)。而我们提出,针对医疗VLMs,需增加第四阶段——专注于医疗数据的专用指令微调,并引入领域专家模型的知识。这些专家模型专为临床任务(如肿瘤检测、异常分类)训练,能学习医学数据中的精细特征,尤其是放射学中复杂细节,是通用VLM难以捕捉的。本文提出VILA-M3框架,通过专家模型注入领域知识。实验表明,该方法在多个医疗任务上达到新SOTA,平均性能比先前最优模型Med-Gemini提升约9%,比特定任务训练模型提升约6%。结果强调了领域专家知识对构建精准、可靠医疗VLMs的重要性。

原文摘要 · Abstract (English)

Generalist vision language models (VLMs) have made significant strides in computer vision, but they fall short in specialized fields like healthcare, where expert knowledge is essential. In traditional computer vision tasks, creative or approximate answers may be acceptable, but in healthcare, precision is paramount.Current large multimodal models like Gemini and GPT-4o are insufficient for medical tasks due to their reliance on memorized internet knowledge rather than the nuanced expertise required in healthcare. VLMs are usually trained in three stages: vision pre-training, vision-language pre-training, and instruction fine-tuning (IFT). IFT has been typically applied using a mixture of generic and healthcare data. In contrast, we propose that for medical VLMs, a fourth stage of specialized IFT is necessary, which focuses on medical data and includes information from domain expert models. Domain expert models developed for medical use are crucial because they are specifically trained for certain clinical tasks, e.g. to detect tumors and classify abnormalities through segmentation and classification, which learn fine-grained features of medical data$-$features that are often too intricate for a VLM to capture effectively especially in radiology. This paper introduces a new framework, VILA-M3, for medical VLMs that utilizes domain knowledge via expert models. Through our experiments, we show an improved state-of-the-art (SOTA) performance with an average improvement of ~9% over the prior SOTA model Med-Gemini and ~6% over models trained on the specific tasks. Our approach emphasizes the importance of domain expertise in creating precise, reliable VLMs for medical applications.

医疗AI视觉语言模型专家知识

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。