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
コメント
ログインするとコメントできます。