通过重采样视觉特征,高效恢复大模型中的细粒度视觉信息。
Vision Remember: Recovering Visual Information in Efficient LVLM with Vision Feature Resampling
- 在LLM解码层中重采样原始视觉特征,保持空间关系
- 相比现有方法,图表理解等任务提升最高5.7分
- 适用于多种高效视觉投影器,通用性强
大型视觉语言模型(LVLM)中冗余视觉标记的计算开销促使许多方法通过视觉投影器压缩它们。然而,这种压缩可能丢失对依赖精细空间关系的任务(如OCR和图表理解)至关重要的视觉信息。本文提出通过在LLM解码层重采样原始视觉特征来恢复视觉信息并实现效率。基于此思想,我们引入Vision Remember,包含两个关键模块:(1) 标记-特征交叉注意力层;(2) 标记双向自注意力层。在双向注意力中,采用自注意力机制维持视觉标记与文本引导标记间的双向交互。在标记-特征交互注意力中,引入局部交叉注意力重采样视觉特征,并利用多级融合丰富视觉表征。我们在多个视觉理解基准上进行了全面实验,结果表明,在LLaVA-NeXT基线上的表现优于TokenPacker达+2.7,优于FastV达+5.7,覆盖几乎所有设置。相较于以往视觉特征重融合方法,本方法也超越DeepStack达+3.9,超越SVA Aggregator达+3.4。实验结果验证了该方法在结合不同高效视觉投影器和LVLM时的泛化能力。
原文摘要 · Abstract (English)
The computational expense of redundant vision tokens in Large Vision-Language Models (LVLMs) has led many existing methods to compress them via a vision projector. However, this compression may lose visual information that is crucial for tasks relying on fine-grained spatial relationships, such as OCR and Chart&Table Understanding. In this paper, we propose to resample original vision features across the LLM decoder layers to recover visual information and attain efficiency. Following this principle, we introduce Vision Remember, which includes two key modules: (1) Token-Feature Cross-Attention Layer and (2) Token Bidirectional Self-Attention Layer. In the Token bidirectional attention, we employ self-attention mechanism to maintain the bidirectional interaction between vision tokens and the text-guided token. In the Token-Feature interaction attention, we introduce local cross-attention to resample the visual feature and utilize the multi-level fusion to enrich the visual representation. We conduct comprehensive experiments on multiple visual understanding benchmarks and the results with the LLaVA-NeXT baseline show that Vision Remember outperforms TokenPacker by +2.7 and FastV by +5.7 across nearly all the settings. Compared with previous vision feature re-fusion methods, our approach also surpasses DeepStack by +3.9 and SVA Aggregator by +3.4 on the same baseline. The experimental results validate the generalization capability of the proposed method when combined with various efficient vision projectors and LVLMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。