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

Yo'LLaVA: Your Personalized Language and Vision Assistant

About

Large Multimodal Models (LMMs) have shown remarkable capabilities across a variety of tasks (e.g., image captioning, visual question answering). While broad, their knowledge remains generic (e.g., recognizing a dog), and they are unable to handle personalized subjects (e.g., recognizing a user's pet dog). Human reasoning, in contrast, typically operates within the context of specific subjects in our surroundings. For example, one might ask, "What should I buy for my dog's birthday?"; as opposed to a generic inquiry about "What should I buy for a dog's birthday?". Similarly, when looking at a friend's image, the interest lies in seeing their activities (e.g., "my friend is holding a cat"), rather than merely observing generic human actions (e.g., "a man is holding a cat"). In this paper, we introduce the novel task of personalizing LMMs, so that they can have conversations about a specific subject. We propose Yo'LLaVA, which learns to embed a personalized subject into a set of latent tokens given a handful of example images of the subject. Our qualitative and quantitative analyses reveal that Yo'LLaVA can learn the concept more efficiently using fewer tokens and more effectively encode the visual attributes compared to strong prompting baselines (e.g., LLaVA).

Thao Nguyen, Haotian Liu, Yuheng Li, Mu Cai, Utkarsh Ojha, Yong Jae Lee• 2024

Related benchmarks

TaskDatasetResultRank
Personalized UnderstandingOmniPBench
Rec Weight0.919
14
MLLM PersonalizationLCMP-E
ACC-C42.08
12
MLLM PersonalizationLCMP-H
ACC-C30.56
12
RecognitionMC-LLaVA
Single Score84.1
11
RecognitionYo'LLaVA
Rec. Single92.4
11
RecognitionMyVLM
Single Recall96.4
11
Visual Question AnsweringMC-LLaVA
VQA BLEU (Single)64.3
11
Multiple-choice Question AnsweringYo'LLaVA
Choice-V & T Accuracy (Single)89.6
11
Visual GroundingMC-LLaVA
VG Score0.702
11
Image CaptioningMyVLM
Caption Recall (Single)0.931
11
Showing 10 of 31 rows

Other info

Code

Follow for update