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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| 5-way 1-shot Classification | CD-FSL ISIC, EuroSAT, CropDisease, ChestX (test) | Accuracy (ISIC)32.97 | 86 | |
| 5-way 5-shot Classification | CD-FSL ISIC, EuroSAT, CropDisease, ChestX (test) | Accuracy (ISIC)52.12 | 72 | |
| Few-shot Image Classification | CD-FSL 5-way 1-shot (test) | ChestX Accuracy22.84 | 38 | |
| Few-shot Image Classification | CD-FSL 5-way 5-shot (test) | ChestX Accuracy26.36 | 38 | |
| 5-way 1-shot Few-Shot Classification | BSCD-FSL Suite (ChestX, ISIC, EuroSAT, CropDisease, CUB, Cars, Places, Plantae) 1.0 (test) | ChestX Accuracy0.2284 | 28 | |
| 5-way Few-shot Image Classification | BSCD-FSL ISIC2018, EuroSAT, CropDiseases, ChestX | Accuracy (ISIC)52.12 | 24 | |
| Few-shot Image Classification | BSCD-FSL ChestX, ISIC, EuroSAT, CropDiseases 5-way 1-shot | ChestX Accuracy22.84 | 17 | |
| 5-way 5-shot Classification | BSCD-FSL (test) | Accuracy (ChestX)26.36 | 17 | |
| Few-shot classification | BSCD-FSL 5-way 5-shot | Accuracy (ISIC, 5-way 5-shot)52.12 | 16 | |
| 5-way 1-shot Classification | ChestX | Accuracy (%)22.84 | 14 |