Understanding Visual Concepts Across Models

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Trabucco, Brandon, Gurinas, Max, Doherty, Kyle, Salakhutdinov, Ruslan
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866929382233210880
author Trabucco, Brandon
Gurinas, Max
Doherty, Kyle
Salakhutdinov, Ruslan
author_facet Trabucco, Brandon
Gurinas, Max
Doherty, Kyle
Salakhutdinov, Ruslan
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2406_07506
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Understanding Visual Concepts Across Models
Trabucco, Brandon
Gurinas, Max
Doherty, Kyle
Salakhutdinov, Ruslan
Computer Vision and Pattern Recognition
Artificial Intelligence
Machine Learning
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.
title Understanding Visual Concepts Across Models
topic Computer Vision and Pattern Recognition
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2406.07506