arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2308.02463cs.CVcs.CL

迈向基于网络规模2D和3D医学数据的放射学通用基础模型

Towards Generalist Foundation Model for Radiology by Leveraging Web-scale 2D&3D Medical Data

Shanghai Jiao Tong University(上海交通大学);Shanghai AI Laboratory(上海人工智能实验室)

Chaoyi Wu, Xiaoman Zhang, Ya Zhang, Yanfeng Wang, Weidi Xie

AI总结:

本研究提出放射学基础模型RadFM,通过构建大规模多模态数据集MedMD和RadMD,设计视觉条件生成预训练架构,并在新基准RadBench上超越现有模型,推动放射学AI发展。

摘要:

在本研究中,我们旨在启动放射学基础模型(称为RadFM)的开发。我们从三个角度考虑基础模型的构建,即数据集构建、模型设计和全面评估。我们的贡献可总结如下:(i)我们构建了一个大规模医学多模态数据集MedMD,该数据集包含1600万张2D和3D医学扫描图像,并附有跨多种数据格式、模态和任务的高质量文本描述或报告,覆盖超过5000种不同疾病。据我们所知,这是首个大规模、高质量、同时包含2D和3D扫描的医学视觉-语言数据集;(ii)我们提出了一种架构,能够实现视觉条件生成预训练,即允许将文本输入与2D或3D医学扫描集成,并为各种放射学任务生成响应。该模型最初在MedMD上进行预训练,随后在特定领域数据集(即MedMD的放射学清理版本,包含300万放射学视觉-语言对,称为RadMD)上进行微调;(iii)我们提出了一个新的评估基准RadBench,该基准包含五项任务,包括模态识别、疾病诊断、视觉问答、报告生成和推理诊断,旨在全面评估基础模型在处理实际临床问题方面的能力。我们在RadBench上进行了自动和人工评估,在两种情况下,RadFM均优于现有可公开访问的多模态基础模型,包括Openflamingo、MedFlamingo、MedVInT和GPT-4V。此外,我们还调整了RadFM以适应不同的公共基准,在多个数据集上超越了现有的最先进水平。所有代码、数据和模型检查点都将公开提供,以促进该领域的进一步研究和发展。

英文摘要:

In this study, we aim to initiate the development of Radiology Foundation Model, termed as RadFM. We consider the construction of foundational models from three perspectives, namely, dataset construction, model design, and thorough evaluation. Our contribution can be concluded as follows: (i), we construct a large-scale Medical Multi-modal Dataset, MedMD, which consists of 16M 2D and 3D medical scans with high-quality text descriptions or reports across various data formats, modalities, and tasks, covering over 5000 distinct diseases. To the best of our knowledge, this is the first large-scale, high-quality, medical visual-language dataset, with both 2D and 3D scans; (ii), we propose an architecture that enables visually conditioned generative pre-training, i.e., allowing for integration of text input with 2D or 3D medical scans, and generate responses for diverse radiologic tasks. The model was initially pre-trained on MedMD and subsequently fine-tuned on the domain-specific dataset, which is a radiologic cleaned version of MedMD, containing 3M radiologic visual-language pairs, termed as RadMD; (iii), we propose a new evaluation benchmark, RadBench, that comprises five tasks, including modality recognition, disease diagnosis, visual question answering, report generation and rationale diagnosis, aiming to comprehensively assess the capability of foundation models in handling practical clinical problems. We conduct both automatic and human evaluation on RadBench, in both cases, RadFM outperforms existing multi-modal foundation models, that are publicaly accessible, including Openflamingo, MedFlamingo, MedVInT and GPT-4V. Additionally, we also adapt RadFM for different public benchmarks, surpassing existing SOTAs on diverse datasets. All codes, data, and model checkpoint will all be made publicly available to promote further research and development in the field.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑