仅用一张图就能预测大模型在冷门领域的表现,省时省力。
Underrepresented in Foundation Model Pretraining Data? A One-Shot Probe
- 用大语言模型生成图像的反事实描述,构造难样本。
- 通过模型区分真实与虚假描述的能力,预测零样本准确率,相关性达0.96。
- 适合资源有限的研究者快速评估模型在非洲等欠发达地区数据上的表现。
大规模视觉-语言基础模型(如CLIP)广泛应用于计算机视觉任务,但在新领域或代表性不足的领域表现不稳定。传统评估需大量标注数据,而小众领域尤其是全球南方地区的数据往往缺乏。本文提出一种高数据效率的方法,仅需每类一个标注图像即可预测基础模型在目标领域的零样本准确率。方法利用大语言模型生成图像的合理反事实描述,通过测量模型区分真实描述与这些困难负样本的能力,构建共享嵌入空间中的判别特征。训练线性回归器,基于相似度得分预测多个视觉领域的零样本准确率,皮尔逊相关系数达0.96。在五个不同数据集上验证,包括标准基准和非洲地区的欠代表数据集。该方法为研究人员提供低成本、可靠的模型探查工具,可在投入大量标注前做出决策。代码、生成的描述和反事实文本已开源:https://github.com/chris-vorster/PreLabellingProbe。
原文摘要 · Abstract (English)
Large-scale Vision-Language Foundation Models (VLFMs), such as CLIP, now underpin a wide range of computer vision research and applications. VLFMs are often adapted to various domain-specific tasks. However, VLFM performance on novel, specialised, or underrepresented domains remains inconsistent. Evaluating VLFMs typically requires labelled test sets, which are often unavailable for niche domains of interest, particularly those from the Global South. We address this gap by proposing a highly data-efficient method to predict a VLFM's zero-shot accuracy on a target domain using only a single labelled image per class. Our approach uses a Large Language Model to generate plausible counterfactual descriptions of a given image. By measuring the VLFM's ability to distinguish the correct description from these hard negatives, we engineer features that capture the VLFM's discriminative power in its shared embedding space. A linear regressor trained on these similarity scores estimates the VLFM's zero-shot test accuracy across various visual domains with a Pearson-r correlation of 0.96. We demonstrate our method's performance across five diverse datasets, including standard benchmark datasets and underrepresented datasets from Africa. Our work provides a low-cost, reliable tool for probing VLFMs, enabling researchers and practitioners to make informed decisions about data annotation efforts before committing significant resources. The model training code, generated captions and counterfactuals are released here: https://github.com/chris-vorster/PreLabellingProbe.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。