arXiv:2410.06542eess.IVcs.CV2024-10被引 55

开源医学影像嵌入模型,跨多种模态达顶尖性能。

MedImageInsight: An Open-Source Embedding Model for General Domain Medical Imaging

  • 基于多模态医学图像与文本训练,支持跨域通用表征。
  • 在胸部X光、皮肤科、OCT等任务上达最先进水平,骨龄评估媲美专家。
  • 轻量级报告生成,兼具临床实用性与可解释性,适合医疗AI研发者。

本文提出MedImageInsight,一个开源的医学影像嵌入模型,涵盖X光、CT、MRI、皮肤镜、OCT、眼底摄影、超声、病理切片和乳腺钼靶等多种模态。经严格评估,该模型在分类、图像检索及微调任务中均达到或超越当前最先进(SOTA)水平。在公开数据集上,其在三维CT图像检索、胸部X光、皮肤病学及OCT疾病分类与搜索任务中表现领先;在骨龄估计任务中达到人类专家水平,多数领域AUC超过0.9。结合文本解码器后,仅需其他模型10%参数即可实现接近最先进水平的单图报告生成。相较于仅用MIMIC-CXR微调GPT-4o,MedImageInsight在临床指标上更优,但在词汇指标上略逊于GPT-4o。重要的是,模型支持生成ROC曲线,可根据临床需求调节敏感性与特异性,并通过图像检索提供循证决策支持(亦可用于检索增强生成)。在独立临床评估中,其胸部X光图像检索性能远超所有公开基础模型(AUC提升超6点),且在年龄与性别维度上展现显著更高的AI公平性。我们希望释放MedImageInsight能推动医学影像AI研究的集体进步。

原文摘要 · Abstract (English)

In this work, we present MedImageInsight, an open-source medical imaging embedding model. MedImageInsight is trained on medical images with associated text and labels across a diverse collection of domains, including X-Ray, CT, MRI, dermoscopy, OCT, fundus photography, ultrasound, histopathology, and mammography. Rigorous evaluations demonstrate MedImageInsight's ability to achieve state-of-the-art (SOTA) or human expert level performance across classification, image-image search, and fine-tuning tasks. Specifically, on public datasets, MedImageInsight achieves SOTA in CT 3D medical image retrieval, as well as SOTA in disease classification and search for chest X-ray, dermatology, and OCT imaging. Furthermore, MedImageInsight achieves human expert performance in bone age estimation (on both public and partner data), as well as AUC above 0.9 in most other domains. When paired with a text decoder, MedImageInsight achieves near SOTA level single image report findings generation with less than 10\% the parameters of other models. Compared to fine-tuning GPT-4o with only MIMIC-CXR data for the same task, MedImageInsight outperforms in clinical metrics, but underperforms on lexical metrics where GPT-4o sets a new SOTA. Importantly for regulatory purposes, MedImageInsight can generate ROC curves, adjust sensitivity and specificity based on clinical need, and provide evidence-based decision support through image-image search (which can also enable retrieval augmented generation). In an independent clinical evaluation of image-image search in chest X-ray, MedImageInsight outperformed every other publicly available foundation model evaluated by large margins (over 6 points AUC), and significantly outperformed other models in terms of AI fairness (across age and gender). We hope releasing MedImageInsight will help enhance collective progress in medical imaging AI research and development.

医学影像嵌入模型多模态AI医疗

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