Marin 社区实验:YOCO 变体复用中点激活替代 KV,有效加速达 1.12 倍

liliang_ren · x · 2026-09-09

Marin 开源训练社区的一个 agent 生成实验:在 YOCO 风格注意力上,让后半层所有层的 K/V 不再取自各自的残差流,而是共享一个中点(midpoint)激活。参数与层数完全保留。

在三个固定种子实验中,d512/d768/d1024 三个宽度的终端 Paloma loss 全部下降(如 d512 从 3.5794 降至 3.5570),训练吞吐仅损失 0.6-0.7%,综合有效加速分别为 1.121x/1.025x/1.077x。核心动机在推理 prefill:前半层算完后,上半部分缓存只需各层做 K/V 投影 GEMM。

原文链接 →

「研究」频道最新

更多「研究」频道 AI 资讯 →