Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Self-Support Few-Shot Semantic Segmentation

About

Existing few-shot segmentation methods have achieved great progress based on the support-query matching framework. But they still heavily suffer from the limited coverage of intra-class variations from the few-shot supports provided. Motivated by the simple Gestalt principle that pixels belonging to the same object are more similar than those to different objects of same class, we propose a novel self-support matching strategy to alleviate this problem, which uses query prototypes to match query features, where the query prototypes are collected from high-confidence query predictions. This strategy can effectively capture the consistent underlying characteristics of the query objects, and thus fittingly match query features. We also propose an adaptive self-support background prototype generation module and self-support loss to further facilitate the self-support matching procedure. Our self-support network substantially improves the prototype quality, benefits more improvement from stronger backbones and more supports, and achieves SOTA on multiple datasets. Codes are at \url{https://github.com/fanq15/SSP}.

Qi Fan, Wenjie Pei, Yu-Wing Tai, Chi-Keung Tang• 2022

Related benchmarks

TaskDatasetResultRank
Few-shot Semantic SegmentationCOCO-20i
mIoU44.1
178
Few-shot Semantic SegmentationPASCAL-5^i (test)--
177
Few-shot SegmentationCOCO 20^i (test)
mIoU44.1
174
Semantic segmentationCOCO-20i
mIoU (Mean)44.1
144
Few-shot Semantic SegmentationCOCO 5-shot 20i
mIoU50.2
85
Few-shot SegmentationDeepGlobe
mIoU54.2
83
Few-shot SegmentationPASCAL 5i (val)
mIoU (Mean)69.3
83
Few-shot SegmentationChest X-ray
mIoU74.4
82
Few-shot Semantic SegmentationCOCO-20i (test)
mIoU (mean)44.1
79
Few-shot SegmentationCOCO-20^i--
78
Showing 10 of 30 rows

Other info

Code

Follow for update