提出新方法缓解视觉语言模型微调时的遗忘问题,提升少样本分类性能。
Complementary Subspace Low-Rank Adaptation of Vision-Language Models for Few-Shot Classification
- 在互补子空间中优化低秩矩阵,保留预训练对齐能力
- 相比基线提升1.0%准确率,零样本性能提升1.3%
- 适合需要高效微调且避免遗忘的少样本视觉任务
视觉语言模型(VLM)作为预训练基础模型,广泛用于大规模图像-文本对齐。针对下游少样本分类任务,参数高效微调(PEFT)方法受到关注。尽管提示调优和线性适配器已被研究,但低秩适应(LoRA)在少样本微调中的应用较少,主要因灾难性遗忘问题:LoRA会干扰预训练权重矩阵中最关键的信息方向,影响模型通用性。本文提出互补子空间低秩适应(Comp-LoRA),将低秩矩阵优化限制在互补子空间,从而在学习新任务时保持原始视觉-语言对齐能力。在少样本分类任务上对比实验表明,该方法相比基线提升约1.0%的Top-1准确率,并使零样本性能提升约1.3%。结果验证了其有效缓解灾难性遗忘的能力。
原文摘要 · Abstract (English)
Vision language model (VLM) has been designed for large scale image-text alignment as a pretrained foundation model. For downstream few shot classification tasks, parameter efficient fine-tuning (PEFT) VLM has gained much popularity in the computer vision community. PEFT methods like prompt tuning and linear adapter have been studied for fine-tuning VLM while low rank adaptation (LoRA) algorithm has rarely been considered for few shot fine-tuning VLM. The main obstacle to use LoRA for few shot fine-tuning is the catastrophic forgetting problem. Because the visual language alignment knowledge is important for the generality in few shot learning, whereas low rank adaptation interferes with the most informative direction of the pretrained weight matrix. We propose the complementary subspace low rank adaptation (Comp-LoRA) method to regularize the catastrophic forgetting problem in few shot VLM finetuning. In detail, we optimize the low rank matrix in the complementary subspace, thus preserving the general vision language alignment ability of VLM when learning the novel few shot information. We conduct comparison experiments of the proposed Comp-LoRA method and other PEFT methods on fine-tuning VLM for few shot classification. And we also present the suppression on the catastrophic forgetting problem of our proposed method against directly applying LoRA to VLM. The results show that the proposed method surpasses the baseline method by about +1.0\% Top-1 accuracy and preserves the VLM zero-shot performance over the baseline method by about +1.3\% Top-1 accuracy.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。