arXiv:2503.21757cs.CVcs.AI2025-03被引 2

用双前向机制让大模型自压缩视觉信息,兼顾生成与判别任务。

Fwd2Bot: LVLM Visual Token Compression with Double Forward Bottleneck

  • 利用大模型自身进行无任务依赖的视觉令牌压缩。
  • 生成任务压缩率提升2倍且不损失能力,判别任务达新最优。
  • 适合需要高效视觉表示的多任务大模型应用。

本文旨在将大型视觉语言模型(LVLM)的视觉令牌压缩为同时适用于生成和判别任务、近乎无损且存储高效的表示。提出一种名为Fwd2Bot的新压缩方法,利用LVLM自身以任务无关方式压缩视觉信息。核心是‘双前向传递’训练策略:第一轮前向传播中,LVLM的LLM部分将视觉信息浓缩为少量摘要令牌;第二轮则将语言指令与这些摘要令牌一起处理,作为图像令牌的直接替代。训练信号来自两个损失:第二轮后的自回归损失提供压缩优化目标,第一轮后的对比损失增强表示能力,尤其提升判别任务性能。训练还引入阶段专用适配器。通过深入消融实验验证,Fwd2Bot生成的高度信息性压缩表示在生成任务上实现2倍压缩率而无性能损失,创下新纪录;在图像检索与组合性任务上亦达新最优水平。

原文摘要 · Abstract (English)

In this work, we aim to compress the vision tokens of a Large Vision Language Model (LVLM) into a representation that is simultaneously suitable for (a) generative and (b) discriminative tasks, (c) is nearly lossless, and (d) is storage-efficient. We propose a novel compression approach, called Fwd2Bot, that uses the LVLM itself to compress the visual information in a task-agnostic manner. At the core of Fwd2bot there exists a "double-forward pass" training strategy, whereby, during the first forward pass, the LLM (of the LVLM) creates a bottleneck by condensing the visual information into a small number of summary tokens. Then, using the same LLM, the second forward pass processes the language instruction(s) alongside the summary tokens, used as a direct replacement for the image ones. The training signal is provided by two losses: an autoregressive one applied after the second pass that provides a direct optimization objective for compression, and a contrastive loss, applied after the first pass, that further boosts the representation strength, especially for discriminative tasks. The training is further enhanced by stage-specific adapters. We accompany the proposed method by an in-depth ablation study. Overall, Fwd2Bot results in highly-informative compressed representations suitable for both generative and discriminative tasks. For generative tasks, we offer a 2x higher compression rate without compromising the generative capabilities, setting a new state-of-the-art result. For discriminative tasks, we set a new state-of-the-art on image retrieval and compositionality.

视觉压缩大模型双前向多任务

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。