Singular Value Fine-tuning: Few-shot Segmentation requires Few-parameters Fine-tuning

Freezing the pre-trained backbone has become a standard paradigm to avoid overfitting in few-shot segmentation. In this paper, we rethink the paradigm and explore a new regime: {\em fine-tuning a small part of parameters in the backbone}. We present a solution to overcome the overfitting problem, leading to better model generalization on learning novel classes. Our method decomposes backbone parameters into three successive matrices via the Singular Value Decomposition (SVD), then {\em only fine-tunes the singular values} and keeps others frozen. The above design allows the model to adjust feature representations on novel classes while maintaining semantic clues within the pre-trained backbone. We evaluate our {\em Singular Value Fine-tuning (SVF)} approach on various few-shot segmentation methods with different backbones. We achieve state-of-the-art results on both Pascal-5$^i$ and COCO-20$^i$ across 1-shot and 5-shot settings. Hopefully, this simple baseline will encourage researchers to rethink the role of backbone fine-tuning in few-shot settings. The source code and models will be available at \url{https://github.com/syp2ysy/SVF}.

PDF Abstract

Datasets


Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Few-Shot Semantic Segmentation COCO-20i (1-shot) BAM (SVF, ResNet-50) Mean IoU 48.47 # 8
Few-Shot Semantic Segmentation COCO-20i (1-shot) PFENet (SVF, ResNet-50) Mean IoU 48.02 # 10
Few-Shot Semantic Segmentation COCO-20i (1-shot) BAM (SVF, VGG-16) Mean IoU 43.76 # 30
Few-Shot Semantic Segmentation COCO-20i (1-shot) PFENet (SVF, VGG-16) Mean IoU 42.24 # 42
Few-Shot Semantic Segmentation COCO-20i (5-shot) BAM (SVF, VGG-16) Mean IoU 49.07 # 36
Few-Shot Semantic Segmentation COCO-20i (5-shot) BAM (SVF, ResNet-50) Mean IoU 53.87 # 14
Few-Shot Semantic Segmentation COCO-20i (5-shot) PFENet (SVF, ResNet-50) Mean IoU 54.38 # 12
Few-Shot Semantic Segmentation COCO-20i (5-shot) PFENet (SVF, VGG-16) Mean IoU 49.49 # 33
Few-Shot Semantic Segmentation PASCAL-5i (1-Shot) PFENet (SVF, VGG-16) Mean IoU 64.33 # 52
Few-Shot Semantic Segmentation PASCAL-5i (1-Shot) BAM (SVF, ResNet-50) Mean IoU 68.95 # 12
FB-IoU 80.13 # 7
Few-Shot Semantic Segmentation PASCAL-5i (1-Shot) PFENet (SVF, ResNet-50) Mean IoU 68.15 # 17
FB-IoU 79.07 # 13
Few-Shot Semantic Segmentation PASCAL-5i (1-Shot) BAM (SVF, VGG-16) Mean IoU 64.87 # 47
Few-Shot Semantic Segmentation PASCAL-5i (5-Shot) BAM (SVF, VGG-16) Mean IoU 69.11 # 46
Few-Shot Semantic Segmentation PASCAL-5i (5-Shot) BAM (SVF, ResNet-50) Mean IoU 72.28 # 15
FB-IoU 83.17 # 6
Few-Shot Semantic Segmentation PASCAL-5i (5-Shot) PFENet (SVF, ResNet-50) Mean IoU 71.82 # 18
FB-IoU 82.77 # 9
Few-Shot Semantic Segmentation PASCAL-5i (5-Shot) PFENet (SVF, VGG-16) Mean IoU 69.8 # 38

Methods


No methods listed for this paper. Add relevant methods here