arXiv:2412.04378cs.CVcs.AI2024-12CVPR被引 17

将生成式大模型改造为判别任务利器,提升图文识别与语言理解能力。

VladVA: Discriminative Fine-tuning of LVLMs

  • 通过对比损失与下一个词预测结合,训练可判别图文的视觉语言模型。
  • 在相同规模下超越CLIP类模型,在组合性任务上显著提升。
  • 采用轻量级提示与LoRA适配器,高效实现模型改造,适合快速部署。

对比训练的视觉语言模型(如CLIP)已成为判别式视觉语言表示学习的标准方法,但其语言理解能力有限,常表现出“词袋”行为。与此同时,结合视觉编码器与大语言模型的大型视觉语言模型(LVLM)具备强大的视觉-语言推理能力,但其自回归特性使其不适合判别任务。本文提出一种新的判别微调方法,将生成式LVLM转化为判别模型,兼具强判别力与增强的语言理解能力。贡献包括:(1) 设计了一种融合可变长度与粒度图像-文本对的训练框架,同时使用对比损失与下一个词预测损失,并通过消融实验验证各组件必要性;(2) 提出基于软提示与LoRA适配器的参数高效适配方法;(3) 在同等规模下显著优于现有CLIP类模型,涵盖标准图文检索基准与组合性任务中的显著提升。

原文摘要 · Abstract (English)

Contrastively-trained Vision-Language Models (VLMs) like CLIP have become the de facto approach for discriminative vision-language representation learning. However, these models have limited language understanding, often exhibiting a "bag of words" behavior. At the same time, Large Vision-Language Models (LVLMs), which combine vision encoders with LLMs, have been shown to be capable of detailed vision-language reasoning, yet their autoregressive nature renders them less suitable for discriminative tasks. In this work, we propose to combine "the best of both worlds": a new training approach for discriminative fine-tuning of LVLMs that results in strong discriminative and compositional capabilities. Essentially, our approach converts a generative LVLM into a discriminative one, unlocking its capability for powerful image-text discrimination combined with enhanced language understanding. Our contributions include (1) a carefully designed training/optimization framework that utilizes image-text pairs of variable length and granularity for training the model with both contrastive and next-token prediction losses. This is accompanied by ablation studies that justify the necessity of our framework's components; (2) a parameter-efficient adaptation method using a combination of soft prompting and LoRA adapters; (3) significant improvements over state-of-the-art CLIP-like models of similar size, including standard image-text retrieval benchmarks and notable gains in compositionality.

视觉语言模型判别任务轻量化微调LoRA

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