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
74
5-way 5-shot ClassificationCD-FSL ISIC, EuroSAT, CropDisease, ChestX (test)
Accuracy (ISIC)52.12
60
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 ClassificationChestX
Accuracy (%)22.84
14
5-way 5-shot ClassificationChestX
Accuracy26.36
14
5-way 1-shot ClassificationCropDisease
Accuracy84.84
14
5-way 5-shot ClassificationISIC
Accuracy52.12
14
5-way 5-shot ClassificationCropDisease
Accuracy96.01
14
5-way 5-shot ClassificationEuroSAT
Accuracy89.4
14
Showing 10 of 12 rows

Other info

Follow for update