用大模型生成提示词,让医学图像识别更准更通用
BiomedCoOp: Learning to Prompt for Biomedical Vision-Language Models
- 用大模型平均提示词做语义对齐,提升提示学习效果
- 在11个医学数据集上实现比现有方法更高的准确率和泛化能力
- 适合医疗影像少样本分类,无需调参或微调模型
视觉语言模型(如CLIP)在自监督表征学习中表现优异,但将其适配到下游任务仍面临挑战,尤其在生物医学图像领域。由于标注数据少、图像对比度弱、视觉特征复杂,传统提示工程耗时且依赖经验,全模型微调成本高。现有提示学习方法如CoOp在泛化性上仍有不足,而医学图像领域的相关研究极少。本文提出BiomedCoOp,一种新型提示学习框架,可高效适配BiomedCLIP,实现精准且高度泛化的少样本医学图像分类。通过利用大语言模型生成的平均提示词保持语义一致性,并结合基于统计的提示选择策略进行知识蒸馏,实现有效提示上下文学习。我们在9种模态、10个器官、共11个医学数据集上进行了全面验证,结果表明该方法在准确率和泛化性能上均显著优于现有最优方法。代码已开源:https://github.com/HealthX-Lab/BiomedCoOp。
原文摘要 · Abstract (English)
Recent advancements in vision-language models (VLMs), such as CLIP, have demonstrated substantial success in self-supervised representation learning for vision tasks. However, effectively adapting VLMs to downstream applications remains challenging, as their accuracy often depends on time-intensive and expertise-demanding prompt engineering, while full model fine-tuning is costly. This is particularly true for biomedical images, which, unlike natural images, typically suffer from limited annotated datasets, unintuitive image contrasts, and nuanced visual features. Recent prompt learning techniques, such as Context Optimization (CoOp) intend to tackle these issues, but still fall short in generalizability. Meanwhile, explorations in prompt learning for biomedical image analysis are still highly limited. In this work, we propose BiomedCoOp, a novel prompt learning framework that enables efficient adaptation of BiomedCLIP for accurate and highly generalizable few-shot biomedical image classification. Our approach achieves effective prompt context learning by leveraging semantic consistency with average prompt ensembles from Large Language Models (LLMs) and knowledge distillation with a statistics-based prompt selection strategy. We conducted comprehensive validation of our proposed framework on 11 medical datasets across 9 modalities and 10 organs against existing state-of-the-art methods, demonstrating significant improvements in both accuracy and generalizability. The code is publicly available at https://github.com/HealthX-Lab/BiomedCoOp.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。