arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2307.08041cs.CV
收藏

在大型语言模型中植入视觉的 SEED

Planting a SEED of Vision in Large Language Model

Tencent AI Lab(腾讯人工智能实验室);Tencent PCG(腾讯平台与内容事业群)

Yuying Ge, Yixiao Ge, Ziyun Zeng, Xintao Wang, Ying Shan

AI总结:

本文提出图像分词器 SEED,通过一维因果视觉 token 与语义判别、重建联合优化,使现成 LLM 经 LoRA 微调即可同时完成图文理解和生成。

摘要:

我们提出了 SEED,这是一种精心设计的图像分词器,使大型语言模型(LLM)能够同时获得“看”和“画”的涌现能力。此前,图像分词器研究曾陷入僵局:采用量化视觉 token 的框架因在多模态理解(相比 BLIP-2 等)或生成(相比 Stable Diffusion 等)方面性能不佳且收敛困难而失去关注。尽管存在这些局限,我们仍相信其天然能够统一视觉与文本表示,从而利用 LLM 的原始训练方案促进可扩展的多模态训练。在本研究中,我们确定了 SEED 架构与训练的两个关键原则,可有效简化后续与 LLM 的对齐。(1)图像 token 应独立于二维物理 patch 位置,而以一维因果依赖生成,表现出与 LLM 中从左到右自回归预测机制相一致的内在相互依赖。(2)图像 token 应捕捉与词语语义抽象程度一致的高级语义,并在分词器训练阶段同时针对判别性和重建进行优化。因此,通过高效的 LoRA 微调将我们的 SEED 纳入后,现成的 LLM 即可执行图像到文本以及文本到图像生成。可能带来更好结果的全面多模态预训练和指令微调留待未来研究。此版本 SEED 仅使用 64 块 V100 GPU 和 500 万个公开可用的图文对,在 5.7 天内完成训练。我们的初步研究强调了离散视觉 token 在通用多模态 LLM 中的巨大潜力,以及合适的图像分词器在更广泛研究中的重要性。

英文摘要:

We present SEED, an elaborate image tokenizer that empowers Large Language Models (LLMs) with the emergent ability to SEE and Draw at the same time. Research on image tokenizers has previously reached an impasse, as frameworks employing quantized visual tokens have lost prominence due to subpar performance and convergence in multimodal comprehension (compared to BLIP-2, etc.) or generation (compared to Stable Diffusion, etc.). Despite the limitations, we remain confident in its natural capacity to unify visual and textual representations, facilitating scalable multimodal training with LLM's original recipe. In this study, we identify two crucial principles for the architecture and training of SEED that effectively ease subsequent alignment with LLMs. (1) Image tokens should be independent of 2D physical patch positions and instead be produced with a 1D causal dependency, exhibiting intrinsic interdependence that aligns with the left-to-right autoregressive prediction mechanism in LLMs. (2) Image tokens should capture high-level semantics consistent with the degree of semantic abstraction in words, and be optimized for both discriminativeness and reconstruction during the tokenizer training phase. As a result, the off-the-shelf LLM is able to perform both image-to-text and text-to-image generation by incorporating our SEED through efficient LoRA tuning. Comprehensive multimodal pretraining and instruction tuning, which may yield improved results, are reserved for future investigation. This version of SEED was trained in 5.7 days using only 64 V100 GPUs and 5M publicly available image-text pairs. Our preliminary study emphasizes the great potential of discrete visual tokens in versatile multimodal LLMs and the importance of proper image tokenizers in broader research.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑