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

Step-wise Distribution Alignment Guided Style Prompt Tuning for Source-free Cross-domain Few-shot Learning

About

Existing cross-domain few-shot learning (CDFSL) methods, which develop source-domain training strategies to enhance model transferability, face challenges with large-scale pre-trained models (LMs) due to inaccessible source data and training strategies. Moreover, fine-tuning LMs for CDFSL demands substantial computational resources, limiting practicality. This paper addresses the source-free CDFSL (SF-CDFSL) problem, tackling few-shot learning (FSL) in the target domain using only pre-trained models and a few target samples without source data or strategies. To overcome the challenge of inaccessible source data, this paper introduces Step-wise Distribution Alignment Guided Style Prompt Tuning (StepSPT), which implicitly narrows domain gaps through prediction distribution optimization. StepSPT proposes a style prompt to align target samples with the desired distribution and adopts a dual-phase optimization process. In the external process, a step-wise distribution alignment strategy factorizes prediction distribution optimization into a multi-step alignment problem to tune the style prompt. In the internal process, the classifier is updated using standard cross-entropy loss. Evaluations on five datasets demonstrate that StepSPT outperforms existing prompt tuning-based methods and SOTAs. Ablation studies further verify its effectiveness. Code will be made publicly available at https://github.com/xuhuali-mxj/StepSPT.

Huali Xu, Li Liu, Tianpeng Liu, Shuaifeng Zhi, Shuzhou Sun, Ming-Ming Cheng• 2024

Related benchmarks

TaskDatasetResultRank
5-way 1-shot ClassificationCD-FSL ISIC, EuroSAT, CropDisease, ChestX (test)
Accuracy (ISIC)32.97
86
5-way 5-shot ClassificationCD-FSL ISIC, EuroSAT, CropDisease, ChestX (test)
Accuracy (ISIC)52.12
72
Few-shot Image ClassificationCD-FSL 5-way 1-shot (test)
ChestX Accuracy22.84
38
Few-shot Image ClassificationCD-FSL 5-way 5-shot (test)
ChestX Accuracy26.36
38
5-way 1-shot Few-Shot ClassificationBSCD-FSL Suite (ChestX, ISIC, EuroSAT, CropDisease, CUB, Cars, Places, Plantae) 1.0 (test)
ChestX Accuracy0.2284
28
5-way Few-shot Image ClassificationBSCD-FSL ISIC2018, EuroSAT, CropDiseases, ChestX
Accuracy (ISIC)52.12
24
Few-shot Image ClassificationBSCD-FSL ChestX, ISIC, EuroSAT, CropDiseases 5-way 1-shot
ChestX Accuracy22.84
17
5-way 5-shot ClassificationBSCD-FSL (test)
Accuracy (ChestX)26.36
17
Few-shot classificationBSCD-FSL 5-way 5-shot
Accuracy (ISIC, 5-way 5-shot)52.12
16
5-way 1-shot ClassificationChestX
Accuracy (%)22.84
14
Showing 10 of 17 rows

Other info

Follow for update