arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2603.05315cs.CV

面向频率的误差有界缓存用于加速扩散变换器

Frequency-Aware Error-Bounded Caching for Accelerating Diffusion Transformers

iFLYTEK

Guandong Li

AI总结:

SpectralCache通过统一缓存框架提升扩散变换器推理速度,实现2.46倍加速,质量与TeaCache相当。

摘要:

扩散变换器(DiTs)已 emerged 为高质量图像和视频生成的主导架构,然而其迭代去噪过程在推理过程中会带来显著的计算成本。现有的缓存方法通过在时间步之间重用中间计算来加速DiTs,但它们有一个共同的局限性:将去噪过程视为时间、深度和特征维度上的均匀过程。在本工作中,我们识别了DiT去噪中的三个正交的非均匀轴:(1)时间轴——去噪轨迹中缓存误差的敏感性变化剧烈;(2)深度轴——连续的缓存决策导致级联的近似误差;(3)特征轴——隐藏状态的不同组件表现出异质的时间动态。基于这些观察,我们提出了SpectralCache,一个统一的缓存框架,包括时间步感知动态调度(TADS)、累积误差预算(CEB)和频率分解缓存(FDC)。在FLUX.1-schnell上以512x512分辨率,SpectralCache实现了2.46倍的加速,LPIPS为0.217,SSIM为0.727,优于TeaCache(2.12倍,LPIPS 0.215,SSIM 0.734)16%的加速,同时保持相似的质量(LPIPS差异<1%)。我们的方法是训练自由的、即插即用的,并且与现有的DiT架构兼容。

英文摘要:

Diffusion Transformers (DiTs) have emerged as the dominant architecture for high-quality image and video generation, yet their iterative denoising process incurs substantial computational cost during inference. Existing caching methods accelerate DiTs by reusing intermediate computations across timesteps, but they share a common limitation: treating the denoising process as uniform across time,depth, and feature dimensions. In this work, we identify three orthogonal axes of non-uniformity in DiT denoising: (1) temporal -- sensitivity to caching errors varies dramatically across the denoising trajectory; (2) depth -- consecutive caching decisions lead to cascading approximation errors; and (3) feature -- different components of the hidden state exhibit heterogeneous temporal dynamics. Based on these observations, we propose SpectralCache, a unified caching framework comprising Timestep-Aware Dynamic Scheduling (TADS), Cumulative Error Budgets (CEB), and Frequency-Decomposed Caching (FDC). On FLUX.1-schnell at 512x512 resolution, SpectralCache achieves 2.46x speedup with LPIPS 0.217 and SSIM 0.727, outperforming TeaCache (2.12x, LPIPS 0.215, SSIM 0.734) by 16% in speed while maintaining comparable quality (LPIPS difference < 1%). Our approach is training-free, plug-and-play, and compatible with existing DiT architectures.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑