Global-Local Medical SAM Adaptor Based on Full Adaption
Visual language models (VLM) have made great breakthroughs in natural image segmentation, and they show consistently high-quality zero-shot transferring capability in many down-stream tasks with prompt engineering. The help of modern parameter-efficient fine-tuning techniques, such as low-rank adaption (LoRA) and Medical SAM adaptor (Med-SA) further enables VLM to be well transferred to medical domains. Regardless of the performance, these methods choose to fine-tune VLM in a local manner, where adapters are specified to adapt some layers vital to the down-stream tasks, making them separated by frozen layers in forward path and lack of collaboration in back-propagation. To resolve this problem, we present a global adaption technique and create a novel global medical SAM adaptor (GMed-SA), which can adapt SAM in a collaborative adaption manner. Specifically, all adapters communicate via fully connected layers as a unified sub-network, enabling them to work collaboratively in fine-tuning. We further combine GMed-SA and Med-SA to propose a global-local medical SAM adaptor (GL-Med-SA) for 2D melanoma segmentation dataset. The extensive experiments show that the combined method GL-Med-SA slightly improve the segmentation performance on the challenging publicly available 2D melanoma dataset while only enlarging minor trainable weight parameters and outperforming several state-of-the-art methods in both Dice metrics and Intersection over union (IoU), which demonstrates its superiority.
Authors
- Chao Lv (ORCID: https://orcid.org/0000-0002-4087-4672)
- Yarong Feng
- Tian Zhang (ORCID: https://orcid.org/0009-0003-1150-3842)
- Yuxin Liang (ORCID: https://orcid.org/0000-0001-6872-5030)
- Yongwei Tang (ORCID: https://orcid.org/0000-0003-0088-3774)
- Meng Wang (ORCID: https://orcid.org/0000-0001-7648-9955)
Institutions
- Liaoning Technical University (CN)
- China Medical University (CN)
- Northeastern University (CN)
Publication Details
- Published
- 2024-11-24
- DOI
- https://doi.org/10.1109/iccvit63928.2024.10872479
- Primary Topic
- Advanced Computing and Algorithms
- Type
- article
- Field-Weighted Citation Impact
- 0.00