arXiv:2511.09973cs.CVcs.AI2025-11中稿 · AAAI被引 1

提出DiVE方法,让视觉语言模型微调后仍保持强泛化能力。

Difference Vector Equalization for Robust Fine-tuning of Vision-Language Models

  • 通过约束样本差异向量相等,保护嵌入空间几何结构。
  • 在ID、OOD和零样本测试上均显著提升性能,零样本准确率提高2.1%。
  • 适合需要微调又不牺牲泛化性的视觉语言模型研究者使用。

对比预训练的视觉语言模型(如CLIP)通过图像与文本编码器提取的嵌入,在零样本分类中表现出强大的泛化能力。本文旨在对分布内(ID)数据稳健微调这些模型,同时不损害其在分布外(OOD)和零样本设置下的泛化能力。现有鲁棒微调方法依赖于预训练阶段使用的对比学习进行微调,但我们发现这些方法会扭曲嵌入的几何结构,影响模型泛化性能。为此,我们提出差异向量等化(DiVE),在微调过程中保持嵌入结构不变。DiVE的核心思想是约束每个样本的差异向量——即同一数据样本在预训练模型与微调模型下嵌入的差值——使其在不同样本间相等。为此,我们引入两种损失:平均向量损失(AVL)通过使差异向量等于其加权平均来全局保持结构;配对向量损失(PVL)通过确保多模态对齐的一致性来局部保持结构。实验表明,DiVE能有效保持嵌入几何结构,在分布内、分布外及零样本任务上均取得优异表现。

原文摘要 · Abstract (English)

Contrastive pre-trained vision-language models, such as CLIP, demonstrate strong generalization abilities in zero-shot classification by leveraging embeddings extracted from image and text encoders. This paper aims to robustly fine-tune these vision-language models on in-distribution (ID) data without compromising their generalization abilities in out-of-distribution (OOD) and zero-shot settings. Current robust fine-tuning methods tackle this challenge by reusing contrastive learning, which was used in pre-training, for fine-tuning. However, we found that these methods distort the geometric structure of the embeddings, which plays a crucial role in the generalization of vision-language models, resulting in limited OOD and zero-shot performance. To address this, we propose Difference Vector Equalization (DiVE), which preserves the geometric structure during fine-tuning. The idea behind DiVE is to constrain difference vectors, each of which is obtained by subtracting the embeddings extracted from the pre-trained and fine-tuning models for the same data sample. By constraining the difference vectors to be equal across various data samples, we effectively preserve the geometric structure. Therefore, we introduce two losses: average vector loss (AVL) and pairwise vector loss (PVL). AVL preserves the geometric structure globally by constraining difference vectors to be equal to their weighted average. PVL preserves the geometric structure locally by ensuring a consistent multimodal alignment. Our experiments demonstrate that DiVE effectively preserves the geometric structure, achieving strong results across ID, OOD, and zero-shot metrics.

视觉语言模型微调泛化能力

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。