Layerwise Repacking:混合注意力变长训练免重分配
原文标题:Layerwise Repacking: Variable-Length Hybrid-Attention Training without an Extra Redistribution Pass
重要度 3/5 · AI 综合评估报道影响力与新鲜度,4–5 分为高价值资讯。
论文提出 Layerwise Repacking,在混合注意力模型中保留 SP16/32 父组处理全注意力,将样本重打包进节点内 SP1/2/4/8 稀疏子组,并复用现有 Ulysses 通信点完成头部交换与重打包,避免额外重分配。在 1F3S 混合注意力模型前 200 步的 22 个步骤上,端到端时间降低 5.95%,MFU 提升 3.75 个百分点;配合改进求解器分别达 7.61% 与 4.83 个百分点。
深读 · 结论先行AI 生成展开看要点与影响 ↓
Layerwise Repacking在混合注意力训练中端到端时间降5.95%、MFU升3.75个百分点。
- •全注意力层保留SP16/32父组,稀疏注意力层将样本重打包进节点内SP1/2/4/8子组
- •跨节点all-to-all实测有效速率低7.0倍,这是避免宽组用于稀疏层的动因
- •重打包复用两个全层Ulysses通信点,一次变长all-to-all同时完成头交换与样本重打包,不增加重分配
- •辅助父组求解器在子组成形前重新平衡父组成员
- •1F3S模型前200步中全部22个相关步骤端到端时间降5.95%、MFU升3.75个百分点;配合改进求解器达7.61%与4.83个百分点
为什么重要:针对混合注意力变长训练中并行度冲突给出免额外重分配的工程方案,对大规模训练系统优化有直接价值。
以上深读由 AI 依据标题/摘要/正文节选生成,仅供参考;事实以原文为准。
AI 处理过程展开/收起
相关推荐
OpenAI 据悉计划采购谷歌 TPU 芯片,降低对英伟达依赖
AI 摘要据报道,OpenAI 正计划采购谷歌自研的张量处理单元(TPU),以扩大算力规模并实现芯片供应多元化,双方尚未敲定最终条款。OpenAI 同时在推进与博通合作、代号 Jalapeno 的自研 AI 芯片。若交易落地,将直接冲击英伟达在 AI 加速芯片市场的主导地位,并重塑云厂商与大模型公司的算力供应格局。
阿联酋因伊朗袭击调整 5GW AI 数据中心计划
AI 摘要据路透社报道,在遭受伊朗袭击后,阿联酋修订了其 5GW AI 数据中心建设计划。该事件凸显地缘政治冲突对中东地区大规模 AI 基础设施投资的直接冲击,可能影响全球算力布局与供应链。
英伟达预计单季营收达1080亿美元
AI 摘要英伟达预测未来几个月内营收将达到1080亿美元,有望成为继亚马逊、苹果和Alphabet之后又一家季度营收超千亿美元的公司。其数据中心营收同比增长一倍以上,利润也大幅增长。
亚马逊计划在 AWS 之外销售 Trainium AI 芯片
AI 摘要亚马逊计划将自研的 Trainium AI 芯片对外销售,不再仅限于 AWS 云服务内部使用。此举意味着亚马逊将直接与英伟达等芯片厂商在 AI 加速器市场展开竞争,拓展其芯片业务的商业化路径。