arrow
Return

Automatic Prompt Generation Using Class Activation Maps for Foundational Models: A Polyp Segmentation Case Study

delete2025-02-24
delete0
delete
OA
AI
H
Hanna Borgli *
H
Håkon Kvale Stensland
P
Pål Halvorsen
DOI:10.3390/make7010022delete
deleteOriginal
deleteShare
deleteSave
View PDF
Abstract

Abstract

En 中文
We introduce a weakly supervised segmentation approach that leverages class activation maps and the Segment Anything Model to generate high-quality masks using only classification data. A pre-trained classifier produces class activation maps that, once thresholded, yield bounding boxes encapsulating the regions of interest. These boxes prompt the SAM to generate detailed segmentation masks, which are then refined by selecting the best overlap with automatically generated masks from the foundational model using the intersection over union metric. In a polyp segmentation case study, our approach outperforms existing zero-shot and weakly supervised methods, achieving a mean intersection over union of 0.63. This method offers an efficient and general solution for image segmentation tasks where segmentation data are scarce.
Keywords:
image segmentation
class activation map
segment anything
annotation tools
zero-shot learning
weakly supervised semantic segmentation
deep learning
medical image processing
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

M
Machine Learning and Knowledge Extraction
IF:
6
Papers:
795
Citations:
1.8K

Organization

U
university of oslo
Scholars:
4.2W
Papers: 3.5W
Citations: 53