中國MoE一夜爆火!大模型新王暴打GPT-4o,訓練成本僅600萬美元
新智元報道
編輯:桃子 好睏
【新智元導讀】600萬美金訓出擊敗GPT-4o大模型,竟被中國團隊實現了!今天,DeepSeek-V3在全網掀起巨大風暴,僅憑671B參數在數學代碼性能上,堪比國外大模型Claude 3.5 Sonnet。
一夜之間,來自中國的大模型刷屏全網。
DeepSeek-V3,一個擁有671B參數的MoE模型,吞吐量每秒高達60 token,比上一代V2直接飆升3倍。
在多項基準測試中,V3性能直接與Claude 3.5 Sonnet、GPT-4o相匹敵。
在數學代碼方面,DeepSeek-V3完全碾壓GPT-4o。尤其是中文能力,全面領先國外的領先大模型。
就看這閃電般的推理速度,就知道模型有多強了。
值得一提的是,DeepSeek-V3在14.8T高質量token上完成了訓練,模型和論文100%開源。
論文地址:https://github.com/deepseek-ai/DeepSeek-V3/blob/main/DeepSeek_V3.pdf
新模型驚豔出世,徹底掀翻了整個AI圈。業界多位AI大佬,紛紛對此表示震驚,將重點轉向其訓練成本GPU之上。
論文中,明確提出了DeepSeek-V3僅使用2048塊GPU訓練了2個月,並且只花費了557.6萬美金。
Karpathy驚歎道,「作爲參考,要達到這種級別的能力,通常需要約1.6萬個GPU的計算集羣。不僅如此,當前業界正在部署的集羣規模甚至已經達到了10萬個GPU。
比如,Llama 3 405B消耗了3080萬GPU小時,而看起來更強大的DeepSeek-V3卻只用了280萬GPU小時(計算量減少了約11倍)。
到目前爲止,模型在實際應用中的表現相當出色——不僅在LLM競技場名列前茅,而且從Karpathy本人的快速測試來看,結果也都很不錯。
這說明,即便是在資源受限情況下,模型也能展現出令人印象深刻的研究和工程能力。
這是否意味着前沿LLM不需要大型GPU集羣?不是的,但這表明,你必須確保不浪費已有的資源,這個案例很好地證明了在數據和算法方面還有很大的優化空間」。
另外,賈揚清針對推理提出了幾點自己的思考:
首先最重要的是,我們正式進入了分佈式推理時代。一臺單GPU機器(80×8=640G)的顯存已經無法容納所有參數。雖然更新大顯存機器確實可以裝下模型,但不論如何,都需要分佈式推理來保證性能和未來擴展。
即使在單個模型中,也需要關注MoE的負載均衡,因爲每次推理只有大約5%的參數激活。
論文中特別提到引入「redundantexpert」概念,正是爲了解決這個問題。這已經不再是「一個模型多個副本」的問題、而是「每個模型子模塊都有多個副本」,然後獨立擴縮容。
輸入token很容易實現盈利。根據個人專業判斷,需要大量優化才能使輸出token盈利或實現收支平衡。但如果我們相信「軟件摩爾定律」,這就不是問題:每18個月單token成本減半。
需要進行分塊(tile)或塊(block)級別的量化。
等硬件支持FP4以後,肯定還有不少可以玩的花樣冷知識:FP4乘法實際上就是個16×16的table lookup等等……
中國模型一夜擊敗GPT-4o,100%開源
DeepSeek-V3不俗表現,是在上一代V2進一步升級和迭代。
在基準測試中,數學領域MATH 500上,DeepSeek-V3拿下了90.2高分,比Claude 3.5 Sonnet、GPT-4o超出10分還要多。
同理,在AIME 2024測試中,DeepSeek-V3也取得了領先優勢,飆升近20分。
在代碼Codeforces基準上,新模型以51.6分刷新SOTA,比國外大模型高出30分左右。
在軟件工程SWE-bench Verified基準上,DeepSeek-V3略顯遜色,Claude 3.5 Sonnet以50.8分碾壓所有模型。
另外,在多語言能力(MMLU-Pro)方面,V3提升並不明顯。知識問答基準(GPQA-Diamond)上,V3也是僅次於Claude 3.5 Sonnet。
如下這張圖表,更詳細地展示了DeepSeek-V3在各種基準測試中的結果。
53頁技術報告中,特比強調了V3的訓練成本取得了最大的突破。
團隊特意強調了,新模型的完整訓練僅需要2.788M個GPU小時。即便如此,它在訓練過程中非常穩定,沒有遇到過任何不可恢復的loss突增,也沒有執行任何rollback操作。
DeepSeek-V3訓練成本如下表1所示,這是背後團隊通過優化算法、框架、硬件協同設計最終實現的。
在預訓練階段,模型每訓練1萬億token僅需要180K個GPU小時,即在配備2048個GPU的集羣上只需3.7天。
因此,DeepSeek-V3預訓練階段耗時不到2個月就完成了,總共消耗2664K個GPU小時。
另外,再加上上下文長度scaling所需的119K GPU小時和後訓練的5K GPU小時,由此V3完整訓練僅消耗2.788M個GPU小時。
團隊表示,假設GPU的租用價格爲2美元/每GPU小時,DeepSeek-V3總訓練成本僅爲557.6萬美元。
那麼,究竟是怎樣的技術突破,使得DeepSeek-V3實現了質的飛昇?
訓練細節
正如開頭所述,DeepSeek-V3是一個強大的混合專家模型(MoE),總參數量爲爲671B,每個token激活37B參數。
它繼續採用了多頭潛在注意力(MLA)來實現高效推理,以及DeepSeekMoE實現低成本訓練。
這兩種架構的優勢,已經在上一代V2中得到了驗證。
除了基本框架之外,研究人員還採用了兩個額外的策略,來進一步增強模型的能力:
採用無輔助損失(auxiliary-loss-free)方法來實現負載均衡,目的是最小化負載均衡對V3性能造成的不利影響。
採用多token預測訓練目標,結果證明能夠提升V3在評估基準上的整體性能。
DeepSeek-V3框架
爲了實現高效訓練,團隊採用了「FP8混合精度訓練」,並對訓練框架進行了全面優化。
通過支持FP8計算和存儲,實現了訓練加速和GPU內存使用的減少。
在預訓練階段,DeepSeek-V3在14.8T高質量且多樣化的token完成了訓練,然後又對模型進行了監督微調、強化學習階段。
由此,我們纔看了DeepSeek-V3在如上評測中,性能超過了其他開源模型,並達到了與領先閉源模型相當的性能水平。
網友炸鍋了
DeepSeek-V3現在已經在官方平臺上直接可以測試,而且代碼全部開源可以直接下載。
國外AI發燒友們紛紛開啓了測試,有人直接將4/8個M4 Mac mini堆疊在一起來運行DeepSeek-V3了...
一位開發者驚訝地表示,DeepSeek-V3無需我解釋就能如此準確地理解一切,這種感覺真讓人毛骨悚然。就好像機器裡真的住着一個幽靈似的。
另有開發者通過DeepSeek-V3創建了一個用AI公司logo製作的小行星遊戲,分分鐘就完成了。
還有的人對用如此低成本,訓練出一個強大得模型,難以置信。
Stability AI前CEO表示,以每秒60個token(相當於人類閱讀速度5倍)的速度全天候運行DeepSeek v3,每天僅需要2美元。
那麼,你是要選擇一杯拿鐵咖啡,還是一個AI助手呢?
參考資料:
https://x.com/karpathy/status/1872362712958906460
https://x.com/jiayq/status/1872382450216915186