将自回归Transformer与扩散模型结合,实现更快更高质量的图像生成。
Marrying Autoregressive Transformer and Diffusion with Multi-Reference Autoregression
- 首创融合自回归Transformer与扩散模型的联合生成框架。
- ImageNet上FID达1.61,推理速度比自回归模型快2倍、比纯扩散模型快112倍。
- 引入多参考自回归机制,提升生成多样性与图像质量。
我们提出TransDiff,首个将自回归Transformer与扩散模型结合的图像生成模型。该框架将标签和图像编码为高层语义特征,并利用扩散模型估计图像样本分布。在ImageNet 256x256基准上,TransDiff显著优于单一使用自回归Transformer或扩散模型的方法:FID为1.61,Inception Score为293.4,推理速度较最先进的自回归方法快2倍,较纯扩散模型快112倍。在此基础上,我们提出多参考自回归(MRAR)新范式,通过预测下一幅图像实现自回归生成,并可参考多个已生成图像,从而学习更丰富的表示,提升后续迭代生成质量。应用MRAR后,TransDiff的FID从1.61降至1.42。我们期待TransDiff开启图像生成领域的新前沿。
原文摘要 · Abstract (English)
We introduce TransDiff, the first image generation model that marries Autoregressive (AR) Transformer with diffusion models. In this joint modeling framework, TransDiff encodes labels and images into high-level semantic features and employs a diffusion model to estimate the distribution of image samples. On the ImageNet 256x256 benchmark, TransDiff significantly outperforms other image generation models based on standalone AR Transformer or diffusion models. Specifically, TransDiff achieves a Frechet Inception Distance (FID) of 1.61 and an Inception Score (IS) of 293.4, and further provides x2 faster inference latency compared to state-of-the-art methods based on AR Transformer and x112 faster inference compared to diffusion-only models. Furthermore, building on the TransDiff model, we introduce a novel image generation paradigm called Multi-Reference Autoregression (MRAR), which performs autoregressive generation by predicting the next image. MRAR enables the model to reference multiple previously generated images, thereby facilitating the learning of more diverse representations and improving the quality of generated images in subsequent iterations. By applying MRAR, the performance of TransDiff is improved, with the FID reduced from 1.61 to 1.42. We expect TransDiff to open up a new frontier in the field of image generation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。