分离编码与解码模块,提升扩散Transformer的生成速度与质量
DDT: Decoupled Diffusion Transformer
- 用专用编码器提取语义,专用解码器恢复高频细节,避免性能冲突
- 在256×256和512×512图像上分别达到1.31和1.28 FID,训练速度更快
- 支持相邻步骤自条件共享,推理速度提升,且通过动态规划优化策略
扩散Transformer虽生成质量高,但需较长训练和推理步数。传统方法在去噪每一步中使用相同模块编码低频语义并解码高频信息,导致语义提取与高频重建间存在固有矛盾。为此,我们提出新的解耦扩散Transformer(DDT),采用独立的条件编码器专注语义提取,配合专用速度解码器恢复高频细节。实验表明,随着模型规模增大,更强编码器带来显著性能提升。在ImageNet 256×256上,DDT-XL/2达到1.31 FID,训练收敛速度比此前方法快近4倍;在512×512上,FID达1.28,刷新纪录。此外,解耦结构使相邻去噪步骤可共享自条件,提升推理效率。为减少性能损失,我们提出一种新颖的统计动态规划方法,自动识别最优共享策略。
原文摘要 · Abstract (English)
Diffusion transformers have demonstrated remarkable generation quality, albeit requiring longer training iterations and numerous inference steps. In each denoising step, diffusion transformers encode the noisy inputs to extract the lower-frequency semantic component and then decode the higher frequency with identical modules. This scheme creates an inherent optimization dilemma: encoding low-frequency semantics necessitates reducing high-frequency components, creating tension between semantic encoding and high-frequency decoding. To resolve this challenge, we propose a new \textbf{\color{ddt}D}ecoupled \textbf{\color{ddt}D}iffusion \textbf{\color{ddt}T}ransformer~(\textbf{\color{ddt}DDT}), with a decoupled design of a dedicated condition encoder for semantic extraction alongside a specialized velocity decoder. Our experiments reveal that a more substantial encoder yields performance improvements as model size increases. For ImageNet $256\times256$, Our DDT-XL/2 achieves a new state-of-the-art performance of {1.31 FID}~(nearly $4\times$ faster training convergence compared to previous diffusion transformers). For ImageNet $512\times512$, Our DDT-XL/2 achieves a new state-of-the-art FID of 1.28. Additionally, as a beneficial by-product, our decoupled architecture enhances inference speed by enabling the sharing self-condition between adjacent denoising steps. To minimize performance degradation, we propose a novel statistical dynamic programming approach to identify optimal sharing strategies.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。