🤗較早收集於 19h

Ulysses 序列並行:百萬 Token 上下文訓練

Ulysses 序列並行:百萬 Token 上下文訓練
PostLinkedIn
🤗閱讀原文: Hugging Face Blog
#long-context#training-parallelism#sequence-modelingulysses-sequence-parallelismulysses-sequence-parallelismhugging-face

💡新方法訓練百萬 Token 上下文 – 長 LLM 研究突破(22字元)

⚡ 30-Second TL;DR

有什麼變化

處理百萬 Token 上下文長度

為什麼重要

加速超長上下文模型研究,提升 LLM 的推理與擷取能力。降低開源長上下文訓練門檻。

下一步行動

將 Ulysses 整合至 Hugging Face 訓練管線,用於長上下文實驗。

誰應關注:Researchers & Academics

關鍵要點

  • 處理百萬 Token 上下文長度
  • 序列級並行提升效率
  • 降低訓練記憶體使用量

🧠 深度解析

背景與延伸:來自公開資料,非原文內容。引用 7 個來源。

🔑 增強重點摘要

  • Ulysses 整合 DeepSpeed 與 Hugging Face Transformers,使用序列分片與 all-to-all 集體通訊實現注意力計算[1][2]
  • Arctic Long Sequence Training (ALST) 結合激活檢查點卸載至 CPU、Tiled MLP 計算及 Liger-kernel,提升 Llama-8B 模型上下文長度超過 400 倍[1][2]
  • 支援與 ZeRO 優化器、FlashAttention 2 等高效注意力內核無縫整合,並可與管線並行或環形注意力正交組合[2][6]
  • 在 Hugging Face Accelerate 中透過 position_ids 實現序列分片,避免注意力遮罩複製問題[3][6]
📊 競品分析▸ Show
技術主要特點記憶體效率上下文長度 (Llama-8B, 8xH100)速度
Ulysses序列並行 + all-to-all高 (ZeRO3 整合)>1M tokens基準匹配
Ring AttentionP2P 通訊中等長序列類似
UPipe注意力頭級分塊極高 (減87.5%)5M tokens匹配 Ulysses
OdysseusAttention/MLP 分離並行優於 Ulysses長序列優類似 TP-SP

🛠️ 技術深入

  • 核心演算法:沿序列長度分片輸入,每 GPU 處理局部片段,非注意力層獨立運算;注意力層使用兩階段 all-to-all 通訊重建全局上下文[2]
  • Tiled MLP 實現:猴子補丁替換 LlamaMLP.forward,使用 num_shards = ceil(seqlen / hidden),跨 rank all_reduce 確保同步,並應用 TiledMLP.apply[1]
  • 注意力模組:注入 UlyssesSPAttentionHF 後端,DistributedAttention 包裝本地注意力,透過 sequence_parallel_group 協調[1][7]
  • DataLoader 適配器處理序列分片,損失計算按片段分擔;需注意力頭數可被並行度整除[3][6]
  • 與 ZeRO-3 組合:權重分片,支援 H100 叢集訓練 Llama-8B 超長上下文[2]

🔮 前景展望AI analysis grounded in cited sources

Ulysses 將成為長上下文 LLM 訓練標準框架
其與 HF Transformers 和 DeepSpeed 的無縫整合及 400 倍上下文提升,使其適用於科學視覺轉換器和多並行設定[2]
混合 Ulysses-Ring SP 可擴展至更大序列長度
兩者正交,可與管線並行組合,解決小頭數大序列的記憶體限制[2]
單節點 8xH100 可訓練 5M+ tokens 模型
ALST 技巧如 CPU 卸載和 Tiled MLP 已證實在 Llama-8B 上實現此規模[1][2]

時間線

2023-01
DeepSpeed 引入 Ulysses 序列並行基礎技術[7]。
2023-12
Jacobs et al. 發表相關序列並行研究[2]。
2024-01
Fang et al. 擴展 Ulysses 至多並行設定[2]。
2024-10
Tsaris et al. 優化 Ulysses 整合[2]。
2025-06
Bekman et al. 發佈 ALST,展示 400 倍上下文提升[1][2]。
2026-03
Hugging Face 正式推出 Ulysses 序列並行技術[原文]。
📰

AI 週報

閱讀本週精選 AI 大事摘要 →

👉相關動態

AI 策展新聞聚合。所有內容版權歸原始發布者所有。
原始來源: Hugging Face Blog

這是摘要,不是原文。去看原站,或訂閱每週簡報。

每週 AI 簡報

每週一封,可隨時退訂。