Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Leveraging Hallucinations to Reduce Manual Prompt Dependency in Promptable Segmentation

About

Promptable segmentation typically requires instance-specific manual prompts to guide the segmentation of each desired object. To minimize such a need, task-generic promptable segmentation has been introduced, which employs a single task-generic prompt to segment various images of different objects in the same task. Current methods use Multimodal Large Language Models (MLLMs) to reason detailed instance-specific prompts from a task-generic prompt for improving segmentation accuracy. The effectiveness of this segmentation heavily depends on the precision of these derived prompts. However, MLLMs often suffer hallucinations during reasoning, resulting in inaccurate prompting. While existing methods focus on eliminating hallucinations to improve a model, we argue that MLLM hallucinations can reveal valuable contextual insights when leveraged correctly, as they represent pre-trained large-scale knowledge beyond individual images. In this paper, we utilize hallucinations to mine task-related information from images and verify its accuracy for enhancing precision of the generated prompts. Specifically, we introduce an iterative Prompt-Mask Cycle generation framework (ProMaC) with a prompt generator and a mask generator.The prompt generator uses a multi-scale chain of thought prompting, initially exploring hallucinations for extracting extended contextual knowledge on a test image.These hallucinations are then reduced to formulate precise instance-specific prompts, directing the mask generator to produce masks that are consistent with task semantics by mask semantic alignment. The generated masks iteratively induce the prompt generator to focus more on task-relevant image areas and reduce irrelevant hallucinations, resulting jointly in better prompts and masks. Experiments on 5 benchmarks demonstrate the effectiveness of ProMaC. Code given in https://lwpyh.github.io/ProMaC/.

Jian Hu, Jiayi Lin, Junchi Yan, Shaogang Gong• 2024

Related benchmarks

TaskDatasetResultRank
Polyp SegmentationKvasir--
128
Camouflaged Object SegmentationCAMO 250 images (test)
Mean Absolute Error (MAE)0.09
40
Camouflaged Object DetectionCOD10K 14 (test)
M Score4.2
21
Camouflaged Object SegmentationCOD10K 2016 (test)
Fw_beta71.6
21
Camouflaged Object SegmentationNC4K 4121 (test)
Fw_beta77.7
21
Camouflaged Object DetectionCHAMELEON 50 (test)
M0.044
21
Camouflaged Object DetectionCAMO 30 (test)
M Score0.09
21
Open-Vocabulary SegmentationPascal Context
mIoU30.7
20
Open-Vocabulary SegmentationCOCO Object
mIoU25.2
20
Camouflaged Object SegmentationCHAMELEON 87 (test)
Fw_beta79
19
Showing 10 of 16 rows

Other info

Code

Follow for update