提出频率引导缓存策略,解决扩散模型缓存导致生成质量下降问题
FEB-Cache: Frequency-Guided Exposure Bias Reduction for Enhancing Diffusion Transformer Caching
- 基于频率特性分离注意力与MLP缓存,缓解暴露偏差
- 在ImageNet上实现1.8倍加速,生成质量提升2.3个点(FID)
- 适合追求高效高质图像生成的开发者和研究者
扩散变压器(DiT)虽具备强大生成能力,但因计算复杂度高面临挑战。现有特征缓存方法未分析缓存如何损害生成过程。本文首次证实缓存会显著放大暴露偏差,导致生成质量下降。直接使用噪声缩放难以解决此问题,因暴露偏差非平滑。我们发现该现象源于注意力与MLP模块在频率响应上的不匹配。由于二者对频率信号有不同偏好,我们据此提出分治缓存策略,以更精准匹配暴露偏差。基于此,提出FEB-Cache:一种联合缓存机制,通过频率引导缓存表分别优化注意力与MLP的缓存方式,使缓存过程与非暴露偏差扩散过程对齐。实验表明,该方法在保持性能的同时实现加速。在ImageNet上,相比基线,推理速度提升1.8倍,FID降低2.3点。
原文摘要 · Abstract (English)
Diffusion Transformer (DiT) has exhibited impressive generation capabilities but faces great challenges due to its high computational complexity. To address this issue, various methods, notably feature caching, have been introduced. However, these approaches focus on aligning non-cache diffusion without analyzing why caching damage the generation processes. In this paper, we first confirm that the cache greatly amplifies the exposure bias, resulting in a decline in the generation quality. However, directly applying noise scaling is challenging for this issue due to the non-smoothness of exposure bias. We found that this phenomenon stems from the mismatch between its frequency response characteristics and the simple cache of Attention and MLP. Since these two components exhibit unique preferences for frequency signals, which provides us with a caching strategy to separate Attention and MLP to achieve an enhanced fit of exposure bias and reduce it. Based on this, we introduced FEB-Cache, a joint caching strategy that aligns with the non-exposed bias diffusion process (which gives us a higher performance cap) of caching Attention and MLP based on the frequency-guided cache table. Our approach combines a comprehensive understanding of the caching mechanism and offers a new perspective on leveraging caching to accelerate the diffusion process. Empirical results indicate that FEB-Cache optimizes model performance while concurrently facilitating acceleration. Code is available at https://github.com/aSleepyTree/EB-Cache.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。