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}.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Few-shot Semantic Segmentation | PASCAL-5^i (test) | -- | 177 | |
| Few-shot Segmentation | COCO 20^i (test) | mIoU44.1 | 174 | |
| Semantic segmentation | COCO-20i | mIoU (Mean)44.1 | 132 | |
| Few-shot Semantic Segmentation | COCO 5-shot 20i | mIoU50.2 | 85 | |
| Few-shot Segmentation | PASCAL 5i (val) | mIoU (Mean)69.3 | 83 | |
| Few-shot Semantic Segmentation | COCO-20i (test) | mIoU (mean)44.1 | 79 | |
| Few-shot Segmentation | COCO-20^i | -- | 78 | |
| Few-shot Semantic Segmentation | COCO 20i 1-shot | mIoU (Overall)42 | 77 | |
| Few-shot Semantic Segmentation | FSS-1000 | mIoU79.4 | 64 | |
| Few-shot Segmentation | DeepGlobe | mIoU54.2 | 61 |