让AI根据图像内容复杂度自动减少视觉标记,提升多模态模型效率
ViCO: A Training Strategy towards Semantic Aware Dynamic High-Resolution
- 用不同压缩率的MLP连接器,按语义复杂度动态调整图像视觉标记数
- 实验显示可减少50%视觉标记数,同时保持感知、推理和OCR能力
- 适合追求高效多模态大模型的开发者与研究者
现有多模态大语言模型因图像输入引入额外视觉标记而导致推理成本上升。本文提出视觉一致性学习(ViCO),一种新型训练算法,使模型能根据图像语义复杂度使用不同数量的视觉标记。核心思路是采用多个具有不同图像压缩比的MLP连接器,基于图像语义复杂度对视觉标记进行下采样。训练时最小化不同连接器条件下的响应间KL散度。推理时引入图像路由器(视觉分辨率路由器,ViR),自动为每张图像块选择合适压缩率。相比仅依据图像分辨率调整视觉标记数的现有动态高分辨率策略,本方法根据语义复杂度动态适应标记数量。实验表明,该方法可减少高达50%的视觉标记数,同时保持模型的感知、推理与OCR能力。代码与模型将开源,以促进后续研究。
原文摘要 · Abstract (English)
Existing Multimodal Large Language Models (MLLMs) suffer from increased inference costs due to the additional vision tokens introduced by image inputs. In this work, we propose Visual Consistency Learning (ViCO), a novel training algorithm that enables the model to represent images of varying semantic complexities using different numbers of vision tokens. The key idea behind our method is to employ multiple MLP connectors, each with a different image compression ratio, to downsample the vision tokens based on the semantic complexity of the image. During training, we minimize the KL divergence between the responses conditioned on different MLP connectors. At inference time, we introduce an image router, termed Visual Resolution Router (ViR), that automatically selects the appropriate compression rate for each image patch. Compared with existing dynamic high-resolution strategies, which adjust the number of visual tokens based on image resolutions, our method dynamically adapts the number of visual tokens according to semantic complexity. Experimental results demonstrate that our method can reduce the number of vision tokens by up to 50% while maintaining the model's perception, reasoning, and OCR capabilities. We hope this work will contribute to the development of more efficient MLLMs. The code and models will be released to facilitate future research.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。