arXiv:2412.15213cs.CV2024-12CVPR被引 30

不用噪声和条件机制,直接跨模态生成图像

Flowing from Words to Pixels: A Noise-Free Framework for Cross-Modality Evolution

  • 用流匹配直接学习一种模态到另一种的映射,跳过噪声中间步骤
  • 文本到图像生成效果优于标准流模型,且训练更稳定
  • 适合追求高效、可解释跨模态生成的研究者与开发者

扩散模型及其推广——流匹配,在媒体生成领域影响深远。传统方法是学习从简单的高斯噪声分布到目标媒体分布的复杂映射。在跨模态任务(如文本到图像生成)中,该映射通过模型中的条件机制实现。流匹配一个尚未被充分探索的关键特性是:与扩散模型不同,其源分布无需为噪声。本文提出范式转变,探讨是否可以直接训练流匹配模型,学习从一模态分布到另一模态分布的直接映射,从而省去噪声分布和条件机制。我们提出了通用且简洁的框架CrossFlow,强调输入数据使用变分编码器的重要性,并引入无分类器引导方法。令人惊讶的是,仅使用基础Transformer(无交叉注意力)的CrossFlow在文本到图像生成上表现优于标准流匹配,且随训练步数和模型规模增长更优,同时支持有意义的潜在空间算术操作,实现语义连贯的输出编辑。为验证方法普适性,我们在图像描述、深度估计和图像超分辨率等跨模态与同模态任务中,证明CrossFlow达到或超越当前最优水平。

原文摘要 · Abstract (English)

Diffusion models, and their generalization, flow matching, have had a remarkable impact on the field of media generation. Here, the conventional approach is to learn the complex mapping from a simple source distribution of Gaussian noise to the target media distribution. For cross-modal tasks such as text-to-image generation, this same mapping from noise to image is learnt whilst including a conditioning mechanism in the model. One key and thus far relatively unexplored feature of flow matching is that, unlike Diffusion models, they are not constrained for the source distribution to be noise. Hence, in this paper, we propose a paradigm shift, and ask the question of whether we can instead train flow matching models to learn a direct mapping from the distribution of one modality to the distribution of another, thus obviating the need for both the noise distribution and conditioning mechanism. We present a general and simple framework, CrossFlow, for cross-modal flow matching. We show the importance of applying Variational Encoders to the input data, and introduce a method to enable Classifier-free guidance. Surprisingly, for text-to-image, CrossFlow with a vanilla transformer without cross attention slightly outperforms standard flow matching, and we show that it scales better with training steps and model size, while also allowing for interesting latent arithmetic which results in semantically meaningful edits in the output space. To demonstrate the generalizability of our approach, we also show that CrossFlow is on par with or outperforms the state-of-the-art for various cross-modal / intra-modal mapping tasks, viz. image captioning, depth estimation, and image super-resolution. We hope this paper contributes to accelerating progress in cross-modal media generation.

跨模态生成流匹配文本到图像无条件生成

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