arXiv:2501.10775cs.CVcs.AI2025-01被引 18

医学影像与报告细粒度对齐,提升疾病诊断准确率

MedFILIP: Medical Fine-grained Language-Image Pre-training

  • 用大模型提取报告中疾病细节,降低文本复杂度
  • 构建图像属性与病种关系,增强模型泛化能力
  • 基于细粒度标注的相似矩阵,实现精准图文匹配

医学视觉语言预训练(VLP)利用自然配对的医学影像-报告数据,对医学影像分析至关重要。然而,现有方法难以准确刻画影像与疾病之间的关联,导致诊断结果不准确或不完整。本文提出MedFILIP,一种细粒度医学VLP模型,通过对比学习引入医学图像特定知识:1)基于大语言模型的信息提取器,通过灵活提示工程解耦报告中的疾病细节,有效降低文本复杂度,同时保留丰富信息,开销极小;2)知识注入模块构建类别与视觉属性间的关系,使模型能基于图像特征做出判断,并实现对未知病种的知识外推;3)基于细粒度标注的语义相似性矩阵,提供更平滑、信息更丰富的标签,促进细粒度图像-文本对齐。我们在多个数据集上验证了MedFILIP,包括RSNA-Pneumonia、NIH ChestX-ray14、VinBigData和COVID-19。在单标签、多标签及细粒度分类任务中,模型均达到领先性能,分类准确率最高提升6.69%。代码已开源。

原文摘要 · Abstract (English)

Medical vision-language pretraining (VLP) that leverages naturally-paired medical image-report data is crucial for medical image analysis. However, existing methods struggle to accurately characterize associations between images and diseases, leading to inaccurate or incomplete diagnostic results. In this work, we propose MedFILIP, a fine-grained VLP model, introduces medical image-specific knowledge through contrastive learning, specifically: 1) An information extractor based on a large language model is proposed to decouple comprehensive disease details from reports, which excels in extracting disease deals through flexible prompt engineering, thereby effectively reducing text complexity while retaining rich information at a tiny cost. 2) A knowledge injector is proposed to construct relationships between categories and visual attributes, which help the model to make judgments based on image features, and fosters knowledge extrapolation to unfamiliar disease categories. 3) A semantic similarity matrix based on fine-grained annotations is proposed, providing smoother, information-richer labels, thus allowing fine-grained image-text alignment. 4) We validate MedFILIP on numerous datasets, e.g., RSNA-Pneumonia, NIH ChestX-ray14, VinBigData, and COVID-19. For single-label, multi-label, and fine-grained classification, our model achieves state-of-the-art performance, the classification accuracy has increased by a maximum of 6.69\%. The code is available in https://github.com/PerceptionComputingLab/MedFILIP.

医学AI细粒度对齐视觉语言模型

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