arXiv:2501.12266cs.CVcs.AI2025-01中稿 · publication in Com…被引 4

无需训练,用大模型实现可解释的医学图像分类

CBVLM: Training-free Explainable Concept-based Large Vision Language Models for Medical Image Classification

  • 用大模型直接判断图像中预设概念是否存在
  • 仅需少量标注样本,准确率超越传统方法
  • 适合医疗场景下对可解释性要求高的研究者

深度学习在医疗流程中的应用受限于标注数据稀缺和模型不可解释。概念瓶颈模型(CBMs)通过限定输出为人类可理解的概念来提升可解释性,但需大量标注且新增概念需重训。受大视觉语言模型(LVLMs)少样本性能启发,我们提出无需训练的CBVLM方法:首先用LVLM判断输入图像中各概念是否存在,再基于预测结果进行分类;两阶段均引入检索模块,选择最优上下文样例用于少样本学习。通过将诊断结果锚定在概念预测上,保证可解释性;利用LVLM的少样本能力,显著降低标注成本。在四个医学数据集和十二个LVLM(通用与医疗专用)上验证,CBVLM无需训练、仅用少量标注样本,性能持续优于CBMs及监督微调方法。

原文摘要 · Abstract (English)

The main challenges limiting the adoption of deep learning-based solutions in medical workflows are the availability of annotated data and the lack of interpretability of such systems. Concept Bottleneck Models (CBMs) tackle the latter by constraining the model output on a set of predefined and human-interpretable concepts. However, the increased interpretability achieved through these concept-based explanations implies a higher annotation burden. Moreover, if a new concept needs to be added, the whole system needs to be retrained. Inspired by the remarkable performance shown by Large Vision-Language Models (LVLMs) in few-shot settings, we propose a simple, yet effective, methodology, CBVLM, which tackles both of the aforementioned challenges. First, for each concept, we prompt the LVLM to answer if the concept is present in the input image. Then, we ask the LVLM to classify the image based on the previous concept predictions. Moreover, in both stages, we incorporate a retrieval module responsible for selecting the best examples for in-context learning. By grounding the final diagnosis on the predicted concepts, we ensure explainability, and by leveraging the few-shot capabilities of LVLMs, we drastically lower the annotation cost. We validate our approach with extensive experiments across four medical datasets and twelve LVLMs (both generic and medical) and show that CBVLM consistently outperforms CBMs and task-specific supervised methods without requiring any training and using just a few annotated examples. More information on our project page: https://cristianopatricio.github.io/CBVLM/.

可解释性医学图像大模型少样本

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