arXiv:2412.07720cs.CV2024-12被引 44

将自回归与扩散模型结合,实现连续视觉生成的灵活插值。

ACDiT: Interpolating Autoregressive Conditional Modeling and Diffusion Transformer

  • 用分块自回归单元实现生成过程的渐变控制。
  • 在图像视频生成上优于同类自回归模型,且支持长序列任务。
  • 适合需要高效推理与跨模态迁移的视觉生成研究者。

自回归与扩散模型分别在语言建模和视觉生成中取得显著进展。本文提出ACDiT,一种新型的分块自回归条件扩散变换器,创新性地将自回归与扩散范式结合用于连续视觉信息生成。通过引入分块自回归单元,ACDiT可在逐标记自回归与全序列扩散间灵活插值,避免离散标记化的局限。每一块的生成被建模为依赖前序块的条件扩散过程。ACDiT实现简单,仅需在标准扩散变压器训练时应用特定跳过因果注意力掩码。推理时,迭代执行去噪与自回归解码,可充分利用KV缓存。在图像、视频及文本生成任务上验证了有效性,显示ACDiT在相似模型规模下于视觉生成任务中优于所有自回归基线。此外,得益于自回归建模,预训练的ACDiT即使仅以生成目标训练,也可有效迁移到视觉理解任务。对自回归与扩散权衡的分析表明,ACDiT具备在长时序视觉生成中的潜力。我们希望ACDiT为视觉自回归生成提供新视角,并开启统一模型的新方向。

原文摘要 · Abstract (English)

Autoregressive and diffusion models have achieved remarkable progress in language models and visual generation, respectively. We present ACDiT, a novel Autoregressive blockwise Conditional Diffusion Transformer, that innovatively combines autoregressive and diffusion paradigms for continuous visual information. By introducing a block-wise autoregressive unit, ACDiT offers a flexible interpolation between token-wise autoregression and full-sequence diffusion, bypassing the limitations of discrete tokenization. The generation of each block is formulated as a conditional diffusion process, conditioned on prior blocks. ACDiT is easy to implement, as simple as applying a specially designed Skip-Causal Attention Mask on the standard diffusion transformer during training. During inference, the process iterates between diffusion denoising and autoregressive decoding that can make full use of KV-Cache. We validate the effectiveness of ACDiT on image, video, and text generation and show that ACDiT performs best among all autoregressive baselines under similar model scales on visual generation tasks. We also demonstrate that, benefiting from autoregressive modeling, pretrained ACDiT can be transferred in visual understanding tasks despite being trained with the generative objective. The analysis of the trade-off between autoregressive and diffusion demonstrates the potential of ACDiT to be used in long-horizon visual generation tasks. We hope that ACDiT offers a novel perspective on visual autoregressive generation and sheds light on new avenues for unified models.

视觉生成扩散模型自回归

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