arXiv:2603.19232cs.CV2026-03中稿 · CVPR被引 1

首个可生成高维离散视觉令牌的模型,实现理解与生成统一

Cubic Discrete Diffusion: Discrete Visual Generation on High-Dimensional Representation Tokens

  • 采用细粒度掩码机制,任意位置任意维度均可预测
  • 在ImageNet-256上参数量达3.7B时性能领先,生成步数固定且远少于特征数量
  • 生成的离散令牌兼具理解与生成能力,适合多模态统一架构研究

基于离散令牌的视觉生成受到广泛关注,因其可与语言模型共享统一的令牌预测范式,有望实现无缝多模态架构。然而,现有离散生成方法仅限于低维潜在令牌(通常8-32维),牺牲了语义丰富性。高维预训练表示(768-1024维)虽可弥补此差距,但其离散生成面临根本挑战。本文提出首个针对高维表示的离散生成模型——立方离散扩散(CubiD)。CubiD在高维离散表示中实施细粒度掩码,任意位置、任意维度均可被掩码并从部分观测中重建。该机制使模型学习到空间位置间及内部的丰富相关性,且生成步数固定为$T$,满足$T \ll hwd$。在ImageNet-256上,CubiD在900M至3.7B参数量下均达到当前最优性能,展现出良好扩展性。关键在于,这些离散化令牌仍保留原始表征能力,同一组离散令牌可有效同时服务于理解与生成任务。代码已公开。

原文摘要 · Abstract (English)

Visual generation with discrete tokens has gained significant attention as it enables a unified token prediction paradigm shared with language models, promising seamless multimodal architectures. However, current discrete generation methods remain limited to low-dimensional latent tokens (typically 8-32 dims), sacrificing the semantic richness essential for understanding. While high-dimensional pretrained representations (768-1024 dims) could bridge this gap, their discrete generation poses fundamental challenges. In this paper, we present Cubic Discrete Diffusion (CubiD), the first discrete generation model for high-dimensional representations. CubiD performs fine-grained masking throughout the high-dimensional discrete representation -- any dimension at any position can be masked and predicted from partial observations. This enables the model to learn rich correlations both within and across spatial positions, with the number of generation steps fixed at $T$ regardless of feature dimensionality, where $T \ll hwd$. On ImageNet-256, CubiD achieves state-of-the-art discrete generation with strong scaling behavior from 900M to 3.7B parameters. Crucially, we validate that these discretized tokens preserve original representation capabilities, demonstrating that the same discrete tokens can effectively serve both understanding and generation tasks. We hope this work will inspire future research toward unified multimodal architectures. Code is available at: https://github.com/YuqingWang1029/CubiD.

视觉生成离散扩散多模态

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