Underrepresented in Foundation Model Pretraining Data? A One-Shot Probe
Organizations: ML-Labs, Dublin City University, Dublin, Ireland
Abstract
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.
Figures & tables
| Dataset | Task | Classes | Focus |
| CIFAR-10 [ 27 ] | Object classification | 10 | Ten common object categories (airplane, automobile, bird, etc.) in low-resolution natural scenes. |
| CIFAR-100 [ 27 ] | Object classification | 100 | One hundred fine-grained classes grouped into 20 super-classes (e.g., maples, lobsters). |
| Oxford 102 Flowers [ 28 ] | Flower classification | 102 | Photographs of 102 flower species with diverse backgrounds, lighting, and viewpoints. |
| Food-101 [ 29 ] | Food classification | 101 | Real-world dishes (e.g., pizza, sushi, ramen) with varied preparation and noisy backgrounds. |
| Caltech-101 [ 30 ] | Object classification | 101 | Images of objects from 101 categories (e.g., faces, motorcycles, crabs) with varying scale, position, and background clutter. |
| Oxford-IIIT Pet [ 31 ] | Pet breed classification | 37 | Cat and dog breeds with diverse poses, backgrounds, and bounding box annotations. |
| Variant | Pearson-r | RMSE |
| LLM-generated only | 0.848922 | 0.144607 |
| Vanilla CLIP prompt-based only | 0.947374 | 0.150432 |
| PreLabellingProbe | 0.961896 | 0.103692 |
| Dataset | CLIP ZS Accuracy (Full testset) | PreLabellingProbe (1-shot) | Error |
| African Food | 38.24 | 41.22 | 2.98 |
| Beans | 39.84 | 26.12 | -13.72 |
| Caltech101 | 89.25 | 84.86 | -4.39 |
| CIFAR10 | 91.68 | 76.41 | -15.27 |
| Food101 | 83.76 | 74.38 | -9.38 |
VLM Evaluation
Vision-language model evaluation measures multimodal systems’ perception, reasoning, grounding, robustness, and reliability across controlled and real-world tasks.
See all 989 papers