Few-shot Semantic Segmentation with Support-induced Graph Convolutional Network

9 Jan 2023  ·  Jie Liu, Yanqi Bao, Wenzhe Yin, Haochen Wang, Yang Gao, Jan-Jakob Sonke, Efstratios Gavves ·

Few-shot semantic segmentation (FSS) aims to achieve novel objects segmentation with only a few annotated samples and has made great progress recently. Most of the existing FSS models focus on the feature matching between support and query to tackle FSS. However, the appearance variations between objects from the same category could be extremely large, leading to unreliable feature matching and query mask prediction. To this end, we propose a Support-induced Graph Convolutional Network (SiGCN) to explicitly excavate latent context structure in query images. Specifically, we propose a Support-induced Graph Reasoning (SiGR) module to capture salient query object parts at different semantic levels with a Support-induced GCN. Furthermore, an instance association (IA) module is designed to capture high-order instance context from both support and query instances. By integrating the proposed two modules, SiGCN can learn rich query context representation, and thus being more robust to appearance variations. Extensive experiments on PASCAL-5i and COCO-20i demonstrate that our SiGCN achieves state-of-the-art performance.

PDF Abstract
No code implementations yet. Submit your code now

Datasets


Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Few-Shot Semantic Segmentation COCO-20i (1-shot) SiGCN (ResNet-50) Mean IoU 41.4 # 47
FB-IoU 62.7 # 25
Few-Shot Semantic Segmentation COCO-20i (5-shot) SiGCN (ResNet-50) Mean IoU 48 # 41
FB-IoU 66.2 # 26
Few-Shot Semantic Segmentation PASCAL-5i (1-Shot) SiGCN (ResNet-101) Mean IoU 65.7 # 40
FB-IoU 78.3 # 20
Few-Shot Semantic Segmentation PASCAL-5i (1-Shot) SiGCN (VGG-16) Mean IoU 60.8 # 70
FB-IoU 73.5 # 38
Few-Shot Semantic Segmentation PASCAL-5i (1-Shot) SiGCN (ResNet-50) Mean IoU 65.3 # 44
FB-IoU 77.5 # 29
Few-Shot Semantic Segmentation PASCAL-5i (5-Shot) SiGCN (ResNet-50) Mean IoU 68.5 # 47
FB-IoU 78.3 # 31

Methods