Pre-trained Vision-Language Models Learn Discoverable Visual Concepts

Yuan Zang, Tian Yun, Hao Tan, Trung Bui, Chen Sun
4/19/2024
cs.CVcs.AIcs.CL

No official code found. Best implementation: brown-palm/Concept-Discovery-and-Learning (21 stars, MIT, last commit 2024-06-05).

Abstract

Do vision-language models (VLMs) pre-trained to caption an image of a "durian" learn visual concepts such as "brown" (color) and "spiky" (texture) at the same time? We aim to answer this question as visual concepts learned "for free" would enable wide applications such as neuro-symbolic reasoning or human-interpretable object classification. We assume that the visual concepts, if captured by pre-trained VLMs, can be extracted by their vision-language interface with text-based concept prompts. We observe that recent works prompting VLMs with concepts often differ in their strategies to define and evaluate the visual concepts, leading to conflicting conclusions. We propose a new concept definition strategy based on two observations: First, certain concept prompts include shortcuts that recognize correct concepts for wrong reasons; Second, multimodal information (e.g. visual discriminativeness, and textual knowledge) should be leveraged when selecting the concepts. Our proposed concept discovery and learning (CDL) framework is thus designed to identify a diverse list of generic visual concepts (e.g. "spiky" as opposed to "spiky durian"), which are ranked and selected based on visual and language mutual information. We carefully design quantitative and human evaluations of the discovered concepts on six diverse visual recognition datasets, which confirm that pre-trained VLMs do learn visual concepts that provide accurate and thorough descriptions for the recognized objects. All code and models are publicly released.

View on arXivView PDF

Code Implementations(1)

MIT

The code for the paper "Pre-trained Vision-Language Models Learn Discoverable Concepts"

210PythonApr 17, 20242 years agoMIT

Similar Papers

Cite this paper

@article{zang2024pretrained,
  title  = {Pre-trained Vision-Language Models Learn Discoverable Visual Concepts},
  author = {Yuan Zang and Tian Yun and Hao Tan and Trung Bui and Chen Sun},
  year   = {2024},
  eprint = {2404.12652},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2404.12652v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2404.12652.svg)](https://codeofpaper.com/paper/2404.12652)

Discussion