PulseAugur
实时 17:09:19
English(EN) Mixture-of-Depths: a per-block router keeps top-k tokens, the rest skip the residual — dynamic depth at a fixed FLOP budget

Mixture-of-Depths 动态分配 Transformer 计算量,节省约 50% 的 FLOPs

研究人员开发了一种名为 Mixture-of-Depths (MoD) 的新颖技术,该技术可以动态分配 Transformer 模型内的计算资源。通过在每个块中引入一个小型路由器,MoD 仅识别和处理最重要的 token,允许不太关键的 token 跳过计算并沿残差连接传递。这种方法通过在固定的总预算下为每个 token 动态分配深度,显著减少了 FLOPs,以大约 50% 的计算量实现了与密集模型相当的性能。 AI

影响 这项技术可以通过降低计算成本而不牺牲性能,从而实现更高效的 LLM 训练和推理。

排序理由 详细介绍新模型架构技术的论文。[lever_c_demoted from research: ic=1 ai=1.0]

在 dev.to — LLM tag 阅读 →

AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →

Mixture-of-Depths 动态分配 Transformer 计算量,节省约 50% 的 FLOPs

报道来源 [1]

  1. dev.to — LLM tag TIER_1 English(EN) · Devanshu Biswas ·

    Mixture-of-Depths: a per-block router keeps top-k tokens, the rest skip the residual — dynamic depth at a fixed FLOP budget

    <p>A normal transformer forces <em>every</em> token through <em>every</em> layer, so the word "the" costs exactly as much compute as "photosynthesis." That's obviously wasteful, and Mixture-of-Depths (DeepMind, 2024) is the fix that finally made the waste addressable to me. It ad…