MC-LLaVA: Multi-Concept Personalized Vision-Language Model

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: An, Ruichuan, Yang, Sihan, Zhang, Renrui, Lu, Ming, Jiang, Tianyi, Zeng, Kai, Luo, Yulin, Cao, Jiajun, Liang, Hao, Chen, Ying, She, Qi, Zhang, Shanghang, Zhang, Wentao
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866910025885155328
author An, Ruichuan
Yang, Sihan
Zhang, Renrui
Lu, Ming
Jiang, Tianyi
Zeng, Kai
Luo, Yulin
Cao, Jiajun
Liang, Hao
Chen, Ying
She, Qi
Zhang, Shanghang
Zhang, Wentao
author_facet An, Ruichuan
Yang, Sihan
Zhang, Renrui
Lu, Ming
Jiang, Tianyi
Zeng, Kai
Luo, Yulin
Cao, Jiajun
Liang, Hao
Chen, Ying
She, Qi
Zhang, Shanghang
Zhang, Wentao
contents Current vision-language models (VLMs) show exceptional abilities across diverse tasks, such as visual question answering. To enhance user experience, recent studies have investigated VLM personalization to understand user-provided concepts. However, they mainly focus on single concepts, neglecting the existence and interplay of multiple concepts, which limits real-world applicability. This paper proposes MC-LLaVA, a multi-concept personalization paradigm. Specifically, MC-LLaVA employs a multi-concept instruction tuning strategy, effectively integrating multiple concepts in a single training step. To reduce the training costs, we propose a personalized textual prompt that uses visual token information to initialize concept tokens. Additionally, we introduce a personalized visual prompt during inference, aggregating location maps for enhanced recognition and grounding capabilities. To further push the performance upper bound, we incorporate an optional auxiliary loss, better enhancing the proposed personalized prompts. To decorate the VLM personalization research, we contribute a high-quality dataset. We carefully collect images with multiple characters and objects from movies and manually create question-answer samples for multi-concept scenarios, featuring superior diversity. Comprehensive experiments demonstrate that MC-LLaVA achieves impressive multi-concept personalized responses, paving the way for VLMs to become better user assistants. The code and dataset will be released at \href{https://github.com/arctanxarc/MC-LLaVA}{https://github.com/arctanxarc/MC-LLaVA}.
format Preprint
id arxiv_https___arxiv_org_abs_2411_11706
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle MC-LLaVA: Multi-Concept Personalized Vision-Language Model
An, Ruichuan
Yang, Sihan
Zhang, Renrui
Lu, Ming
Jiang, Tianyi
Zeng, Kai
Luo, Yulin
Cao, Jiajun
Liang, Hao
Chen, Ying
She, Qi
Zhang, Shanghang
Zhang, Wentao
Computer Vision and Pattern Recognition
Artificial Intelligence
Current vision-language models (VLMs) show exceptional abilities across diverse tasks, such as visual question answering. To enhance user experience, recent studies have investigated VLM personalization to understand user-provided concepts. However, they mainly focus on single concepts, neglecting the existence and interplay of multiple concepts, which limits real-world applicability. This paper proposes MC-LLaVA, a multi-concept personalization paradigm. Specifically, MC-LLaVA employs a multi-concept instruction tuning strategy, effectively integrating multiple concepts in a single training step. To reduce the training costs, we propose a personalized textual prompt that uses visual token information to initialize concept tokens. Additionally, we introduce a personalized visual prompt during inference, aggregating location maps for enhanced recognition and grounding capabilities. To further push the performance upper bound, we incorporate an optional auxiliary loss, better enhancing the proposed personalized prompts. To decorate the VLM personalization research, we contribute a high-quality dataset. We carefully collect images with multiple characters and objects from movies and manually create question-answer samples for multi-concept scenarios, featuring superior diversity. Comprehensive experiments demonstrate that MC-LLaVA achieves impressive multi-concept personalized responses, paving the way for VLMs to become better user assistants. The code and dataset will be released at \href{https://github.com/arctanxarc/MC-LLaVA}{https://github.com/arctanxarc/MC-LLaVA}.
title MC-LLaVA: Multi-Concept Personalized Vision-Language Model
topic Computer Vision and Pattern Recognition
Artificial Intelligence
url https://arxiv.org/abs/2411.11706