專案檔案
Zero-Bubble Pipeline Parallelism
新型流水線並行訓練效率最佳化
GitHub Stars
452
方向
流水線並行
目標
減少空泡
- 機構
- Sea AI Lab (SAIL)
- 分組
- 國際企業實驗室
- 類別
- 訓練效率最佳化
- 狀態
- 研究開源
- 啟動
- 2023-11
- 語言 / 形態
- Python
- 協議
- 未明確
- GitHub Stars
- 461
- 資訊更新
- 2026-05-04
Zero-Bubble Pipeline Parallelism 是 Sea AI Lab 在大模型訓練效率上的系統研究,目標是減少流水線並行中的空閒時間。
是什麼
流水線並行會把模型切成多個階段,在不同裝置上順序處理 micro-batches。問題是裝置之間容易出現等待,也就是"bubble"。
Zero-Bubble 的思路是改進排程和反向傳播安排,讓裝置更少空轉,提高大模型訓練吞吐。
與 AI 的關係
訓練效率是基礎模型競爭的隱藏戰場。少一點空轉,就意味著同樣算力可以訓練更多 token、更大模型或更快迭代。
這種系統論文和開源實現對大模型實驗室很有實際價值。
與新加坡的關係
Sea AI Lab 做這類訓練系統研究,說明新加坡本土企業實驗室正在進入更底層的模型基礎設施層,而不僅是應用層。
它和 Colossal-AI 一起構成新加坡開源生態中"訓練系統"這一條線。
關鍵里程碑
- 2023-11Zero-Bubble 倉庫建立