Towards Foundation Models and Few-Shot Parameter-Efficient Fine-Tuning for Volumetric Organ Segmentation
arXiv:2303.17051 · doi:10.1016/j.media.2025.103596
Abstract
The recent popularity of foundation models and the pre-train-and-adapt paradigm, where a large-scale model is transferred to downstream tasks, is gaining attention for volumetric medical image segmentation. However, current transfer learning strategies devoted to full fine-tuning for transfer learning may require significant resources and yield sub-optimal results when the labeled data of the target task is scarce. This makes its applicability in real clinical settings challenging since these institutions are usually constrained on data and computational resources to develop proprietary solutions. To address this challenge, we formalize Few-Shot Efficient Fine-Tuning (FSEFT), a novel and realistic scenario for adapting medical image segmentation foundation models. This setting considers the key role of both data- and parameter-efficiency during adaptation. Building on a foundation model pre-trained on open-access CT organ segmentation sources, we propose leveraging Parameter-Efficient Fine-Tuning and black-box Adapters to address such challenges. Furthermore, novel efficient adaptation methodologies are introduced in this work, which include Spatial black-box Adapters that are more appropriate for dense prediction tasks and constrained transductive inference, leveraging task-specific prior knowledge. Our comprehensive transfer learning experiments confirm the suitability of foundation models in medical image segmentation and unveil the limitations of popular fine-tuning strategies in few-shot scenarios.
Accepted in Medical Image Analysis. The pre-trained model and adaptation code is available at: https://github.com/jusiro/fewshot-finetuning
References in corpus (16)
- Very Deep Convolutional Networks for Large-Scale Image Recognition
- A Simple Framework for Contrastive Learning of Visual Representations
- Learning to Prompt for Vision-Language Models
- The Medical Segmentation Decathlon
- Recent advances and clinical applications of deep learning in medical image analysis
- CHAOS Challenge -- Combined (CT-MR) Healthy Abdominal Organ Segmentation
- Segment Anything Model for Medical Image Analysis: an Experimental Study
- Med3D: Transfer Learning for 3D Medical Image Analysis
- Models Genesis
- CLIP-Driven Universal Model for Organ Segmentation and Tumor Detection
- Tent: Fully Test-time Adaptation by Entropy Minimization
- WORD: A large scale dataset, benchmark and clinical applicable study for abdominal organ segmentation from CT image
- Fine-Tuning can Distort Pretrained Features and Underperform Out-of-Distribution
- Tip-Adapter: Training-free CLIP-Adapter for Better Vision-Language Modeling
- A Foundation Language-Image Model of the Retina (FLAIR): Encoding Expert Knowledge in Text Supervision
- Training BatchNorm and Only BatchNorm: On the Expressive Power of Random Features in CNNs