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

Unified Language-Vision Pretraining in LLM with Dynamic Discrete Visual Tokenization

About

Recently, the remarkable advance of the Large Language Model (LLM) has inspired researchers to transfer its extraordinary reasoning capability to both vision and language data. However, the prevailing approaches primarily regard the visual input as a prompt and focus exclusively on optimizing the text generation process conditioned upon vision content by a frozen LLM. Such an inequitable treatment of vision and language heavily constrains the model's potential. In this paper, we break through this limitation by representing both vision and language in a unified form. Specifically, we introduce a well-designed visual tokenizer to translate the non-linguistic image into a sequence of discrete tokens like a foreign language that LLM can read. The resulting visual tokens encompass high-level semantics worthy of a word and also support dynamic sequence length varying from the image. Coped with this tokenizer, the presented foundation model called LaVIT can handle both image and text indiscriminately under the same generative learning paradigm. This unification empowers LaVIT to serve as an impressive generalist interface to understand and generate multi-modal content simultaneously. Extensive experiments further showcase that it outperforms the existing models by a large margin on massive vision-language tasks. Our code and models are available at https://github.com/jy0205/LaVIT.

Yang Jin, Kun Xu, Kun Xu, Liwei Chen, Chao Liao, Jianchao Tan, Quzhe Huang, Bin Chen, Chenyi Lei, An Liu, Chengru Song, Xiaoqiang Lei, Di Zhang, Wenwu Ou, Kun Gai, Yadong Mu• 2023

Related benchmarks

TaskDatasetResultRank
Visual Question AnsweringVQA v2
Accuracy68.3
1165
Visual Question AnsweringVizWiz
Accuracy41
1043
Visual Question AnsweringGQA
Accuracy47.9
963
Visual Question AnsweringVQA v2 (test)
Accuracy66
131
Text-to-Image GenerationMS-COCO
FID7.4
75
Visual Question AnsweringVQAv2 (test)
VQA Accuracy66
72
Knowledge-based Visual Question AnsweringOKVQA
Accuracy0.546
52
Image-Text RetrievalFlickr30K
R@183
25
Image EditingMagicBrush
CLIPim81.1
6
Image EditingMA5K
LPIPS36.9
6
Showing 10 of 11 rows

Other info

Follow for update