Skip to main navigation Skip to search Skip to main content

Med-Tuning: A New Parameter-Efficient Tuning Framework for Medical Volumetric Segmentation

  • Jiachen Shen
  • , Wenxuan Wang
  • , Chen Chen
  • , Jianbo Jiao
  • , Jing Liu
  • , Yan Zhang
  • , Shanshan Song
  • , Jiangyun Li*
  • *Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference contribution

69 Downloads (Pure)

Abstract

The “pre-training then fine-tuning (FT)” paradigm is widely adopted to boost the model performance of deep learning-based methods for medical volumetric segmentation. However, conventional full FT incurs high computational and memory costs. Thus, it is of increasing importance to fine-tune pre-trained models for medical volumetric segmentation tasks in a both effective and parameter-efficient manner. In this paper, we introduce a new framework named Med-Tuning to realize parameter-efficient tuning (PET) for medical volumetric segmentation task and an efficient plug-and-play module named Med-Adapter for task-specific feature extraction. With a small number of tuned parameters, our framework enhances the 2D baselines’s precision on segmentation tasks, which are pre-trained on natural images. Extensive experiments on three benchmark datasets (CT and MRI modalities) show that our method achieves better results than previous PET methods on volumetric segmentation tasks. Compared to full FT, Med-Tuning reduces the fine-tuned model parameters by up to 4×, with even better segmentation performance. Our project webpage is at https://rubics-xuan.github.io/Med-Tuning/.

Original languageEnglish
Title of host publicationProceedings of The 7nd International Conference on Medical Imaging with Deep Learning
EditorsNinon Burgos, Caroline Petitjean, Maria Vakalopoulou, Stergios Christodoulidis, Pierrick Coupe, Hervé Delingette, Carole Lartizien, Diana Mateus
PublisherPMLR
Pages1412–1433
Number of pages22
Publication statusPublished - 5 Jul 2024
Event7th International Conference on Medical Imaging with Deep Learning, MIDL 2024 - Paris, France
Duration: 3 Jul 20245 Jul 2024

Publication series

NameProceedings of Machine Learning Research
Volume250
ISSN (Electronic)2640-3498

Conference

Conference7th International Conference on Medical Imaging with Deep Learning, MIDL 2024
Country/TerritoryFrance
CityParis
Period3/07/245/07/24

Bibliographical note

Copyright:
© 2024 CC-BY 4.0, J. Shen, W. Wang, C. Chen, J. Jiao, J. Liu, Y. Zhang, S. Song & J. Li.

Keywords

  • Medical Volumetric Segmentation
  • Parameter-Efficient Tuning
  • Transformer

ASJC Scopus subject areas

  • Artificial Intelligence
  • Software
  • Control and Systems Engineering
  • Statistics and Probability

Fingerprint

Dive into the research topics of 'Med-Tuning: A New Parameter-Efficient Tuning Framework for Medical Volumetric Segmentation'. Together they form a unique fingerprint.

Cite this