DeepSeek開源第四彈:并行優(yōu)化策略3項齊開

2月27日消息,DeepSeek開源周第四日,DeepSeek開源了并行優(yōu)化策略(Optimized Parallelism Strategies),一次開源了3項:

DualPipe:一種用于V3/R1模型訓練中實現(xiàn)計算與通信重疊的雙向流水線并行算法

EPLB:一個針對V3/R1的專家并行負載均衡工具

Profile-data:訓練和推理框架的分析數(shù)據(jù)

GitHub地址:

https://github.com/deepseek-ai/DualPipe

https://github.com/deepseek-ai/eplb

https://github.com/deepseek-ai/profile-data

一、DualPipe

DualPipe是DeepSeek-V3技術報告中提出的一種創(chuàng)新雙向流水線并行算法。

它能夠實現(xiàn)前向與后向計算和通信階段的完全重疊,同時有效減少流水線氣泡(空閑時間)。

DeepSeek展示了在8個流水線并行階段和20個micro-batches情況下,DualPipe在兩個方向上的調度示例。(來自DeepSeek-V3技術報告)

反向方向的微批次與前向方向對稱,為了簡化圖示,這里省略了反向方向的批次 ID。圖中由同一個黑色邊框包圍的兩個單元格具有相互重疊的計算和通信。

流水線氣泡與內存使用比較

表中,

免責聲明:本網(wǎng)站內容主要來自原創(chuàng)、合作伙伴供稿和第三方自媒體作者投稿,凡在本網(wǎng)站出現(xiàn)的信息,均僅供參考。本網(wǎng)站將盡力確保所提供信息的準確性及可靠性,但不保證有關資料的準確性及可靠性,讀者在使用前請進一步核實,并對任何自主決定的行為負責。本網(wǎng)站對有關資料所引致的錯誤、不確或遺漏,概不負任何法律責任。任何單位或個人認為本網(wǎng)站中的網(wǎng)頁或鏈接內容可能涉嫌侵犯其知識產(chǎn)權或存在不實內容時,應及時向本網(wǎng)站提出書面權利通知或不實情況說明,并提供身份證明、權屬證明及詳細侵權或不實情況證明。本網(wǎng)站在收到上述法律文件后,將會依法盡快聯(lián)系相關文章源頭核實,溝通刪除相關內容或斷開相關鏈接。

2025-02-27
DeepSeek開源第四彈:并行優(yōu)化策略3項齊開
DeepSeek展示了在8個流水線并行階段和20個micro-batches情況下,DualPipe在兩個方向上的調度示例。反向方向的微批次與前向方向對稱,為了簡化圖示,這里省略了反向方向的批次 ID。如DeepSeek-V3論文所述,研究人員采用了冗余專家策略,對高負載專家進行復制。為了便于復制和部署,DeepSeek團隊在eplb.py中開源了部署的EP負載平衡算法。同時,DeepSeek還公開了這些數(shù)據(jù)的訓練和推理過程。

長按掃碼 閱讀全文