arXiv:2412.20682cs.CVcs.LG2024-12被引 1

无需标注数据,自动选出最适合下游任务的视觉语言模型。

Learning to Rank Pre-trained Vision-Language Models for Downstream Tasks

  • 通过构建视觉与文本特征图,计算双模态图的相似性来评估模型性能。
  • 在三个基准上验证,能准确预测模型在未标注数据上的表现。
  • 适合资源受限场景下快速选择最佳视觉语言模型。

视觉语言模型(如CLIP)在分类任务中表现出色的零样本能力。然而,在未标注的下游任务中选择最优模型仍具挑战。现有方法依赖类别名和有监督大规模数据集,部署时可能不可行。本文提出无监督视觉语言模型选择问题,仅使用无标注下游数据。为此,我们提出视觉-文本图对齐方法(VEGA),通过衡量模型在下游任务中双模态特征的对齐程度来选型,无需任何标注。该方法基于双模态特征在预训练中对齐的原理,将视觉与文本特征分别建图,以节点和边级别的整体相似性定义VEGA。在三个不同基准、多种应用场景和数据集上的实验表明,VEGA能持续可靠地估计模型在未标注任务中的性能。

原文摘要 · Abstract (English)

Vision language models (VLMs) like CLIP show stellar zero-shot capability on classification benchmarks. However, selecting the VLM with the highest performance on the unlabeled downstream task is non-trivial. Existing VLM selection methods focus on the class-name-only setting, relying on a supervised large-scale dataset and large language models, which may not be accessible or feasible during deployment. This paper introduces the problem of \textbf{unsupervised vision-language model selection}, where only unsupervised downstream datasets are available, with no additional information provided. To solve this problem, we propose a method termed Visual-tExtual Graph Alignment (VEGA), to select VLMs without any annotations by measuring the alignment of the VLM between the two modalities on the downstream task. VEGA is motivated by the pretraining paradigm of VLMs, which aligns features with the same semantics from the visual and textual modalities, thereby mapping both modalities into a shared representation space. Specifically, we first construct two graphs on the vision and textual features, respectively. VEGA is then defined as the overall similarity between the visual and textual graphs at both node and edge levels. Extensive experiments across three different benchmarks, covering a variety of application scenarios and downstream datasets, demonstrate that VEGA consistently provides reliable and accurate estimates of VLMs' performance on unlabeled downstream tasks.

视觉语言模型模型选择无监督学习

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