用1个视觉标记实现高效多模态理解,大幅降低计算开销。
LLaVA-Mini: Efficient Image and Video Large Multimodal Models with One Vision Token
- 提前融合视觉与文本信息,将视觉标记压缩至仅1个。
- 在11项图像和7项视频任务中超越原版模型,仅用1个视觉标记。
- 支持高分辨率图像与长视频处理,40毫秒内响应,节省77%算力。
实时大型多模态模型(如GPT-4o)的出现引发了对高效多模态模型的关注。现有框架通常将视觉输入编码为大量视觉标记,与文本指令一起输入大语言模型,导致显著计算开销。以往优化多模态模型主要聚焦于替换小规模语言模型,却忽视了标记数量这一关键问题。本文提出LLaVA-Mini,一种仅使用极少视觉标记的高效多模态模型。通过分析发现,多数视觉标记仅在语言模型早期层中起关键作用,主要用于融合视觉信息到文本标记中。基于此,LLaVA-Mini引入模态预融合机制,在输入语言模型前预先融合视觉信息,从而实现视觉标记极端压缩——仅需1个视觉标记即可输入。该模型可统一处理图像、高分辨率图像及视频。在11项图像与7项视频基准测试中,其性能超越LLaVA-v1.5,仅使用1个视觉标记而非576个。效率分析显示,模型可减少77%的浮点运算量,响应延迟低于40毫秒,并可在24GB显存的GPU上处理超10,000帧视频。
原文摘要 · Abstract (English)
The advent of real-time large multimodal models (LMMs) like GPT-4o has sparked considerable interest in efficient LMMs. LMM frameworks typically encode visual inputs into vision tokens (continuous representations) and integrate them and textual instructions into the context of large language models (LLMs), where large-scale parameters and numerous context tokens (predominantly vision tokens) result in substantial computational overhead. Previous efforts towards efficient LMMs always focus on replacing the LLM backbone with smaller models, while neglecting the crucial issue of token quantity. In this paper, we introduce LLaVA-Mini, an efficient LMM with minimal vision tokens. To achieve a high compression ratio of vision tokens while preserving visual information, we first analyze how LMMs understand vision tokens and find that most vision tokens only play a crucial role in the early layers of LLM backbone, where they mainly fuse visual information into text tokens. Building on this finding, LLaVA-Mini introduces modality pre-fusion to fuse visual information into text tokens in advance, thereby facilitating the extreme compression of vision tokens fed to LLM backbone into one token. LLaVA-Mini is a unified large multimodal model that can support the understanding of images, high-resolution images, and videos in an efficient manner. Experiments across 11 image-based and 7 video-based benchmarks demonstrate that LLaVA-Mini outperforms LLaVA-v1.5 with just 1 vision token instead of 576. Efficiency analyses reveal that LLaVA-Mini can reduce FLOPs by 77%, deliver low-latency responses within 40 milliseconds, and process over 10,000 frames of video on the GPU hardware with 24GB of memory.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。