Attributes Should Come from Images, Not Class Names: Distribution-Conditioned Attribute Selection for Vision-Language Models

· Source: Artificial Intelligence · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Computer Vision · Depth: Expert, quick

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

Method

Score a large attribute pool against target images in CLIP's joint embedding space, selecting top-scoring attributes per class.

In practice

Topics

Best for: Research Scientist, AI Scientist, Computer Vision Engineer, Machine Learning Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.