全部文章0

科学空间苏剑林··访问 1

DeepSeek V4的tid2eid是怎么来的?

原网页
原始内容

训过MoE的同学都知道,如果把整个模型的MLP部分都换成MoE,那么靠近Embedding的前几层MoE往往很难实现负载均衡。对此,DeepSeek V3,包括我们的Kimi K2,采取的应对策略是“first_k_dense”——顾名思义,就是前$k$层不用MoE而用常规的Dense型GLU。到了DeepSeek V4,这个策略改成了“first_k_hash”。

这里的“hash”指的是“Hash Routing”,提出自《Hash Layers For Large Sparse Models》,它通过一张事先确定的映射表tid2eid来为每个Token分配Expert。本文来探讨一下如何生成这个tid2eid。

[...]