Back to Basics: Let Denoising Generative Models Denoise

AK's Picks ML Transformer Generative Models DM
如今的去噪扩散模型并不以传统意义上的“去噪”方式进行工作,也就是说,它们并不会直接预测干净的图像。相反,神经网络预测的是噪声或含噪的数据。本文指出,预测干净数据与预测含噪数据在本质上是不同的。根据流形假设,自然数据应当位于一个低维流形上,而含噪数据则不然。基于这一假设,我们主张采用直接预测干净数据的模型,这使得看似容量不足的网络也能在非常高维的空间中有效运行。我们证明了仅使用像素作为输入的简单大尺寸图像块Transformer即可成为强大的生成模型:无需使用分词器(tokenizer)、无需预训练、也无需额外的损失函数。我们的方法在概念上无非就是“**纯图像Transformer**”,即我们所称的 **JiT**(Just image Transformers)。我们在ImageNet上以256和512分辨率、使用16和32的大图像块尺寸进行实验,取得了具有竞争力的结果——而在这些设置下,预测高维含噪量的方法可能会彻底失败。通过让网络回归到流形的基本原理,我们的研究回归本质,致力于在原始自然数据上建立一种自成一体的、基于Transformer的扩散建模范式。
Today's denoising diffusion models do not "denoise" in the classical sense, i.e., they do not directly predict clean images. Rather, the neural networks predict noise or a noised quantity. In this paper, we suggest that predicting clean data and predicting noised quantities are fundamentally different. According to the manifold assumption, natural data should lie on a low-dimensional manifold, whereas noised quantities do not. With this assumption, we advocate for models that directly predict clean data, which allows apparently under-capacity networks to operate effectively in very high-dimensional spaces. We show that simple, large-patch Transformers on pixels can be strong generative models: using no tokenizer, no pre-training, and no extra loss. Our approach is conceptually nothing more than "$\textbf{Just image Transformers}$", or $\textbf{JiT}$, as we call it. We report competitive results using JiT with large patch sizes of 16 and 32 on ImageNet at resolutions of 256 and 512, where predicting high-dimensional noised quantities can fail catastrophically. With our networks mapping back to the basics of the manifold, our research goes back to basics and pursues a self-contained paradigm for Transformer-based diffusion on raw natural data.
许愿