Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Cross-Domain Few-Shot Segmentation via Iterative Support-Query Correspondence Mining

About

Cross-Domain Few-Shot Segmentation (CD-FSS) poses the challenge of segmenting novel categories from a distinct domain using only limited exemplars. In this paper, we undertake a comprehensive study of CD-FSS and uncover two crucial insights: (i) the necessity of a fine-tuning stage to effectively transfer the learned meta-knowledge across domains, and (ii) the overfitting risk during the na\"ive fine-tuning due to the scarcity of novel category examples. With these insights, we propose a novel cross-domain fine-tuning strategy that addresses the challenging CD-FSS tasks. We first design Bi-directional Few-shot Prediction (BFP), which establishes support-query correspondence in a bi-directional manner, crafting augmented supervision to reduce the overfitting risk. Then we further extend BFP into Iterative Few-shot Adaptor (IFA), which is a recursive framework to capture the support-query correspondence iteratively, targeting maximal exploitation of supervisory signals from the sparse novel category samples. Extensive empirical evaluations show that our method significantly outperforms the state-of-the-arts (+7.8\%), which verifies that IFA tackles the cross-domain challenges and mitigates the overfitting simultaneously. The code is available at: https://github.com/niejiahao1998/IFA.

Jiahao Nie, Yun Xing, Gongjie Zhang, Pei Yan, Aoran Xiao, Yap-Peng Tan, Alex C. Kot, Shijian Lu• 2024

Related benchmarks

TaskDatasetResultRank
Semantic segmentationLoveDA
mIoU46.38
192
Semantic segmentationiSAID--
146
Few-shot SegmentationDeepGlobe
mIoU58.8
83
Few-shot SegmentationChest X-ray
mIoU74.6
82
Medical Image SegmentationISIC
DICE42.59
79
Few-shot Semantic SegmentationCOCO-20i -> PASCAL-5i cross-dataset
mIoU79.6
70
Few-shot Semantic SegmentationFSS-1000
mIoU82.4
64
Few-shot Semantic SegmentationAverage Deepglobe, ISIC, Chest X-Ray, FSS-1000
mIoU71.4
54
Few-shot SegmentationISIC 2018
mIoU69.8
51
Few-shot Semantic SegmentationCD-FSS 1-shot 1.0 (test)
mIoU (Average)67.8
34
Showing 10 of 62 rows

Other info

Follow for update