研究人员开发了一种名为 Mixture-of-Depths (MoD) 的新颖技术,该技术可以动态分配 Transformer 模型内的计算资源。通过在每个块中引入一个小型路由器,MoD 仅识别和处理最重要的 token,允许不太关键的 token 跳过计算并沿残差连接传递。这种方法通过在固定的总预算下为每个 token 动态分配深度,显著减少了 FLOPs,以大约 50% 的计算量实现了与密集模型相当的性能。 AI
影响 这项技术可以通过降低计算成本而不牺牲性能,从而实现更高效的 LLM 训练和推理。
排序理由 详细介绍新模型架构技术的论文。[lever_c_demoted from research: ic=1 ai=1.0]
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →