arXiv:2412.18390cs.CVcs.AI2024-12被引 2

将扩散模型用于离散生成,用递归预测实现高效图像合成。

RDPM: Solve Diffusion Probabilistic Models via Recurrent Token Prediction

  • 通过递归预测离散令牌,将连续扩散过程转为离散域生成。
  • 仅需少数推理步即可生成高质量图像,速度优于传统方法。
  • 适合追求高效多模态生成的开发者与研究者。

扩散概率模型(DPMs)已成为高保真图像生成的主流方法,其在连续的VAE潜在空间中执行扩散过程,与大语言模型(LLMs)用于文本生成的方法显著不同。本文提出一种新型生成框架——循环扩散概率模型(RDPM),通过递归令牌预测机制增强扩散过程,开创了离散扩散的新方向。该模型逐步向图像潜在表示引入高斯噪声,并以递归方式将其编码为向量量化令牌,从而在离散值域上实现独特的扩散过程。该过程迭代预测后续时间步的令牌码,将初始标准高斯噪声逐步转化为源数据分布,损失函数设计与GPT风格模型一致。实验表明,RDPM在保持高质量生成的同时,具备仅需少量推理步骤即可完成生成的速度优势。该模型不仅利用扩散过程保证生成质量,还将连续信号转换为一系列高保真离散令牌,实现了与其他离散符号(如文本)统一的优化策略。我们预期该工作将推动多模态生成的统一模型发展,特别是整合图像、视频、音频等连续信号域与文本。代码与模型权重将开源发布。

原文摘要 · Abstract (English)

Diffusion Probabilistic Models (DPMs) have emerged as the de facto approach for high-fidelity image synthesis, operating diffusion processes on continuous VAE latent, which significantly differ from the text generation methods employed by Large Language Models (LLMs). In this paper, we introduce a novel generative framework, the Recurrent Diffusion Probabilistic Model (RDPM), which enhances the diffusion process through a recurrent token prediction mechanism, thereby pioneering the field of Discrete Diffusion. By progressively introducing Gaussian noise into the latent representations of images and encoding them into vector-quantized tokens in a recurrent manner, RDPM facilitates a unique diffusion process on discrete-value domains. This process iteratively predicts the token codes for subsequent timesteps, transforming the initial standard Gaussian noise into the source data distribution, aligning with GPT-style models in terms of the loss function. RDPM demonstrates superior performance while benefiting from the speed advantage of requiring only a few inference steps. This model not only leverages the diffusion process to ensure high-quality generation but also converts continuous signals into a series of high-fidelity discrete tokens, thereby maintaining a unified optimization strategy with other discrete tokens, such as text. We anticipate that this work will contribute to the development of a unified model for multimodal generation, specifically by integrating continuous signal domains such as images, videos, and audio with text. We will release the code and model weights to the open-source community.

扩散模型离散生成多模态

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