一种混合型稀疏模型:34 层,128 个专家
三种类型的层,一个重复单元
- 14 内存· Gated DeltaNet:每个头具有一个线性注意力状态,逐个 token 更新
- 13 窗口· 对最近 512 个 token 的注意力,RoPE 位置
- 7 全局· 全局注意力,无位置编码;最后一层始终为全局
每一层都采用“专家混合”(Mixture of Experts)机制
- 对于每个 token,路由器会从 128 个路由专家中选择 4 个
- 1 个共享专家始终处于启用状态
- 每个 token 激活7.2B 参数中的 0.6B ;隐藏层大小为 1,024,输入与输出嵌入共享
优化器
Muon + AdamW
每次动量更新在应用前都会进行正交化(Newton–Schulz),因此权重矩阵的每个方向都以相近的速度变化,而非由少数方向主导。
训练时使用 4,096 个 token 的上下文窗口;词汇量为 151,643 个 token。