Understanding Visual Concepts Across Models

Brandon Trabucco, Max Gurinas, Kyle Doherty, Ruslan Salakhutdinov
6/11/2024
cs.CVcs.AIcs.LG

No official code found. Best implementation: visual-words/visual-words (7 stars, MIT, last commit 2024-06-12).

Abstract

Large multimodal models such as Stable Diffusion can generate, detect, and classify new visual concepts after fine-tuning just a single word embedding. Do models learn similar words for the same concepts (i.e. <orange-cat> = orange + cat)? We conduct a large-scale analysis on three state-of-the-art models in text-to-image generation, open-set object detection, and zero-shot classification, and find that new word embeddings are model-specific and non-transferable. Across 4,800 new embeddings trained for 40 diverse visual concepts on four standard datasets, we find perturbations within an $ε$-ball to any prior embedding that generate, detect, and classify an arbitrary concept. When these new embeddings are spliced into new models, fine-tuning that targets the original model is lost. We show popular soft prompt-tuning approaches find these perturbative solutions when applied to visual concept learning tasks, and embeddings for visual concepts are not transferable. Code for reproducing our work is available at: https://visual-words.github.io.

View on arXivView PDF

Code Implementations(1)

MIT

Official code for "Understanding Visual Concepts Across Models"

70Jun 11, 20242 years agoMIT

Similar Papers

Cite this paper

@article{trabucco2024understanding,
  title  = {Understanding Visual Concepts Across Models},
  author = {Brandon Trabucco and Max Gurinas and Kyle Doherty and Ruslan Salakhutdinov},
  year   = {2024},
  eprint = {2406.07506},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2406.07506v1}
}

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/2406.07506.svg)](https://codeofpaper.com/paper/2406.07506)

Discussion