SoftCache让扩散模型缓存步长自适应
SoftCache: Adaptive Step Size Caching for Diffusion Transformers
中文摘要
由模型生成 · 人工抽检SoftCache是一种免训练缓存方法,利用动态缓存中陈旧度信号的余量预测复用误差,将二值刷新开关改为自适应步长:余量大时跳过多个调度节点,临近刷新时步长收缩回原值。在FLUX图像及HunyuanVideo、Wan2.1视频上实现3.5至6.9倍加速,四项保真度指标全面领先基线。
全文
已译中文扩散模型与流模型通过迭代步骤生成样本,每一步都需要对大型网络做一次完整的前向传播,而这次前向传播主导了生成成本。免训练缓存通过复用近期评估中的激活值而非重新计算来加速这一过程。动态缓存方法将网络输入的一个廉价探针累积为陈旧度信号,当该信号越过阈值时便执行一次新的评估,以此决定何时刷新。这类方法执行的是硬缓存:信号是刷新与复用之间的二值开关,且每一步——无论新算还是缓存——都按规定的步长推进。我们观察到,该信号所承载的信息远不止这个开关。它低于阈值的幅度,即其余量(headroom),可以预测复用所引入的误差。我们提出 SoftCache,执行软缓存:不再使用二值开关,而是由余量决定每个缓存步推进多远。在余量较大处,采样器跳过若干个计划节点;随着刷新临近,其步长逐渐收缩回原始步长。这些自适应跳跃可能落在计划节点之间,使采样器偏离其固定网格。SoftCache 无需训练,也不增加任何针对骨干网络的设置:在图像任务上的 FLUX 以及视频任务上的 HunyuanVideo 和 Wan2.1 上,在 3.5 倍到 6.9 倍的加速下,它在全部四项保真度指标上均领先于所有基线。
本译文由模型自动生成,仅供参考与快速理解;涉及数字、版本、命令、法律条文等请以原文为准。
原文附件
8 个 · 均跳转原站,本站不托管文件正文为原站公开内容的抓取与清洗结果,英文条目附模型生成的中文译文;两者均仅供参考, 准确内容请以原站为准。内容版权归原作者所有,若认为侵权请联系我们下线。