The Devil is in Low-Level Features for Cross-Domain Few-Shot Segmentation
About
Cross-Domain Few-Shot Segmentation (CDFSS) is proposed to transfer the pixel-level segmentation capabilities learned from large-scale source-domain datasets to downstream target-domain datasets, with only a few annotated images per class. In this paper, we focus on a well-observed but unresolved phenomenon in CDFSS: for target domains, particularly those distant from the source domain, segmentation performance peaks at the very early epochs, and declines sharply as the source-domain training proceeds. We delve into this phenomenon for an interpretation: low-level features are vulnerable to domain shifts, leading to sharper loss landscapes during the source-domain training, which is the devil of CDFSS. Based on this phenomenon and interpretation, we further propose a method that includes two plug-and-play modules: one to flatten the loss landscapes for low-level features during source-domain training as a novel sharpness-aware minimization method, and the other to directly supplement target-domain information to the model during target-domain testing by low-level-based calibration. Extensive experiments on four target datasets validate our rationale and demonstrate that our method surpasses the state-of-the-art method in CDFSS signifcantly by 3.71% and 5.34% average MIoU in 1-shot and 5-shot scenarios, respectively.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Few-shot Segmentation | DeepGlobe | mIoU49.7 | 83 | |
| Few-shot Segmentation | Chest X-ray | mIoU82.7 | 82 | |
| Few-shot Semantic Segmentation | Average Deepglobe, ISIC, Chest X-Ray, FSS-1000 | mIoU65 | 54 | |
| Few-shot Segmentation | ISIC 2018 | mIoU62.4 | 51 | |
| Few-shot Semantic Segmentation | CD-FSS 1-shot 1.0 (test) | mIoU (Average)65 | 34 | |
| Semantic segmentation | FSS-1000 1-shot | mIoU81.1 | 32 | |
| Semantic segmentation | FSS-1000 5-shot | mIoU83.7 | 29 | |
| Few-shot Semantic Segmentation | Chest X-ray 1-shot | mIoU83.9 | 22 | |
| Few-shot Semantic Segmentation | DeepGlobe 5-shot | mIoU51.5 | 22 | |
| Few-shot Semantic Segmentation | ISIC 1-shot 2018 | mIoU52.9 | 22 |