Attributes Should Come from Images, Not Class Names: Distribution-Conditioned Attribute Selection for Vision-Language Models
Summary
A new study reveals that attributes for interpretable zero-shot classification in Vision-Language Models (VLMs) like CLIP should be derived from images, not solely from class names. Existing methods, which use large language models (LLMs) to generate class descriptors, show a significant accuracy drop from 59.5% to 15.5% on ImageNet when the class name is removed from the prompt. This indicates LLM descriptors are conditioned on labels, not visual evidence, leading to failures in data shift scenarios, such as colorless strawberries in ImageNet-Sketch. Researchers propose a "distribution-conditioned attribute selection" method, scoring a large attribute pool against target images in CLIP's embedding space. This approach achieves 23.8% accuracy on ImageNet with class-name-free prompts, outperforming LLM descriptors (15.5%) and maintaining gains across four shifted ImageNet variants. The method also surpasses prompt-tuning technique CoOp by 3 points using one image per class, completing in under a minute compared to CoOp's 14 hours, and provides a readable dataset summary for understanding distribution shifts.
Key takeaway
For Machine Learning Engineers developing interpretable zero-shot classification systems, you should prioritize deriving attributes directly from image collections rather than relying solely on LLM-generated class descriptions. This approach significantly improves VLM accuracy and robustness against data shifts, as demonstrated by a 3-point gain over CoOp and faster adaptation times (under a minute vs. 14 hours). Consider implementing distribution-conditioned attribute selection to enhance model reliability and gain clearer insights into dataset characteristics.
Key insights
LLM-generated class attributes are label-conditioned; image-derived attributes improve VLM zero-shot classification and robustness.
Principles
- Attributes should reflect visual data, not just class names.
- Data-driven attribute selection enhances VLM robustness.
- Explicit attribute sets can summarize dataset characteristics.
Method
Score a large attribute pool against target images in CLIP's joint embedding space, selecting top-scoring attributes per class.
In practice
- Use image-conditioned attributes for robust zero-shot VLM inference.
- Generate dataset summaries by analyzing selected attributes.
- Replace prompt-tuning with attribute selection for faster VLM adaptation.
Topics
- Vision-Language Models
- Zero-shot Classification
- CLIP
- Attribute Selection
- Data Shift
- Interpretability
Best for: Research Scientist, AI Scientist, Computer Vision Engineer, Machine Learning Engineer
Related on AIssential
See Counsel's argued verdicts on the open AI decisions leaders are weighing →
Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.