CAD: Memory Efficient Convolutional Adapter for Segment Anything

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Kim, Joohyeok, Song, Joonhyeon, Yun, Seohwan, Yoon, Seongho, Lee, Sangmin
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910618193231872
author Kim, Joohyeok
Song, Joonhyeon
Yun, Seohwan
Yoon, Seongho
Lee, Sangmin
author_facet Kim, Joohyeok
Song, Joonhyeon
Yun, Seohwan
Yoon, Seongho
Lee, Sangmin
contents The Foundation model for image segmentation, Segment Anything (SAM), has been actively researched in various fields since its proposal. Various researches have been proposed to adapt SAM to specific domains, with one notable approach involving the addition and training of lightweight adapter modules. While adapter-based fine-tuning approaches have reported parameter efficiency and significant performance improvements, they face a often overlooked issue: the excessive consumption of GPU memory relative to the number of trainable parameters. Addressing this issue, this paper proposes a memory-efficient parallel convolutional adapter architecture. This architecture connects in parallel with SAM's image encoder, eliminating the need to store activations and gradients of the image encoder during model training. Our proposed architecture demonstrated competitive experimental results while using less than half the GPU memory compared to SAM Adapter, indicating its value as an alternative to simple decoder fine-tuning when hardware limitations preclude adapter-based learning. Our code implementation is available at our github.
format Preprint
id arxiv_https___arxiv_org_abs_2409_15889
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle CAD: Memory Efficient Convolutional Adapter for Segment Anything
Kim, Joohyeok
Song, Joonhyeon
Yun, Seohwan
Yoon, Seongho
Lee, Sangmin
Computer Vision and Pattern Recognition
The Foundation model for image segmentation, Segment Anything (SAM), has been actively researched in various fields since its proposal. Various researches have been proposed to adapt SAM to specific domains, with one notable approach involving the addition and training of lightweight adapter modules. While adapter-based fine-tuning approaches have reported parameter efficiency and significant performance improvements, they face a often overlooked issue: the excessive consumption of GPU memory relative to the number of trainable parameters. Addressing this issue, this paper proposes a memory-efficient parallel convolutional adapter architecture. This architecture connects in parallel with SAM's image encoder, eliminating the need to store activations and gradients of the image encoder during model training. Our proposed architecture demonstrated competitive experimental results while using less than half the GPU memory compared to SAM Adapter, indicating its value as an alternative to simple decoder fine-tuning when hardware limitations preclude adapter-based learning. Our code implementation is available at our github.
title CAD: Memory Efficient Convolutional Adapter for Segment Anything
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2409.15889