<p>Diffusion models have recently gained widespread adoption in visual generation. However, their iterative denoising process is computationally intensive, making real-time inference on embedded devices with limited power highly challenging. As a result, accelerating diffusion models has become a critical research focus. While existing acceleration techniques are primarily designed for UNet-based architectures, they are not directly applicable to Transformer-based diffusion models (DiT). A natural approach to speed up DiT is to skip certain blocks, but this often leads to significant degradation in generation quality. To address the unique challenges of DiT, we propose <InlineEquation ID="IEq3"> <EquationSource Format="TEX">\(\Delta \)</EquationSource> <EquationSource Format="MATHML"><math> <mi mathvariant="normal">Δ</mi> </math></EquationSource> </InlineEquation>-Cache, a novel caching method that captures and stores the incremental changes between different blocks, effectively reducing computational costs while maintaining fidelity to the original output. Furthermore, we conduct a quantitative analysis of the relationship between DiT block depth and image generation quality. Our findings reveal that shallow DiT blocks primarily define global structures such as composition and outlines, while deeper blocks focus on refining details, with middle blocks playing an intermediate role. Building on these insights, we introduce a denoising property alignment method that adaptively skips computations for different blocks at various timesteps while preserving performance. Comprehensive experiments on PIXART-<InlineEquation ID="IEq4"> <EquationSource Format="TEX">\(\alpha \)</EquationSource> <EquationSource Format="MATHML"><math> <mi>α</mi> </math></EquationSource> </InlineEquation>, SD3, and DiT-XL demonstrate that <InlineEquation ID="IEq5"> <EquationSource Format="TEX">\(\Delta \)</EquationSource> <EquationSource Format="MATHML"><math> <mi mathvariant="normal">Δ</mi> </math></EquationSource> </InlineEquation>-DiT achieves a 1.6<InlineEquation ID="IEq6"> <EquationSource Format="TEX">\(\times \)</EquationSource> <EquationSource Format="MATHML"><math> <mo>×</mo> </math></EquationSource> </InlineEquation> speedup while even enhancing generation quality. Additionally, on the NVIDIA Jetson AGX Orin, our method delivers a 2.03<InlineEquation ID="IEq7"> <EquationSource Format="TEX">\(\times \)</EquationSource> <EquationSource Format="MATHML"><math> <mo>×</mo> </math></EquationSource> </InlineEquation> real-world inference speedup without sacrificing performance, highlighting its effectiveness in resource-constrained scenarios.</p>

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

\(\Delta \)-DiT: Accelerating Diffusion Transformers without Training via Denoising Property Alignment

  • Pengtao Chen,
  • Mingzhu Shen,
  • Peng Ye,
  • Jianjian Cao,
  • Chongjun Tu,
  • Christos-Savvas Bouganis,
  • Yiren Zhao,
  • Tao Chen

摘要

Diffusion models have recently gained widespread adoption in visual generation. However, their iterative denoising process is computationally intensive, making real-time inference on embedded devices with limited power highly challenging. As a result, accelerating diffusion models has become a critical research focus. While existing acceleration techniques are primarily designed for UNet-based architectures, they are not directly applicable to Transformer-based diffusion models (DiT). A natural approach to speed up DiT is to skip certain blocks, but this often leads to significant degradation in generation quality. To address the unique challenges of DiT, we propose \(\Delta \) Δ -Cache, a novel caching method that captures and stores the incremental changes between different blocks, effectively reducing computational costs while maintaining fidelity to the original output. Furthermore, we conduct a quantitative analysis of the relationship between DiT block depth and image generation quality. Our findings reveal that shallow DiT blocks primarily define global structures such as composition and outlines, while deeper blocks focus on refining details, with middle blocks playing an intermediate role. Building on these insights, we introduce a denoising property alignment method that adaptively skips computations for different blocks at various timesteps while preserving performance. Comprehensive experiments on PIXART- \(\alpha \) α , SD3, and DiT-XL demonstrate that \(\Delta \) Δ -DiT achieves a 1.6 \(\times \) × speedup while even enhancing generation quality. Additionally, on the NVIDIA Jetson AGX Orin, our method delivers a 2.03 \(\times \) × real-world inference speedup without sacrificing performance, highlighting its effectiveness in resource-constrained scenarios.