跳到正文
Hugging Face Daily Papers·· 1 天前AI 评分36

LatentLoop:用非对称计算分配扩展循环 MoE 模型

LatentLoop: Scaling Looped MoE Models with Asymmetric Compute Allocation

阅读原文

本站未展示全文,请前往来源网站阅读。

AI 导读

LatentLoop 通过压缩循环 MoE 模型第一次循环的序列、在第二次保留全分辨率,并把省下的计算转给隐藏层维度,从而降低验证 NLL。在 1.3B–7B 参数、最大预算超 10^21 FLOPs 的 15 组配置中,拟合 NLL 为 1.988 时相对原生 MoE 和循环基线分别预测节省 19.0% 和 7.9% 计算,并在 9 个基准的 0-shot 与 5-shot 设置下全面提分。

来源:Hugging Face Daily Papers · arxiv.org