一種混合式稀疏模型:34 層,128 個專家
三種層,一種重複單元
- 14 記憶體· Gated DeltaNet:每個頭部都有一個線性注意力狀態,逐個 token 更新
- 13 視窗· 對最近 512 個 token 的注意力,RoPE 位置
- 7 全局· 全注意力,無位置編碼;最後一層始終為全局
每個層級都採用專家混合 (Mixture of Experts) 架構
- 每個 token 由路由器從 128 位專家中選出 4位進行路由
- 1 位共用專家始終啟用
- 每個 token有 0.6B 的參數處於活躍狀態,總參數量為 7.2B ;隱層大小為 1,024,輸入與輸出嵌入向量綁定。
最佳化器
Muon + AdamW
每次動量更新在套用前都會經過正交化處理(Newton–Schulz),因此權重矩陣的每個方向都能以相近的速度移動,而非由少數幾個方向主導。
訓練時的上下文視窗為 4,096 個 token;詞彙量為 151,643 個 token。