欧美性猛交xxxx免费看_牛牛在线视频国产免费_天堂草原电视剧在线观看免费_国产粉嫩高清在线观看_国产欧美日本亚洲精品一5区

0
  • 聊天消息
  • 系統(tǒng)消息
  • 評論與回復(fù)
登錄后你可以
  • 下載海量資料
  • 學(xué)習(xí)在線課程
  • 觀看技術(shù)視頻
  • 寫文章/發(fā)帖/加入社區(qū)
會員中心
創(chuàng)作中心

完善資料讓更多小伙伴認識你,還能領(lǐng)取20積分哦,立即完善>

3天內(nèi)不再提示

基于NVIDIA Megatron Core的MOE LLM實現(xiàn)和訓(xùn)練優(yōu)化

NVIDIA英偉達企業(yè)解決方案 ? 來源:NVIDIA英偉達企業(yè)解決方案 ? 2024-03-22 09:50 ? 次閱讀

本文將分享阿里云人工智能平臺 PAI 團隊與 NVIDIA Megatron-Core 團隊在 MoE (Mixture of Experts) 大語言模型(LLM)實現(xiàn)與訓(xùn)練優(yōu)化上的創(chuàng)新工作。分享內(nèi)容將按以下脈絡(luò)展開:

首先簡短回顧 MoE 技術(shù)的發(fā)展歷程,提煉核心概念及其在實踐應(yīng)用中亟待解決的關(guān)鍵挑戰(zhàn)。

接著詳述雙方合作研發(fā)的 MoE 框架所具有的獨特能力和卓越性能,展示其在提升模型訓(xùn)練效率、資源利用以及模型表現(xiàn)等方面取得的驗證結(jié)果。

最后,扼要介紹阿里云基于此合作成果所搭建的平臺工具及推薦的最佳實踐方案,賦能開發(fā)者高效運用 MoE 技術(shù),促進大規(guī)模模型訓(xùn)練的深入探索與廣泛應(yīng)用。

362ba03a-e78c-11ee-a297-92fbcf53809c.jpg

圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講

《基于 NVIDIA Megatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》

MoE 是一種模型結(jié)構(gòu),它將稠密模型結(jié)構(gòu)拆分為多個子結(jié)構(gòu),每個子結(jié)構(gòu)成為一個專家,通過在訓(xùn)練和推理過程中動態(tài)選擇一組專家進行計算,實現(xiàn)了模型參數(shù)的稀疏更新。

簡而言之,MoE 模型將整體模型拆分為多個專業(yè)子模塊(專家),每次僅激活和更新少數(shù)與輸入相關(guān)的專家子結(jié)構(gòu),而選擇哪些專家參與計算則是通過一個路由機制決定。在 GPT 等超大規(guī)模語言模型中,采用MoE 技術(shù)能夠顯著減少訓(xùn)練和推理時的計算負擔(dān),因為不是所有參數(shù)都需要在每次操作時都更新,這極大提高了訓(xùn)練效率,并且在推理階段僅使用部分活躍的網(wǎng)絡(luò)參數(shù),極大地削減了計算資源需求。

3640ff2a-e78c-11ee-a297-92fbcf53809c.jpg

圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講

《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》

2023 年末,Mistral AI 推出開源 MoE 架構(gòu)大語言模型 Mixtral 8x7B,憑借 46.7B 參數(shù)量,在多項下游任務(wù)榜單的效果勝過當(dāng)時的最佳開源稠密模型 Llama-2 70B。右側(cè)圖表揭示,在同等推理資源條件下,MoE 模型性能顯著優(yōu)于稠密模型。這一成就引發(fā)業(yè)界對 MoE 模型的強烈關(guān)注,進而推動阿里云與 NVIDIA Megatron-Core 團隊,共同深化在大模型領(lǐng)域的 MoE 技術(shù)合作與應(yīng)用。

3658ffbc-e78c-11ee-a297-92fbcf53809c.jpg

圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講

《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》

為了描述使用 MoE 結(jié)構(gòu)來實現(xiàn)工業(yè)級應(yīng)用需要解決的問題,本文首先回顧 MoE 的發(fā)展歷史。90 年代初提出的多專家模型概念為 MoE 打下了理論基礎(chǔ),即通過集合多個專家模型協(xié)同解決任務(wù),并采用門控路由進行專家選擇。

隨著 2017 年深度學(xué)習(xí)突飛猛進,參數(shù)量對模型性能的重要性日益顯現(xiàn),傳統(tǒng)稠密結(jié)構(gòu)深度學(xué)習(xí)框架面臨容量瓶頸。谷歌率先將 MoE 與深度學(xué)習(xí)結(jié)合,首次在 RNN 中驗證了 MoE 的可行性。至 2020 年,Transformer 結(jié)構(gòu)在語言模型領(lǐng)域展現(xiàn)出卓越效果和優(yōu)秀的可擴展性。此后,Google Gshard 項目將 MoE 融入 Transformer,通過深度設(shè)計與實驗顯著提升效果和性能,并在 Switch Transformers 中將參數(shù)總量推向萬億級別,奠定了 MoE 架構(gòu)在大模型領(lǐng)域的基礎(chǔ)。

后續(xù)研究如 ST-MoE、Tutel MoE、FasterMOE 和 MegaBlocks 等,針對速度和穩(wěn)定性進行深入探索。2023 年,GPT-4 的驚艷表現(xiàn)引發(fā)了業(yè)內(nèi)對其是否采用 MoE 結(jié)構(gòu)的熱議。同年,Mixtral 的實現(xiàn)似乎驗證了這一猜想,表明 MoE 在大模型架構(gòu)中的應(yīng)用正逐步走向成熟和廣泛認可。

366a86e2-e78c-11ee-a297-92fbcf53809c.jpg

圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講

《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》

后續(xù)我們將簡要剖析幾個標(biāo)志性工作,首先是將 MoE 與深度學(xué)習(xí)融為一體的 Sparsely-Gated MoE。該研究將 RNN 網(wǎng)絡(luò)劃分為多個專家子網(wǎng)絡(luò),并采用 Softmax Gating 機制來控制 token 的路由分配。然而,原始 Softmax Gating 邏輯可能導(dǎo)致負載不均衡,隨著訓(xùn)練推進,部分專家網(wǎng)絡(luò)可能過于活躍,而其他專家的參數(shù)訓(xùn)練不足。

為解決此問題,研究者在計算 Softmax 之前,對門控矩陣參數(shù)和輸入特征進行噪聲 (noise) 注入,以實現(xiàn)更為均衡的專家選擇。此外,在門控網(wǎng)絡(luò)損失函數(shù) (loss function) 中增設(shè)輔助損失,引導(dǎo)模型實現(xiàn)更佳的負載分配。同時,路由策略上嘗試每次挑選 top-2 至 top-4 的專家,這些優(yōu)化舉措最終助力模型參數(shù)容量首次突破千億門檻。

3683b5d6-e78c-11ee-a297-92fbcf53809c.jpg

圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講

《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》

而谷歌的 Gshard 項目進一步將 MoE 應(yīng)用于 Transformer 架構(gòu)中,其主要創(chuàng)新包括:

設(shè)計了 MoE Transformer 的基本結(jié)構(gòu)和并行模式,每間隔一層前向?qū)硬捎?MoE 層替代,專家分布在不同設(shè)備上獨立計算,其他層的參數(shù)共享。

針對專家負載均衡問題,提出了“專家容量(expert capacity)”概念,限制每個專家處理的 token 數(shù)量,并采用殘差連接策略繞過已滿負荷的專家。

引入輔助損失與隨機 top-k 路由策略,以優(yōu)化專家選擇過程。

通信效率上,創(chuàng)新提出“l(fā)ocal group dispatching”方案,通過門控網(wǎng)絡(luò)預(yù)篩選后再按專家 ID 分組傳輸數(shù)據(jù),有效提升通信效率。

36dc08da-e78c-11ee-a297-92fbcf53809c.jpg

圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講

《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》

實驗結(jié)果顯示,Gshard 成功將 Transformer 參數(shù)量從 150B 提升至 600B,而計算開銷僅增加 50%,首次驗證了 MoE 在模型擴展性上的顯著優(yōu)勢。

36ea3a68-e78c-11ee-a297-92fbcf53809c.jpg

圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講

《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》

之后的 Switch Transformer 工作對 MoE Transformer 的訓(xùn)練流程進行了深度優(yōu)化,以實現(xiàn)模型參數(shù)規(guī)模的極限拓展。首先,路由策略選擇了更為激進的 top-1 方案,即每次僅選擇一個專家進行參數(shù)更新。此舉不僅能顯著降低路由的計算負擔(dān),與 top-2 策略相比計算量近乎減半,同時每個專家處理的 batch size 也因此減少,進一步減輕了通信成本。

36f697e0-e78c-11ee-a297-92fbcf53809c.jpg

圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講

《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》

其次,針對先前存在的專家負載均衡問題,工作引入了輔助損失函數(shù)。該函數(shù)中,N 代表專家總數(shù),f 表示第 i 個專家分配到的 token 比例,P 代表路由器分配給第i個專家的概率。優(yōu)化目標(biāo)旨在促使 token 均勻分配至每個專家。通過最小化損失函數(shù),使得 f 和 P 趨向于 N 分之一,此時損失函數(shù)中的調(diào)節(jié)參數(shù) α 通常取值 0.01 左右,能夠?qū)崿F(xiàn)較好的負載均衡效果。

370787b2-e78c-11ee-a297-92fbcf53809c.jpg

圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講

《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》

在分布式訓(xùn)練方面,除了先前提及的數(shù)據(jù)并行和專家并行外,該工作對非專家結(jié)構(gòu)部分實施了模型并行策略,來進一步降低單卡顯存需求,使得模型參數(shù)量得以顯著提升。

針對路由負載均衡引起的模型訓(xùn)練隨機性與不穩(wěn)定問題,該工作提出了一系列穩(wěn)定訓(xùn)練的策略。其中包括在本地計算時采用較高的 FP32 精度,而在通信階段使用較低的 BF16 精度的隨機精度策略;采用具有較小縮放系數(shù)的正態(tài)分布初始化權(quán)重;引入專家 dropout 即專家內(nèi)部的 dropout 技術(shù)以減少過擬合現(xiàn)象。

通過這些優(yōu)化措施,MoE transformer 的訓(xùn)練收斂速度相較于稠密模型在迭代次數(shù)和時間上均有數(shù)倍提升。其中,最大的 Switch-C 模型參數(shù)量高達 1.5 萬億。后續(xù)的 ST-MoE 工作則更深入地探究了如何進一步改善 MoE 模型的訓(xùn)練穩(wěn)定性和提升模型性能。

3724cd2c-e78c-11ee-a297-92fbcf53809c.jpg

圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講

《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》

在預(yù)訓(xùn)練穩(wěn)定性方面,為解決專家路由隨機性和數(shù)值精度造成的 roundoff 誤差,即路由網(wǎng)絡(luò)計算 logits 值過大導(dǎo)致的訓(xùn)練不收斂問題,工作提出了 z-loss 函數(shù)。該函數(shù)中,B 表示一個 batch 的 token 數(shù)量,N 代表專家數(shù)量,X 則對應(yīng)輸入路由網(wǎng)絡(luò)的 logits 的維度。通過引入 z-loss 有效抑制了由精度等因素導(dǎo)致的logits 值過高現(xiàn)象,從而增強了模型訓(xùn)練的穩(wěn)定性,z-loss 同樣適用于稠密 LLM 訓(xùn)練的穩(wěn)定性提升。

3743920c-e78c-11ee-a297-92fbcf53809c.jpg

圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講

《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》

此外,該工作還深入探究了影響 MoE 訓(xùn)練穩(wěn)定性和效果均衡的多種因素,并通過細致實驗歸納出一系列最佳實踐。

譬如,向路由網(wǎng)絡(luò)引入適量噪聲 (noise),例如 dropout 操作,能夠增強訓(xùn)練穩(wěn)定性,但可能模型效果受損。在調(diào)優(yōu) (Fine-tuning) 階段,ST-MoE 的研究進一步發(fā)現(xiàn),增大 dropout 參數(shù)對稠密 MoE 模型和非稠密 MoE 模型的影響存在顯著差異,對于稠密模型影響更大。同時,與稠密模型不同,小型 batch size 和較大學(xué)習(xí)率 (learning rate) 對 MoE 模型訓(xùn)練效果的提升更為有利。

3762b6b4-e78c-11ee-a297-92fbcf53809c.jpg

圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講

《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》

Mixtral 模型在實現(xiàn)上借鑒了 Megablocks 框架的獨特思路,該框架的主要特點是將 MoE 層的計算表述為塊稀疏運算。面對前面提及的動態(tài)路由問題,即不同專家處理的 token 數(shù)量各異,導(dǎo)致用戶在計算過程中需抉擇使用 drop token 或 padding,前者關(guān)乎模型性能,后者則會增加訓(xùn)練成本。

Megablocks 創(chuàng)新性地將多個矩陣乘的操作統(tǒng)一定義為一個大型塊稀疏矩陣,即將多位專家的計算視作一個固定尺寸的大矩陣,其中的計算任務(wù)則細分為多個小矩陣塊的 GEMM(General Matrix Multiply)操作。

為優(yōu)化這種塊稀疏矩陣的處理,Megablocks 框架利用了 Block Compressed Sparse Row (BCSR) 數(shù)據(jù)結(jié)構(gòu),對矩陣的行和列訪問速度以及轉(zhuǎn)換操作進行了優(yōu)化。通過這種方法實現(xiàn)了路由矩陣的高效操作。

377a0e86-e78c-11ee-a297-92fbcf53809c.jpg

圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講

《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》

上述 MoE 發(fā)展史詳細說明了其開發(fā)需要解決的問題,下文將簡介阿里云為解決這些問題,與 NVIDIA Megatron-Core 團隊的技術(shù)合作及產(chǎn)出。Megatron-Core 是一個成熟且輕量化的大規(guī)模 LLM 訓(xùn)練框架,集成了訓(xùn)練大規(guī)模 LLM 的核心技術(shù),比如多元化的模型并行支持、算子優(yōu)化、通信優(yōu)化、顯存優(yōu)化,以及低精度訓(xùn)練(如 FP8)等先進技術(shù)。

Megatron-Core 沿襲了 Megatron-LM 的優(yōu)秀能力,并在代碼質(zhì)量、穩(wěn)定性、功能完備度及測試覆蓋范圍等維度全面提升。尤為關(guān)鍵的是,該框架設(shè)計上更注重解耦和模塊化,開發(fā)者在做二次開發(fā)或探索新模型架構(gòu)時享有高度靈活性。因此我們選擇了與 Megatron-Core 團隊開展合作。

379108e8-e78c-11ee-a297-92fbcf53809c.jpg

圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講

《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》

接下來探討 Megatron-Core 對 MoE 架構(gòu)的主要支持特性。在并行化方面,Megatron-Core MoE 不僅支持專家并行,還支持 3D 并行,包括數(shù)據(jù)并行、張量并行、流水并行以及序列并行等。對于超大規(guī)模 MoE 模型,它能夠靈活地將專家并行與其他并行策略有機結(jié)合。

在 token 分發(fā)機制上,Megatron-Core MoE 采用了 dropless MoE 操作,即不丟棄任何 token。在路由和負載均衡優(yōu)化層面,它支持多種路由策略,如通用的 top-k,并在負載均衡算法上支持 Sinkhorn 算法、z-loss 以及 load balancing loss 等多種方案。

此外,為解決多個專家接收變長輸入問題,Megatron-Core MoE 引入了 GroupedGMM 技術(shù),并優(yōu)化效率較低的操作,將其替換為優(yōu)化的 CUDA kernel。

同時,在模型遷移與適配上,Megatron-Core MoE 提供了豐富的模型 checkpoint 轉(zhuǎn)換功能,允許用戶導(dǎo)入HuggingFace 模型,并自由調(diào)整 TP(tensor parallelism)、PP(pipeline parallelism)和 EP(expert parallelism)等結(jié)構(gòu),隨后利用 Megatron-Core 高效啟動模型訓(xùn)練任務(wù)。

37a13fe2-e78c-11ee-a297-92fbcf53809c.jpg

圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講

《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》

阿里云人工智能平臺 PAI 團隊基于 Megatron-Core,開發(fā)了一套簡易、高效的大模型訓(xùn)練工具。依托阿里云靈駿計算服務(wù),團隊推出了 PAI-Megatron-Patch 工具庫,可實現(xiàn)從十余種主流開源大模型的模型格式到Megatron-LM 和 Megatron-Core 的無縫轉(zhuǎn)換。用戶通過 PAI DSW 和 DLC 等產(chǎn)品,能輕松啟動 Megatron-LM 及 Megatron-Core 進行大規(guī)模稠密模型和 MoE 模型訓(xùn)練。

37b3b6fe-e78c-11ee-a297-92fbcf53809c.jpg

圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講

《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》

針對開源大模型多采用 HuggingFace 格式,與 Megatron 訓(xùn)練框架存在一定差異的問題,PAI-Megatron-patch 的主要功能在于提供從 HuggingFace 模型到 Megatron 框架的權(quán)重轉(zhuǎn)換服務(wù)。該轉(zhuǎn)換實質(zhì)上是對模型命名空間進行映射,涵蓋了 layernorm 層、attention 層以及 MLP 層等核心組件的定義轉(zhuǎn)換。PAI-Megatron-patch 內(nèi)置簡潔的權(quán)重轉(zhuǎn)換腳本,使得用戶能夠便捷地執(zhí)行模型格式轉(zhuǎn)換操作,極大地簡化了遷移流程。

37c81e00-e78c-11ee-a297-92fbcf53809c.jpg

圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講

《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》

在進行 MoE 模型轉(zhuǎn)換時,需確保轉(zhuǎn)換后的文件能直接用于分布式訓(xùn)練,為此我們也做了大量驗證工作。以Mixtral 模型為例,在擁有 8 個專家和 16 張卡的場景下,采取 TP=4 和 EP=4 的切分策略。每個 TP 分區(qū)內(nèi)的文件夾中包含四個文件,存儲全部八個專家的 FFN 權(quán)重,即每個文件承載兩位專家的 FFN 權(quán)重信息。

經(jīng)過轉(zhuǎn)換,模型在零樣本(Zero-shot)損失精度方面的表現(xiàn)如途中下表所示,數(shù)據(jù)顯示轉(zhuǎn)換前后模型的精度差異非常微小,確保了模型轉(zhuǎn)換的有效性和準(zhǔn)確性。

我們的工作不僅關(guān)注模型的轉(zhuǎn)換環(huán)節(jié),同樣嚴(yán)謹(jǐn)?shù)仳炞C了整個訓(xùn)練流程的穩(wěn)健性。仍以 Mixtral 8x7B 模型為例,我們在三個訓(xùn)練階段——從頭預(yù)訓(xùn)練、基于 checkpoint 的續(xù)訓(xùn)及指令微調(diào) (Finetune),均進行了細致測試。

37e125bc-e78c-11ee-a297-92fbcf53809c.jpg

圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講

《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》

在從頭預(yù)訓(xùn)練階段,我們設(shè)定的參數(shù)包括:global_size=256,LR (learning rate) =1e-4,seq_len=2048,TP=4。經(jīng)過 24 小時至 2.4K 個訓(xùn)練步驟后,損失 loss 成功收斂至約 1.9。

37fc1b42-e78c-11ee-a297-92fbcf53809c.jpg

圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講

《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》

在繼續(xù)預(yù)訓(xùn)練階段,設(shè)定 global_size = 256,LR=5e-5,seq_len=2048,TP=4。在 18 小時達到 2,000 個訓(xùn)練步驟后,loss 亦表現(xiàn)出正常的收斂行為。

3812e85e-e78c-11ee-a297-92fbcf53809c.jpg

圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講

《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》

針對 Finetune 階段,我們做了三組測試:

橙色曲線代表 finetune base 模型,設(shè)定 global_size=128,LR=1e-5,seq_len=2048,TP=4。

藍色曲線代表 finetune instruct 模型,其參數(shù)配置與橙色曲線一致。

紅色曲線代表僅計算 answer loss 的 finetune instruct 模型,其 global_size=32,LR 依舊為1e-5,seq_len 調(diào)整為 128,TP 依舊為 4。

所有模型在訓(xùn)練過程中均呈現(xiàn)出正常收斂態(tài)勢,其中 finetune base 模型導(dǎo)致 loss 下降幅度較大,這一現(xiàn)象符合預(yù)期。

38313980-e78c-11ee-a297-92fbcf53809c.jpg

圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講

《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》

為了進一步驗證 Finetune 的效果,我們選取了一個代碼生成任務(wù)作為評估基準(zhǔn)。我們利用大約 80K 條公開的代碼訓(xùn)練樣本,測試集在 HumanEvol 平臺上進行評估,此處的參數(shù)配置沿襲了上述 SFT 設(shè)定。經(jīng)過 2,500 步訓(xùn)練后,模型在 HumanEvol 上的性能指標(biāo)從最初的 45.73% 顯著提升至 53.05%,有力證明了訓(xùn)練流程的合理性和有效性。此外,在速度對比方面,該方法優(yōu)于同等資源條件下運行的 Megablocks。

38495786-e78c-11ee-a297-92fbcf53809c.jpg

圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講

《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》

我們通過阿里云人工智能平臺 PAI 提供了一套官方最佳實踐指南,指南是對我們從數(shù)據(jù)處理、訓(xùn)練、推理、評估、直至在線服務(wù)部署的完整 AI 開發(fā)流程的測試成果總結(jié):

https://mp.weixin.qq.com/s?__biz=Mzg4MzgxNDk2OA==&mid=2247492431&idx=1&sn=1e135a9e61ac65e88a156969d66ab5be&chksm=cf430418f8348d0e1ed97db178536e73d3ae7093e428c6e3e25bc0dc84a5754f49e3c20ce98e&cur_album_id=2918365856378880004&scene=189#wechat_redirect

該指南涵蓋了從數(shù)據(jù)準(zhǔn)備階段起,指導(dǎo)用戶如何從 OSS (對象存儲服務(wù))和 NAS (網(wǎng)絡(luò)附加存儲)讀取原始數(shù)據(jù),并在 PAI DSW 環(huán)境中執(zhí)行高效的數(shù)據(jù)預(yù)處理操作。預(yù)處理后的數(shù)據(jù)能夠方便地回存至 NAS 或 OSS,以為后續(xù)的模型訓(xùn)練做好準(zhǔn)備。

這種方式的模型訓(xùn)練支持在 PAI DLC 進行大規(guī)模分布式訓(xùn)練,同時也兼容 DSW 提供的單機訓(xùn)練環(huán)境,確保用戶可根據(jù)實際需求靈活選擇訓(xùn)練方式。訓(xùn)練完成后,模型的 checkpoint 可以直接導(dǎo)出至 OSS 或 NAS 存儲系統(tǒng)。

完成訓(xùn)練的 checkpoint 可先進行離線推理和模型性能評估。一旦推理和評估驗證無誤,開發(fā)者只需一鍵即可部署至 EAS 提供的模型在線服務(wù)。依托此服務(wù)的接口,開發(fā)者可以輕松構(gòu)建各類 APP 和業(yè)務(wù)場景。

38591d9c-e78c-11ee-a297-92fbcf53809c.jpg

圖片來源于 GTC 2024 大會 China AI Day 線上專場的演講

《基于 NVIDIAMegatron-Core 的 MoE LLM 實現(xiàn)和訓(xùn)練優(yōu)化》

這份最佳實踐指南提供了詳細的步驟說明,確保用戶能清晰掌握每一步驟的操作方法。

未來,阿里云人工智能平臺 PAI 團隊將繼續(xù)深化與 NVIDIA Megatron-Core 團隊的合作,致力于在密集型和稀疏型模型的訓(xùn)練表現(xiàn)和效率上取得更大突破,為推進 AGI(通用人工智能)技術(shù)的發(fā)展貢獻力量。我們熱忱歡迎全球開發(fā)者共同參與到開源社區(qū)項目以及阿里云的建設(shè)之中,攜手共進,共創(chuàng)智能未來。




審核編輯:劉清

聲明:本文內(nèi)容及配圖由入駐作者撰寫或者入駐合作網(wǎng)站授權(quán)轉(zhuǎn)載。文章觀點僅代表作者本人,不代表電子發(fā)燒友網(wǎng)立場。文章及其配圖僅供工程師學(xué)習(xí)之用,如有內(nèi)容侵權(quán)或者其他違規(guī)問題,請聯(lián)系本站處理。 舉報投訴
  • NVIDIA
    +關(guān)注

    關(guān)注

    14

    文章

    5078

    瀏覽量

    103770
  • 人工智能
    +關(guān)注

    關(guān)注

    1796

    文章

    47734

    瀏覽量

    240411
  • GPT
    GPT
    +關(guān)注

    關(guān)注

    0

    文章

    360

    瀏覽量

    15523
  • 阿里云
    +關(guān)注

    關(guān)注

    3

    文章

    976

    瀏覽量

    43247
  • LLM
    LLM
    +關(guān)注

    關(guān)注

    0

    文章

    300

    瀏覽量

    407

原文標(biāo)題:GTC24|SE63222 線上演講回顧:基于 NVIDIA Megatron Core 的 MOE LLM 實現(xiàn)和訓(xùn)練優(yōu)化

文章出處:【微信號:NVIDIA-Enterprise,微信公眾號:NVIDIA英偉達企業(yè)解決方案】歡迎添加關(guān)注!文章轉(zhuǎn)載請注明出處。

收藏 人收藏

    評論

    相關(guān)推薦

    解析DeepSeek MoE并行計算優(yōu)化策略

    本期Kiwi Talks將從集群Scale Up互聯(lián)的需求出發(fā),解析DeepSeek在張量并行及MoE專家并行方面采用的優(yōu)化策略。DeepSeek大模型的工程優(yōu)化以及國產(chǎn)AI 產(chǎn)業(yè)鏈的開源與快速部署預(yù)示著國產(chǎn)AI網(wǎng)絡(luò)自主自控將大
    的頭像 發(fā)表于 02-07 09:20 ?294次閱讀
    解析DeepSeek <b class='flag-5'>MoE</b>并行計算<b class='flag-5'>優(yōu)化</b>策略

    NVIDIA TensorRT-LLM中啟用ReDrafter的一些變化

    Recurrent Drafting (簡稱 ReDrafter) 是蘋果公司為大語言模型 (LLM) 推理開發(fā)并開源的一種新型推測解碼技術(shù),該技術(shù)現(xiàn)在可與 NVIDIA TensorRT-LLM 一起使用。
    的頭像 發(fā)表于 12-25 17:31 ?266次閱讀
    在<b class='flag-5'>NVIDIA</b> TensorRT-<b class='flag-5'>LLM</b>中啟用ReDrafter的一些變化

    解鎖NVIDIA TensorRT-LLM的卓越性能

    NVIDIA TensorRT-LLM 是一個專為優(yōu)化大語言模型 (LLM) 推理而設(shè)計的庫。它提供了多種先進的優(yōu)化技術(shù),包括自定義 Att
    的頭像 發(fā)表于 12-17 17:47 ?284次閱讀

    NVIDIA TensorRT-LLM Roadmap現(xiàn)已在GitHub上公開發(fā)布

    感謝眾多用戶及合作伙伴一直以來對NVIDIA TensorRT-LLM的支持。TensorRT-LLM 的 Roadmap 現(xiàn)已在 GitHub 上公開發(fā)布!
    的頭像 發(fā)表于 11-28 10:43 ?341次閱讀
    <b class='flag-5'>NVIDIA</b> TensorRT-<b class='flag-5'>LLM</b> Roadmap現(xiàn)已在GitHub上公開發(fā)布

    TensorRT-LLM低精度推理優(yōu)化

    本文將分享 TensorRT-LLM 中低精度量化內(nèi)容,并從精度和速度角度對比 FP8 與 INT8。首先介紹性能,包括速度和精度。其次,介紹量化工具 NVIDIA TensorRT Model
    的頭像 發(fā)表于 11-19 14:29 ?466次閱讀
    TensorRT-<b class='flag-5'>LLM</b>低精度推理<b class='flag-5'>優(yōu)化</b>

    如何訓(xùn)練自己的LLM模型

    訓(xùn)練自己的大型語言模型(LLM)是一個復(fù)雜且資源密集的過程,涉及到大量的數(shù)據(jù)、計算資源和專業(yè)知識。以下是訓(xùn)練LLM模型的一般步驟,以及一些關(guān)鍵考慮因素: 定義目標(biāo)和需求 : 確定你的
    的頭像 發(fā)表于 11-08 09:30 ?830次閱讀

    LLM和傳統(tǒng)機器學(xué)習(xí)的區(qū)別

    訓(xùn)練方法 LLM: 預(yù)訓(xùn)練和微調(diào): LLM通常采用預(yù)訓(xùn)練(Pre-training)和微調(diào)(Fine-tuning)的方法。預(yù)
    的頭像 發(fā)表于 11-08 09:25 ?817次閱讀

    端到端InfiniBand網(wǎng)絡(luò)解決LLM訓(xùn)練瓶頸

    ChatGPT對技術(shù)的影響引發(fā)了對人工智能未來的預(yù)測,尤其是多模態(tài)技術(shù)的關(guān)注。OpenAI推出了具有突破性的多模態(tài)模型GPT-4,使各個領(lǐng)域取得了顯著的發(fā)展。 這些AI進步是通過大規(guī)模模型訓(xùn)練實現(xiàn)
    的頭像 發(fā)表于 10-23 11:26 ?559次閱讀
    端到端InfiniBand網(wǎng)絡(luò)解決<b class='flag-5'>LLM</b><b class='flag-5'>訓(xùn)練</b>瓶頸

    NVIDIA Nemotron-4 340B模型幫助開發(fā)者生成合成訓(xùn)練數(shù)據(jù)

    Nemotron-4 340B 是針對 NVIDIA NeMo 和 NVIDIA TensorRT-LLM 優(yōu)化的模型系列,該系列包含最先進的指導(dǎo)和獎勵模型,以及一個用于生成式 AI
    的頭像 發(fā)表于 09-06 14:59 ?387次閱讀
    <b class='flag-5'>NVIDIA</b> Nemotron-4 340B模型幫助開發(fā)者生成合成<b class='flag-5'>訓(xùn)練</b>數(shù)據(jù)

    LLM預(yù)訓(xùn)練的基本概念、基本原理和主要優(yōu)勢

    在人工智能和自然語言處理(NLP)領(lǐng)域,大型語言模型(Large Language Model,簡稱LLM)的興起極大地推動了技術(shù)的進步和應(yīng)用的發(fā)展。LLM通過在大規(guī)模文本數(shù)據(jù)上進行預(yù)訓(xùn)練,獲得了
    的頭像 發(fā)表于 07-10 11:03 ?1335次閱讀

    llm模型訓(xùn)練一般用什么系統(tǒng)

    LLM(Large Language Model,大型語言模型)是近年來在自然語言處理領(lǐng)域取得顯著成果的一種深度學(xué)習(xí)模型。它通常需要大量的計算資源和數(shù)據(jù)來進行訓(xùn)練。以下是關(guān)于LLM模型訓(xùn)練
    的頭像 發(fā)表于 07-09 10:02 ?503次閱讀

    llm模型和chatGPT的區(qū)別

    基于Transformer架構(gòu)的預(yù)訓(xùn)練語言模型,它可以生成連貫、自然的文本。ChatGPT使用GPT模型作為基礎(chǔ),通過微調(diào)和訓(xùn)練實現(xiàn)對話生成和理解。 以下是一
    的頭像 發(fā)表于 07-09 09:55 ?1347次閱讀

    進一步解讀英偉達 Blackwell 架構(gòu)、NVlink及GB200 超級芯片

    能比上一代NVIDIA Hopper?提高了15倍。采用第二代Transformer引擎、定制的Blackwell Tensor Core技術(shù)、TensorRT?-LLM和Nemo?框架的創(chuàng)新,顯著加速了
    發(fā)表于 05-13 17:16

    NVIDIA加速微軟最新的Phi-3 Mini開源語言模型

    NVIDIA 宣布使用 NVIDIA TensorRT-LLM 加速微軟最新的 Phi-3 Mini 開源語言模型。TensorRT-LLM 是一個開源庫,用于
    的頭像 發(fā)表于 04-28 10:36 ?643次閱讀

    MOE與MOT:提升LLM效能的關(guān)鍵策略比較

    MoE 與 MoT:在專家混合中(左),每個令牌都被路由到不同的專家前饋層。在令牌混合(右)中,每組內(nèi)的令牌被混合,并且混合令牌由專家前饋層處理。
    的頭像 發(fā)表于 04-15 09:53 ?978次閱讀
    <b class='flag-5'>MOE</b>與MOT:提升<b class='flag-5'>LLM</b>效能的關(guān)鍵策略比較