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

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
DOI: https://doi.org/10.48550/arXiv.2309.04669
2024-03-22
Abstract: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 <a class="link-external link-https" href="https://github.com/jy0205/LaVIT" rel="external noopener nofollow">this https URL</a>.
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The paper aims to address the issue of unequal treatment in Multimodal Large Language Models (MLLMs) when processing image and text data, and attempts to enhance the model's multimodal understanding and generation capabilities through a unified representation approach. Specifically, the paper proposes a new model named LaVIT (Language-Vision Transformer), whose main objectives include: 1. **Unified Visual and Language Representation**: Current mainstream methods typically treat visual input as prompts, optimizing only the process of generating text based on visual content. This unbalanced approach limits the model's potential. LaVIT achieves equal treatment of images and text by converting images into discrete visual tokens, allowing these tokens to be understood by large language models just like text. 2. **Dynamic Discrete Visual Tokens**: To further improve representation compatibility and computational efficiency, the paper develops a mechanism capable of producing dynamic-length discrete visual tokens. This method not only reduces redundancy among visual patches but also enhances the consistency of image and text representations within the LLM. 3. **End-to-End Training Framework**: By introducing a novel visual tokenizer, LaVIT can directly predict the next image or text token without modifying the original architecture. This allows the model to handle image and text sequences simultaneously under a unified autoregressive training paradigm. In summary, LaVIT aims to overcome the unequal treatment of visual and language processing in existing methods, achieving more efficient and unified multimodal information processing capabilities through innovative visual tokenization techniques. Experimental results show that LaVIT significantly outperforms existing models on various vision-language tasks.