近日，DeepSeek“開源周”活動如火如荼地進行到了第四天，一項旨在優(yōu)化并行策略的新項目正式對外公布。該項目聚焦于提升模型訓(xùn)練和推理過程中的計算與通信協(xié)同效率，力求為人工智能領(lǐng)域帶來更為出色的性能表現(xiàn)。
此次開源的核心內(nèi)容涵蓋了三大亮點：
首先，DualPipe雙向管道并行算法脫穎而出。該算法專為V3/R1訓(xùn)練設(shè)計，通過實現(xiàn)計算與通信階段的高效重疊，顯著減少了流水線中的空閑時間，即所謂的“流水線氣泡”。這一創(chuàng)新策略極大地提高了訓(xùn)練資源的利用率，使得訓(xùn)練過程更加高效。
其次，專家并行負載均衡器（EPLB）的推出，為專家并行架構(gòu)中的負載均衡問題提供了有效解決方案。在專家并行架構(gòu)中，不同的專家模塊被分配到不同的GPU上運行。然而，由于任務(wù)需求的變化，各專家模塊的工作負載也會相應(yīng)調(diào)整。為了保持GPU間的負載均衡，EPLB采用了冗余專家策略，通過復(fù)制高負載的專家模塊，并利用啟發(fā)式方法將其合理分配到不同的GPU上。為了減少節(jié)點間的通信開銷，EPLB還嘗試將同一組的專家模塊盡可能部署在同一節(jié)點上。這一策略的具體實現(xiàn)細節(jié)已在`eplb.py`文件中開源，開發(fā)者可借此工具計算出平衡的專家復(fù)制與放置方案。盡管專家負載的預(yù)測方法未在此次開源范圍內(nèi)，但開發(fā)者仍可采用移動平均法等常見手段進行估算。
最后，項目團隊還公開了V3/R1計算-通信重疊分析的相關(guān)數(shù)據(jù)。這些數(shù)據(jù)詳細展示了計算與通信重疊策略的實現(xiàn)細節(jié)及其優(yōu)化效果，為研究社區(qū)提供了深入了解和改進相關(guān)技術(shù)的寶貴資料。
關(guān)于DualPipe算法，它首次亮相于深度搜索-V3技術(shù)報告中。該算法的核心思想在于，通過雙向流水線機制實現(xiàn)正向與反向計算階段的完全重疊，從而大幅提升整體效率。而專家并行負載均衡器（EPLB）則不僅解決了專家并行架構(gòu)中的負載均衡難題，還通過優(yōu)化專家模塊的部署策略，進一步降低了節(jié)點間的通信成本。
此次開源活動，項目團隊旨在向社區(qū)提供更多實用的技術(shù)參考，推動人工智能領(lǐng)域相關(guān)技術(shù)的進一步發(fā)展。這些開源內(nèi)容和數(shù)據(jù)的發(fā)布，無疑將為研究者和開發(fā)者提供更為豐富的資源和靈感，助力他們在人工智能的道路上不斷前行。
