研究人员开发了BAG(Budget-Aware Gating),一种旨在加速Diffusion Transformer(DiTs)的新型缓存策略。与之前缺乏预算感知或实例适应性的方法不同,BAG使用轻量级门控网络动态决定是重用缓存的特征还是执行完整计算。该方法通过离线到在线调度蒸馏进行训练,在FLUX.1-dev和Wan2.1数据集上,在各种加速级别和分辨率下,始终优于现有的缓存技术。 AI
影响 这种新的缓存策略可能导致扩散模型更快的推理时间,提高生成式AI应用的效率。
排序理由 该集群描述了在arXiv上的一篇研究论文中提出的一种新颖方法。[lever_c_demoted from research: ic=1 ai=1.0]
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →