🤖較早收集於 19m

Kimi 注意力殘差解決 LLM 深度問題

PostLinkedIn
🤖閱讀原文: Reddit r/MachineLearning
#residual-connections#scaling-lawsattnreskimiattnreskimi-linear

💡新型殘差修正提升深度 LLM 效能與穩定性(48B 驗證),跨尺度適用。(38字)

⚡ 30-Second TL;DR

有什麼變化

以輸入依賴注意力取代均勻殘差,涵蓋先前層。

為什麼重要

AttnRes 提供低開銷殘差即插即用方案,實現穩定訓練的更深 LLM。跨尺度一致改善使其對生產 LLM 開發極具價值。

下一步行動

閱讀 arXiv 2603.15031 並在您的 PyTorch LLM 中實作 Block AttnRes。

誰應關注:Researchers & Academics

關鍵要點

  • 以輸入依賴注意力取代均勻殘差,涵蓋先前層。
  • Block AttnRes 透過層分割與區塊摘要降低記憶體。
  • 實驗證實跨模型尺寸的縮放法則獲益。
  • 在 48B Kimi Linear 上緩解 PreNorm 稀釋(1.4T 權杖)。
  • 改善跨深度的均勻輸出幅度與梯度流動。

🧠 深度解析

背景與延伸:來自公開資料,非原文內容。引用 5 個來源。

🔑 增強重點摘要

  • AttnRes 使用每個層的固定學習偽查詢向量(pseudo-query vector),而非輸入依賴查詢,並對前層輸出應用 RMSNorm 以防止幅度主導注意力權重。[1][3]
  • 在 Kimi Linear MoE 架構中,AttnRes 帶來 GPQA-Diamond +7.5 分、Minerva Math +3.6 分及 HumanEval 顯著提升,尤其在多步推理任務。[1][3]
  • Block AttnRes 將層分為區塊並對區塊摘要進行跨區塊注意力,匹配需 1.25 倍計算的基準模型損失。[1][3]

🛠️ 技術深入

  • Full AttnRes:每個層使用固定偽查詢向量 w_l ∈ R^d,對所有前層輸出(經 RMSNorm)計算 softmax 注意力,初始權重設為零以確保訓練起始均勻。[1][3]
  • Block AttnRes:將層分區塊,區塊內輸出求和成單一表示,僅對區塊摘要應用注意力以降低記憶體需求。[1]
  • Kimi Linear 整合:48B 總參數、3B 激活參數 MoE,使用 3:1 KDA(Kimi Delta Attention)與 MLA(Multi-Head Latent Attention)混合,每四層一全注意力層,無 RoPE 於 MLA 以支援長上下文。[2][3][4]
  • KDA:基於 Gated DeltaNet 的線性注意力,引入通道級閘門(channel-wise gating),使用 Diagonal-Plus-Low-Rank (DPLR) 轉移矩陣變體提升硬體效率,KV 快取減 75%、解碼速度達 6 倍於 1M 上下文。[4][5]

🔮 前景展望AI analysis grounded in cited sources

AttnRes 將成為深度 LLM 標準殘差替代,提升超過 1.25 倍計算效率的模型效能
縮放實驗顯示 Block AttnRes 匹配需更多計算的基準損失,並在 48B 模型中改善梯度與基準表現。[1][3]
Kimi Linear 混合注意力將取代純全注意力架構於長上下文代理任務
其在短/長上下文與 RL 基準優於全注意力,KV 快取減 75% 並加速 6 倍解碼。[4][5]

時間線

2026-03
Moonshot AI 發布 AttnRes 論文與整合至 Kimi Linear 48B MoE 模型
2026-03
Kimi Linear arXiv 論文發布,展示 KDA 與 MLA 混合優於全注意力
📰

AI 週報

閱讀本週精選 AI 大事摘要 →

👉相關動態

AI 策展新聞聚合。所有內容版權歸原始發布者所有。
原始來源: Reddit r/MachineLearning

這是摘要,不是原文。去看原站,或訂閱每週簡報。

每週 AI 簡報

每週一封,可隨時退訂。