arXiv:2505.07538cs.CV2025-05被引 33

用扩散模型生成视觉自回归令牌,实现无需文本对的高效视觉生成。

Selftok: Discrete Visual Tokens of Autoregression, by Diffusion, and for Reasoning

  • 通过反向扩散过程构建自回归视觉令牌,打破传统空间先验
  • 在无文本图像对情况下,强化学习显著提升生成质量,超越现有模型
  • 支持视觉生成的强化学习,适合多模态大模型研究者

本文摒弃图像表征中的传统空间先验,提出一种新型离散视觉令牌——自洽令牌(Selftok)。其核心设计是利用图像生成的反向扩散过程,将自回归(AR)先验引入视觉令牌中,模拟语言模型的因果结构。该方法使Selftok在两个关键方面区别于传统空间令牌:首先,它为视觉语言模型(VLMs)提供了一种优雅的统一范式,仅用纯离散自回归架构即可训练VLM,无需额外模块或训练目标;其次,理论证明自回归先验满足贝尔曼方程,而空间先验不满足,因此Selftok可有效支持视觉生成中的强化学习(RL),效果媲美语言模型。此外,Selftok在重建质量与压缩率间达到当前最优平衡。基于Selftok构建的纯自回归VLM在视觉理解与生成任务上表现优异,尤其在未使用任何文本-图像训练对的情况下,仅通过简单的策略梯度强化学习,便显著提升生成基准性能,大幅超越现有模型。这表明Selftok有效解决了视觉令牌难以支持高效强化学习的长期难题。结合强化学习在大语言模型中的成熟优势,推动迈向真正多模态大模型更进一步。

原文摘要 · Abstract (English)

We completely discard the conventional spatial prior in image representation and introduce a novel discrete visual tokenizer: Self-consistency Tokenizer (Selftok). At its design core, we compose an autoregressive (AR) prior -- mirroring the causal structure of language -- into visual tokens by using the reverse diffusion process of image generation. The AR property makes Selftok fundamentally distinct from traditional spatial tokens in the following two key ways: - Selftok offers an elegant and minimalist approach to unify diffusion and AR for vision-language models (VLMs): By representing images with Selftok tokens, we can train a VLM using a purely discrete autoregressive architecture -- like that in LLMs -- without requiring additional modules or training objectives. - We theoretically show that the AR prior satisfies the Bellman equation, whereas the spatial prior does not. Therefore, Selftok supports reinforcement learning (RL) for visual generation with effectiveness comparable to that achieved in LLMs. Besides the AR property, Selftok is also a SoTA tokenizer that achieves a favorable trade-off between high-quality reconstruction and compression rate. We use Selftok to build a pure AR VLM for both visual comprehension and generation tasks. Impressively, without using any text-image training pairs, a simple policy gradient RL working in the visual tokens can significantly boost the visual generation benchmark, surpassing all the existing models by a large margin. Therefore, we believe that Selftok effectively addresses the long-standing challenge that visual tokens cannot support effective RL. When combined with the well-established strengths of RL in LLMs, this brings us one step closer to realizing a truly multimodal LLM. Project Page: https://selftok-team.github.io/report/.

视觉生成自回归扩散模型强化学习

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