Hierarchical Dense Correlation Distillation for Few-Shot Segmentation

Few-shot semantic segmentation (FSS) aims to form class-agnostic models segmenting unseen classes with only a handful of annotations. Previous methods limited to the semantic feature and prototype representation suffer from coarse segmentation granularity and train-set overfitting. In this work, we design Hierarchically Decoupled Matching Network (HDMNet) mining pixel-level support correlation based on the transformer architecture. The self-attention modules are used to assist in establishing hierarchical dense features, as a means to accomplish the cascade matching between query and support features. Moreover, we propose a matching module to reduce train-set overfitting and introduce correlation distillation leveraging semantic correspondence from coarse resolution to boost fine-grained segmentation. Our method performs decently in experiments. We achieve $50.0\%$ mIoU on \coco~dataset one-shot setting and $56.0\%$ on five-shot segmentation, respectively.

PDF Abstract CVPR 2023 PDF CVPR 2023 Abstract
Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Few-Shot Semantic Segmentation COCO-20i (1-shot) HDMNet (ResNet-50) Mean IoU 50 # 6
FB-IoU 72.2 # 4
Few-Shot Semantic Segmentation COCO-20i (1-shot) HDMNet (VGG-16) Mean IoU 45.9 # 21
Few-Shot Semantic Segmentation COCO-20i (5-shot) HDMNet (ResNet-50) Mean IoU 56 # 10
FB-IoU 77.7 # 2
Few-Shot Semantic Segmentation COCO-20i (5-shot) HDMNet (VGG-16) Mean IoU 52.4 # 18
Few-Shot Semantic Segmentation PASCAL-5i (1-Shot) HDMNet (VGG-16) Mean IoU 65.1 # 45
Few-Shot Semantic Segmentation PASCAL-5i (1-Shot) HDMNet (ResNet-50) Mean IoU 69.4 # 7
Few-Shot Semantic Segmentation PASCAL-5i (5-Shot) HDMNet (ResNet-50) Mean IoU 71.8 # 19
Few-Shot Semantic Segmentation PASCAL-5i (5-Shot) HDMNet (VGG-16) Mean IoU 69.3 # 41

Methods


No methods listed for this paper. Add relevant methods here