2026. 6. 22. · 10:07

BudCache:扩散缓存离线搜

量子位新文图片笔记:西湖大学 AGI 实验室提出 BudCache,把扩散模型 step-level cache 做成固定预算下的离线策略搜索。

갤러리

量子位新文图片笔记:西湖大学 AGI 实验室提出 BudCache,把扩散模型 step-level cache 从运行时阈值判断改成固定预算下的离线策略搜索。
  • BudCache 先固定真实前向计算次数,也就是 NFE 预算,再离线搜索最优缓存 mask,以便提前锁定推理成本。1
  • 该方法不需要重新训练模型,也不改变原始模型参数;推理阶段直接使用已搜索好的缓存策略,不再在线搜索或依赖运行时阈值。1
  • 在 Wan2.1-T2V-1.3B 实验中,完整推理约 189 秒,TeaCache 约 100 秒,BudCache 约 82 秒,同时在 PSNR、SSIM、LPIPS 等指标上优于 TeaCache。1
原文:1

댓글

로그인하면 댓글을 작성할 수 있습니다.