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 | 74 | |
| 5-way 5-shot Classification | CD-FSL ISIC, EuroSAT, CropDisease, ChestX (test) | Accuracy (ISIC)52.12 | 60 | |
| 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 Classification | ChestX | Accuracy (%)22.84 | 14 | |
| 5-way 5-shot Classification | ChestX | Accuracy26.36 | 14 | |
| 5-way 1-shot Classification | CropDisease | Accuracy84.84 | 14 | |
| 5-way 5-shot Classification | ISIC | Accuracy52.12 | 14 | |
| 5-way 5-shot Classification | CropDisease | Accuracy96.01 | 14 | |
| 5-way 5-shot Classification | EuroSAT | Accuracy89.4 | 14 |