提出两阶段少样本适配新方法,提升视觉语言模型在少量数据下的表现。
Rethinking Few-Shot Adaptation of Vision-Language Models in Two Stages
- 分两阶段:先用参数高效微调学任务特征,再训练线性分类器
- 在11个数据集上表现优于现有方法,尤其在跨设置时更稳定
- 适合资源有限且需快速适配新类别的实际应用场景
少样本适配(FSA)中数据稀缺导致大型预训练视觉语言模型难以有效微调。本文分析参数高效微调(PEFT)在少数类别上的学习动态,发现其自然分为两个阶段:任务级特征提取与概念专化。基于此,提出两阶段少样本适配(2SFS):在固定计算预算下,先通过PEFT学习任务特定特征提取器,再在其上训练线性分类器。该方法实现类别级别的选择性推理——测试时仅对新类别使用适配后的文本编码器嵌入,基类嵌入则保留在分类器中。在三个主干网络、两个设置和十一个数据集上,2SFS在固定超参下达到或超越当前最优,而传统方法性能显著下降。
原文摘要 · Abstract (English)
An old-school recipe for training a classifier is to (i) learn a good feature extractor and (ii) optimize a linear layer atop. When only a handful of samples are available per category, as in Few-Shot Adaptation (FSA), data are insufficient to fit a large number of parameters, rendering the above impractical. This is especially true with large pre-trained Vision-Language Models (VLMs), which motivated successful research at the intersection of Parameter-Efficient Fine-tuning (PEFT) and FSA. In this work, we start by analyzing the learning dynamics of PEFT techniques when trained on few-shot data from only a subset of categories, referred to as the ``base'' classes. We show that such dynamics naturally splits into two distinct phases: (i) task-level feature extraction and (ii) specialization to the available concepts. To accommodate this dynamic, we then depart from prompt- or adapter-based methods and tackle FSA differently. Specifically, given a fixed computational budget, we split it to (i) learn a task-specific feature extractor via PEFT and (ii) train a linear classifier on top. We call this scheme Two-Stage Few-Shot Adaptation (2SFS). Differently from established methods, our scheme enables a novel form of selective inference at a category level, i.e., at test time, only novel categories are embedded by the adapted text encoder, while embeddings of base categories are available within the classifier. Results with fixed hyperparameters across two settings, three backbones, and eleven datasets, show that 2SFS matches or surpasses the state-of-the-art, while established methods degrade significantly across settings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。