←新聞 11 / 12→

硬體基建

SLAI T-Rex:昇騰 SuperPOD 上 DeepSeek-V4 全參數後訓練優化

SLAI T-Rex: Full-Parameter Post-training of the DeepSeek-V4 Family on Ascend SuperPOD

SLAI T-Rex:昇騰 SuperPOD 上 DeepSeek-V4 全參數後訓練優化

arXiv cs.CL · 2026-07-23

摘要

研究團隊在華為昇騰 NPU 超級計算集群上成功完成了 DeepSeek-V4 兆參數級 MoE 模型的全參數後訓練,開發了涵蓋模型並行、計算通信協調和低層核心執行的分層優化框架。相比開源基線方案,該系統將模型浮點運算利用率(MFU)提升至 34.22%,性能提升了 2.93 倍,並在此基礎上建立了複雜運籌最佳化任務的 CPT 和 SFT 工作流。

研究團隊在華為昇騰 NPU SuperPOD 超級計算集群上,針對 DeepSeek-V4 兆參數級 MoE 模型完成了全參數後訓練。面對大規模分佈式訓練中嚴重的記憶體壓力、非重疊通訊開銷以及低效核心執行等系統層級挑戰,團隊開發了名為 SLAI T-Rex 的分層優化框架。該框架涵蓋模型層級並行、計算與通訊協調以及低層核心執行,旨在解決現有基於 GPU 集群的訓練系統所面臨的瓶頸。

優化後的系統在模型浮點運算利用率(MFU)上達到 34.22%,相比開源基線方案提升了 2.93 倍,同時維持了訓練的穩定性。基於此優化基礎設施,團隊進一步建立了針對複雜運籌最佳化(OR)任務的 CPT 和 SFT 工作流。透過 DeepSeek-V4-Flash 模型,團隊開發了面向 OR 的 CPT 和 SFT 數據管道,結合收集的領域資源與求解器驗證的合成最佳化文件。

最終構建的數據集包含 10K 個高品質 SFT 樣本,涵蓋四個任務類別與三種問題表示形式。在評估的模型中,該專用模型取得了最高的平均零樣本 Pass@1 分數,達到 71.81%。此成績優於 GPT-5.4-Mini 模型 3.98 個百分點,也優於基礎 DeepSeek-V4-Flash 模型 11.27 個百分點。

●開發者:可參考該優化框架提升大規模 MoE 模型訓練效率

●投資人:國產 NPU 芯片在大模型訓練中的競爭力持續增強

●一般用戶:推動國內高效能 AI 基礎設施發展,有助於降低大模型部署成本

重要性評分

76/100

🟠 值得關注

大模型訓練MoE 架構NPU 優化
原文出處
上一則← OpenAI 推進美國科學研究新時代下一則護欄成替罪羊:審計工具增強型 LLM 代理中的不誠實安全拒絕 →

喜歡這篇?每天早晨還有更多。

訂閱 5min AI,讓 AI 替你追蹤整個 AI 世界。

相關指南

🤖 本文摘要由 AI 自動生成,內容源自原始報導。如有疑慮,請參閱關於我們。

喜歡這篇?每天早晨還有更多。

訂閱 5min AI,讓 AI 替你追蹤整個 AI 世界。