Cost Aggregation Is All You Need for Few-Shot Segmentation

22 Dec 2021  ·  Sunghwan Hong, Seokju Cho, Jisu Nam, Seungryong Kim ·

We introduce a novel cost aggregation network, dubbed Volumetric Aggregation with Transformers (VAT), to tackle the few-shot segmentation task by using both convolutions and transformers to efficiently handle high dimensional correlation maps between query and support. In specific, we propose our encoder consisting of volume embedding module to not only transform the correlation maps into more tractable size but also inject some convolutional inductive bias and volumetric transformer module for the cost aggregation... Our encoder has a pyramidal structure to let the coarser level aggregation to guide the finer level and enforce to learn complementary matching scores. We then feed the output into our affinity-aware decoder along with the projected feature maps for guiding the segmentation process. Combining these components, we conduct experiments to demonstrate the effectiveness of the proposed method, and our method sets a new state-of-the-art for all the standard benchmarks in few-shot segmentation task. Furthermore, we find that the proposed method attains state-of-the-art performance even for the standard benchmarks in semantic correspondence task although not specifically designed for this task. We also provide an extensive ablation study to validate our architectural choices. The trained weights and codes are available at: https://seokju-cho.github.io/VAT/. read more

PDF Abstract

Results from the Paper


Task Dataset Model Metric Name Metric Value Global Rank Benchmark
Few-Shot Semantic Segmentation COCO-20i (1-shot) VAT (ResNet-50) Mean IoU 41.3 # 1
Few-Shot Semantic Segmentation COCO-20i (5-shot) VAT (ResNet-50) Mean IoU 47.9 # 2
Few-Shot Semantic Segmentation FSS-1000 VAT Mean IoU 90.0 # 1
Few-Shot Semantic Segmentation FSS-1000 (1-shot) VAT Mean IoU 90.0 # 1
Few-Shot Semantic Segmentation FSS-1000 (5-shot) VAT Mean IoU 90.6 # 1
Few-Shot Semantic Segmentation PASCAL-5i (1-Shot) VAT Mean IoU 67.5 # 1
Few-Shot Semantic Segmentation PASCAL-5i (5-Shot) VAT Mean IoU 71.6 # 1
Semantic correspondence PF-PASCAL VAT PCK 92.3 # 2
Semantic correspondence PF-WILLOW VAT PCK 81.0 # 1
Semantic correspondence SPair-71k VAT PCK 54.2 # 1

Methods


No methods listed for this paper. Add relevant methods here