针对3D医学影像的多模态预训练,提升报告生成与病灶识别能力。
Comprehensive language-image pre-training for 3D medical image understanding
- 融合图文与图像仅预训练,利用双类数据增强模型泛化
- 在多个任务上达到当前最优,零样本分类准确率超基线12.5%
- 专为医学场景设计,适合临床辅助诊断与科研应用
在3D医学影像领域,视觉-语言预训练用于构建视觉-语言编码器(VLE),以支持放射科医生检索相似病例、预测异常概率或生成放射科报告。尽管该方法前景广阔,但当前3D VLE受限于三大挑战:隐私问题导致数据稀缺、体积图像带来的高计算成本,以及训练时使用长报告与推理时使用短提示之间的领域偏移。因此,自然图像的VLE方法难以直接迁移至3D医学影像。本文通过引入报告生成目标作为额外监督,并结合视觉-语言与纯视觉预训练,使模型可同时利用图像仅和图文配对的3D数据集。此外,提出一种新损失函数缓解长报告与短提示间的领域差异。结合3D医学影像领域的最佳实践,我们构建了综合语言-图像预训练(COLIPRI)编码器系列。COLIPRI在报告生成、语义分割、分类探测及零样本分类任务中均达到最先进性能。模型权重与推理代码已公开于 https://huggingface.co/microsoft/colipri。
原文摘要 · Abstract (English)
In the 3D medical image domain, vision-language pre-training is used to create vision-language encoders (VLEs) that can support radiologists by retrieving patients with similar abnormalities, predicting likelihoods of abnormality, or, with downstream adaptation, generating radiological reports. While the methodology holds promise, three challenges limit the capabilities of current 3D VLEs: data scarcity due to privacy concerns, high computational costs resulting from the volumetric nature of the images, and a domain shift between the long reports used for training and the short prompts used during inference for, e.g., zero-shot classification. As a consequence, natural-image VLE recipes do not directly transfer to 3D medical imaging. In this paper, we overcome these challenges by injecting additional supervision via a report generation objective and combining vision-language with vision-only pre-training, allowing us to leverage both image-only and paired image-text 3D datasets. Further, we propose a novel loss that addresses the domain shift between long reports and short textual prompts. Through these additional objectives, paired with best practices of the 3D medical imaging domain, we develop the Comprehensive Language-Image Pre-training (COLIPRI) encoder family. Our COLIPRI encoders achieve state-of-the-art performance in report generation, semantic segmentation, classification probing, and zero-shot classification. The model weights and inference code are freely available at https://huggingface.co/microsoft/colipri.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。