Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

"This is my unicorn, Fluffy": Personalizing frozen vision-language representations

About

Large Vision & Language models pretrained on web-scale data provide representations that are invaluable for numerous V&L problems. However, it is unclear how they can be used for reasoning about user-specific visual concepts in unstructured language. This problem arises in multiple domains, from personalized image retrieval to personalized interaction with smart devices. We introduce a new learning setup called Personalized Vision & Language (PerVL) with two new benchmark datasets for retrieving and segmenting user-specific "personalized" concepts "in the wild". In PerVL, one should learn personalized concepts (1) independently of the downstream task (2) allowing a pretrained model to reason about them with free language, and (3) does not require personalized negative examples. We propose an architecture for solving PerVL that operates by extending the input vocabulary of a pretrained model with new word embeddings for the new personalized concepts. The model can then reason about them by simply using them in a sentence. We demonstrate that our approach learns personalized visual concepts from a few examples and can effectively apply them in image retrieval and semantic segmentation using rich textual queries.

Niv Cohen, Rinon Gal, Eli A. Meirom, Gal Chechik, Yuval Atzmon• 2022

Related benchmarks

TaskDatasetResultRank
Composed Image RetrievalCIRR (test)
Recall@141.61
481
Composed Image RetrievalFashionIQ (val)
Shirt Recall@1021.5
455
Composed Image RetrievalCIRCO (test)
mAP@105.32
234
Composed Image RetrievalFashion-IQ (test)
Dress Recall@100.1725
145
Composed Image Retrieval (Image-Text to Image)CIRR
Recall@116.6
75
Composed Image RetrievalCIRCO
mAP@54.6
63
Compositional Image RetrievalFashionIQ 1.0 (val)
Average Recall@1019.8
42
Composed Image RetrievalCIRCO 1.0 (test)
mAP@54.6
36
Composed Image RetrievalCIRR Index Set v1.0
Recall@116.6
23
Composed Image RetrievalCIRR v1.0
Recall@141.6
21
Showing 10 of 18 rows

Other info

Code

Follow for update