arXiv:2505.10562cs.CV2025-05NeurIPS被引 5

让视觉分词器与下游任务一起优化,提升图文理解与生成效果

End-to-End Vision Tokenizer Tuning

  • 联合优化视觉分词器与自回归任务,打破原有分离训练的局限
  • 在多模态任务上性能提升2-6%,同时保持原有重建能力
  • 无需修改原模型结构,可轻松集成到现有训练流程中

现有视觉分词方法将分词器优化与下游任务训练分离,隐含假设视觉令牌能跨任务通用,但为低层重构优化的分词器对需要多样化表征与语义的任务不敏感。这种解耦范式导致分词损失成为目标任务的表征瓶颈,例如图像中文字分词错误会严重影响识别或生成结果。为此,我们提出端到端视觉分词器调优(ETT),实现视觉分词与目标自回归任务的联合优化。不同于以往仅使用冻结分词器离散索引的自回归模型,ETT利用分词器码本的视觉嵌入,并端到端优化分词器,同时兼顾重建与描述生成目标。ETT可无缝集成至现有训练流程,无需调整原代码本或大语言模型架构。大量实验表明,相比冻结分词器基线,该方法在多模态理解与视觉生成任务上取得2-6%的显著性能提升,同时保留原始重建能力。我们希望这一简单而有效的方法能推动多模态基础模型在图像生成与理解之外的应用。

原文摘要 · Abstract (English)

Existing vision tokenization isolates the optimization of vision tokenizers from downstream training, implicitly assuming the visual tokens can generalize well across various tasks, e.g., image generation and visual question answering. The vision tokenizer optimized for low-level reconstruction is agnostic to downstream tasks requiring varied representations and semantics. This decoupled paradigm introduces a critical misalignment: The loss of the vision tokenization can be the representation bottleneck for target tasks. For example, errors in tokenizing text in a given image lead to poor results when recognizing or generating them. To address this, we propose ETT, an end-to-end vision tokenizer tuning approach that enables joint optimization between vision tokenization and target autoregressive tasks. Unlike prior autoregressive models that use only discrete indices from a frozen vision tokenizer, ETT leverages the visual embeddings of the tokenizer codebook, and optimizes the vision tokenizers end-to-end with both reconstruction and caption objectives. ETT can be seamlessly integrated into existing training pipelines with minimal architecture modifications. Our ETT is simple to implement and integrate, without the need to adjust the original codebooks or architectures of the employed large language models. Extensive experiments demonstrate that our proposed end-to-end vision tokenizer tuning unlocks significant performance gains, i.e., 2-6% for multimodal understanding and visual generation tasks compared to frozen tokenizer baselines, while preserving the original reconstruction capability. We hope this very simple and strong method can empower multimodal foundation models besides image generation and understanding.

视觉分词多模态端到端优化

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