Self-Support Few-Shot Semantic Segmentation

23 Jul 2022  ·  Qi Fan, Wenjie Pei, Yu-Wing Tai, Chi-Keung Tang ·

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}.

PDF Abstract
Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Few-Shot Semantic Segmentation COCO-20i (1-shot) SSP (ResNet-101) Mean IoU 42 # 43
Few-Shot Semantic Segmentation COCO-20i (1-shot) SSP (ResNet-50) Mean IoU 37.4 # 60
Few-Shot Semantic Segmentation COCO-20i (5-shot) SSP (ResNet-101) Mean IoU 50.2 # 29
Few-Shot Semantic Segmentation COCO-20i (5-shot) SSP (ResNet-50) Mean IoU 44.1 # 55
Few-Shot Semantic Segmentation FSS-1000 (1-shot) SSP Mean IoU 87.3 # 14
Few-Shot Semantic Segmentation FSS-1000 (5-shot) SSP Mean IoU 88.6 # 12
Few-Shot Semantic Segmentation PASCAL-5i (1-Shot) SSP (ResNet-101) Mean IoU 64.6 # 50
Few-Shot Semantic Segmentation PASCAL-5i (1-Shot) SSP (ResNet-50) Mean IoU 61.4 # 68
Few-Shot Semantic Segmentation PASCAL-5i (5-Shot) SSP (ResNet-50) Mean IoU 69.3 # 41
Few-Shot Semantic Segmentation PASCAL-5i (5-Shot) SSP (ResNet-101) Mean IoU 73.1 # 12

Methods


No methods listed for this paper. Add relevant methods here