HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

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

Yanpeng Sun Qiang Chen Xiangyu He Jian Wang Haocheng Feng Junyu Han Errui Ding Jian Cheng Zechao Li Jingdong Wang

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

Abstract

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 https://github.com/syp2ysy/SVF.

Code Repositories

syp2ysy/SVF
Official
paddle
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
few-shot-semantic-segmentation-on-coco-20i-1BAM (SVF, VGG-16)
Mean IoU: 43.76
few-shot-semantic-segmentation-on-coco-20i-1BAM (SVF, ResNet-50)
Mean IoU: 48.47
few-shot-semantic-segmentation-on-coco-20i-1PFENet (SVF, VGG-16)
Mean IoU: 42.24
few-shot-semantic-segmentation-on-coco-20i-1PFENet (SVF, ResNet-50)
Mean IoU: 48.02
few-shot-semantic-segmentation-on-coco-20i-5PFENet (SVF, ResNet-50)
Mean IoU: 54.38
few-shot-semantic-segmentation-on-coco-20i-5BAM (SVF, ResNet-50)
Mean IoU: 53.87
few-shot-semantic-segmentation-on-coco-20i-5PFENet (SVF, VGG-16)
Mean IoU: 49.49
few-shot-semantic-segmentation-on-coco-20i-5BAM (SVF, VGG-16)
Mean IoU: 49.07
few-shot-semantic-segmentation-on-pascal-5i-1BAM (SVF, VGG-16)
Mean IoU: 64.87
few-shot-semantic-segmentation-on-pascal-5i-1BAM (SVF, ResNet-50)
FB-IoU: 80.13
Mean IoU: 68.95
few-shot-semantic-segmentation-on-pascal-5i-1PFENet (SVF, ResNet-50)
FB-IoU: 79.07
Mean IoU: 68.15
few-shot-semantic-segmentation-on-pascal-5i-1PFENet (SVF, VGG-16)
Mean IoU: 64.33
few-shot-semantic-segmentation-on-pascal-5i-5PFENet (SVF, VGG-16)
Mean IoU: 69.8
few-shot-semantic-segmentation-on-pascal-5i-5BAM (SVF, ResNet-50)
FB-IoU: 83.17
Mean IoU: 72.28
few-shot-semantic-segmentation-on-pascal-5i-5BAM (SVF, VGG-16)
Mean IoU: 69.11
few-shot-semantic-segmentation-on-pascal-5i-5PFENet (SVF, ResNet-50)
FB-IoU: 82.77
Mean IoU: 71.82

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp