论文:短多项式程序加速 LLM 特殊函数运算
原文标题:Fast Polynomial Transcendentals for LLMs
重要度 3/5 · AI 综合评估报道影响力与新鲜度,4–5 分为高价值资讯。
论文探索用短多项式程序加速大模型中的特殊函数单元运算,在 NVIDIA Blackwell 上替换 sigmoid、tanh、SiLU 等为三至四次 BF16 多项式。隔离路径提速 1.19–2.19 倍,训练步吞吐提升最高 8%,且训练损失差异极小。
深读 · 结论先行AI 生成展开看要点与影响 ↓
用三至四次BF16多项式替换sigmoid/tanh/SiLU,训练步吞吐最高提升8%。
- •在NVIDIA Blackwell上,将sigmoid、tanh、SiLU替换为三至四次BF16多项式程序
- •隔离路径在L2提速1.19–2.19倍,在HBM提速1.00–1.70倍
- •dense-SiLU、tanh-softcapped attention、routed-expert SwiGLU分别提升完整训练步吞吐2.7%、2.9%、8.0%
- •sigmoid attention替换使完整attention前向提升7.4%,但完整GPU步仅提升0.3%
- •训练损失差异极小,并做了同检查点开放权重消融与配对预训练对比
为什么重要:GPU各流水线扩展速度不均,SFU成为新瓶颈;该结果提示用短多项式程序可低成本换取训练吞吐。
以上深读由 AI 依据标题/摘要/正文节选生成,仅供参考;事实以原文为准。
AI 处理过程展开/收起
相关推荐
OpenAI 据悉计划采购谷歌 TPU 芯片,降低对英伟达依赖
AI 摘要据报道,OpenAI 正计划采购谷歌自研的张量处理单元(TPU),以扩大算力规模并实现芯片供应多元化,双方尚未敲定最终条款。OpenAI 同时在推进与博通合作、代号 Jalapeno 的自研 AI 芯片。若交易落地,将直接冲击英伟达在 AI 加速芯片市场的主导地位,并重塑云厂商与大模型公司的算力供应格局。
阿联酋因伊朗袭击调整 5GW AI 数据中心计划
AI 摘要据路透社报道,在遭受伊朗袭击后,阿联酋修订了其 5GW AI 数据中心建设计划。该事件凸显地缘政治冲突对中东地区大规模 AI 基础设施投资的直接冲击,可能影响全球算力布局与供应链。
英伟达预计单季营收达1080亿美元
AI 摘要英伟达预测未来几个月内营收将达到1080亿美元,有望成为继亚马逊、苹果和Alphabet之后又一家季度营收超千亿美元的公司。其数据中心营收同比增长一倍以上,利润也大幅增长。
Marvell 与 Lumentum 宣布合并,打造 AI 连接领域领导者
AI 摘要Marvell 与 Lumentum 宣布合并,双方称将组建 AI 连接(AI Connectivity)领域的品类领导者。交易旨在整合光通信与互连能力,服务数据中心 AI 基础设施需求。