Leveraging Visual Tokens for Extended Text Contexts in Multi-Modal Learning

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Wang, Alex Jinpeng, Li, Linjie, Lin, Yiqi, Li, Min, Wang, Lijuan, Shou, Mike Zheng
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866911903761039360
author Wang, Alex Jinpeng
Li, Linjie
Lin, Yiqi
Li, Min
Wang, Lijuan
Shou, Mike Zheng
author_facet Wang, Alex Jinpeng
Li, Linjie
Lin, Yiqi
Li, Min
Wang, Lijuan
Shou, Mike Zheng
contents Training models with longer in-context lengths is a significant challenge for multimodal model due to substantial GPU memory and computational costs. This exploratory study does not present state-of-the-art models; rather, it introduces an innovative method designed to increase in-context text length in multi-modality large language models (MLLMs) efficiently. We present Visualized In-Context Text Processing (VisInContext), which processes long in-context text using visual tokens. This technique significantly reduces GPU memory usage and floating point operations (FLOPs) for both training and inferenceing stage. For instance, our method expands the pre-training in-context text length from 256 to 2048 tokens with nearly same FLOPs for a 56 billion parameter MOE model. Experimental results demonstrate that model trained with VisInContext delivers superior performance on common downstream benchmarks for in-context few-shot evaluation. Additionally, VisInContext is complementary to existing methods for increasing in-context text length and enhances document understanding capabilities, showing great potential in document QA tasks and sequential document retrieval.
format Preprint
id arxiv_https___arxiv_org_abs_2406_02547
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Leveraging Visual Tokens for Extended Text Contexts in Multi-Modal Learning
Wang, Alex Jinpeng
Li, Linjie
Lin, Yiqi
Li, Min
Wang, Lijuan
Shou, Mike Zheng
Computer Vision and Pattern Recognition
Training models with longer in-context lengths is a significant challenge for multimodal model due to substantial GPU memory and computational costs. This exploratory study does not present state-of-the-art models; rather, it introduces an innovative method designed to increase in-context text length in multi-modality large language models (MLLMs) efficiently. We present Visualized In-Context Text Processing (VisInContext), which processes long in-context text using visual tokens. This technique significantly reduces GPU memory usage and floating point operations (FLOPs) for both training and inferenceing stage. For instance, our method expands the pre-training in-context text length from 256 to 2048 tokens with nearly same FLOPs for a 56 billion parameter MOE model. Experimental results demonstrate that model trained with VisInContext delivers superior performance on common downstream benchmarks for in-context few-shot evaluation. Additionally, VisInContext is complementary to existing methods for increasing in-context text length and enhances document understanding capabilities, showing great potential in document QA tasks and sequential document retrieval.
title Leveraging Visual Tokens for Extended Text Contexts in Multi-Modal Learning
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2406.02547