研究人员开发了 Step-dLLM,一种在推理过程中提高扩散大型语言模型 (dLLM) 效率的新颖方法。与自回归模型不同,dLLM 不能使用 KV 缓存并在每一步重新计算注意力图,导致成本高昂。Step-dLLM 通过在锚点步骤自适应地刷新稀疏注意力掩码并采用头部自适应预算来优先处理注意力集中的头部来解决此问题。这种方法通过融合的 Triton 内核实现,在 LLaDA-1.5 和 Dream-7B-Instruct 等模型上实现了显著的加速,无需重新训练,同时与现有的稀疏注意力方法相比,还改善了准确性-延迟权衡。 AI
影响 这种新方法可以显著降低运行扩散 LLM 的计算成本,使其在各种应用中更易于访问和更高效。
排序理由 研究论文,详细介绍了一种提高 LLM 推理效率的新方法。[lever_c_demoted from research: ic=1 ai=1.0]
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →