Seonghyeon Moon, Samuel S. Sohn, Honglu Zhou, Sejong Yoon, Vladimir Pavlovic, Muhammad Haris Khan, Mubbasir Kapadia
FSS(Few-shot segmentation) aims to segment a target class using a small number of labeled images(support set). To extract information relevant to the target class, a dominant approach in best-performing FSS methods removes background features using a support mask. We observe that this feature excision through a limiting support mask introduces an information bottleneck in several challenging FSS cases, e.g., for small targets and/or inaccurate target boundaries. To this end, we present a novel method(MSI), which maximizes the support-set information by exploiting two complementary sources of features to generate super correlation maps. We validate the effectiveness of our approach by instantiating it into three recent and strong FSS methods. Experimental results on several publicly available FSS benchmarks show that our proposed method consistently improves performance by visible margins and leads to faster convergence. Our code and trained models are available at: https://github.com/moonsh/MSI-Maximize-Support-Set-Information
| Task | Dataset | Metric | Value | Model |
|---|---|---|---|---|
| Few-Shot Learning | COCO-20i -> Pascal VOC (1-shot) | Mean IoU | 69.2 | VAT + MSI (ResNet101) |
| Few-Shot Learning | FSS-1000 (1-shot) | Mean IoU | 90.6 | VAT + MSI (ResNet-101) |
| Few-Shot Learning | PASCAL-5i (1-Shot) | FB-IoU | 82.3 | VAT + MSI (ResNet-101) |
| Few-Shot Learning | PASCAL-5i (1-Shot) | Mean IoU | 70.1 | VAT + MSI (ResNet-101) |
| Few-Shot Learning | COCO-20i (1-shot) | Mean IoU | 49.8 | VAT + MSI (ResNet-101) |
| Few-Shot Semantic Segmentation | COCO-20i -> Pascal VOC (1-shot) | Mean IoU | 69.2 | VAT + MSI (ResNet101) |
| Few-Shot Semantic Segmentation | FSS-1000 (1-shot) | Mean IoU | 90.6 | VAT + MSI (ResNet-101) |
| Few-Shot Semantic Segmentation | PASCAL-5i (1-Shot) | FB-IoU | 82.3 | VAT + MSI (ResNet-101) |
| Few-Shot Semantic Segmentation | PASCAL-5i (1-Shot) | Mean IoU | 70.1 | VAT + MSI (ResNet-101) |
| Few-Shot Semantic Segmentation | COCO-20i (1-shot) | Mean IoU | 49.8 | VAT + MSI (ResNet-101) |
| Meta-Learning | COCO-20i -> Pascal VOC (1-shot) | Mean IoU | 69.2 | VAT + MSI (ResNet101) |
| Meta-Learning | FSS-1000 (1-shot) | Mean IoU | 90.6 | VAT + MSI (ResNet-101) |
| Meta-Learning | PASCAL-5i (1-Shot) | FB-IoU | 82.3 | VAT + MSI (ResNet-101) |
| Meta-Learning | PASCAL-5i (1-Shot) | Mean IoU | 70.1 | VAT + MSI (ResNet-101) |
| Meta-Learning | COCO-20i (1-shot) | Mean IoU | 49.8 | VAT + MSI (ResNet-101) |