Few-shot Adaptation of Medical Vision-Language Models

Fereshteh Shakeri, Yunshi Huang, Julio Silva-Rodríguez, Houda Bahig, An Tang +2 more
9/5/2024
cs.CV

Official code: FereshteShakeri/few-shot-MedVLMs (34 stars, last commit 2024-10-06).

Abstract

Integrating image and text data through multi-modal learning has emerged as a new approach in medical imaging research, following its successful deployment in computer vision. While considerable efforts have been dedicated to establishing medical foundation models and their zero-shot transfer to downstream tasks, the popular few-shot setting remains relatively unexplored. Following on from the currently strong emergence of this setting in computer vision, we introduce the first structured benchmark for adapting medical vision-language models (VLMs) in a strict few-shot regime and investigate various adaptation strategies commonly used in the context of natural images. Furthermore, we evaluate a simple generalization of the linear-probe adaptation baseline, which seeks an optimal blending of the visual prototypes and text embeddings via learnable class-wise multipliers. Surprisingly, such a text-informed linear probe yields competitive performances in comparison to convoluted prompt-learning and adapter-based strategies, while running considerably faster and accommodating the black-box setting. Our extensive experiments span three different medical modalities and specialized foundation models, nine downstream tasks, and several state-of-the-art few-shot adaptation methods. We made our benchmark and code publicly available to trigger further developments in this emergent subject: \url{https://github.com/FereshteShakeri/few-shot-MedVLMs}.

View on arXivView PDF

Code Implementations(1)

344Shell, PythonJul 8, 20242 years ago

Similar Papers

Cite this paper

@article{shakeri2024fewshot,
  title  = {Few-shot Adaptation of Medical Vision-Language Models},
  author = {Fereshteh Shakeri and Yunshi Huang and Julio Silva-Rodríguez and Houda Bahig and An Tang and Jose Dolz and Ismail Ben Ayed},
  year   = {2024},
  eprint = {2409.03868},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2409.03868v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2409.03868.svg)](https://codeofpaper.com/paper/2409.03868)

Discussion