英伟达 Blackwell 上的 FLAMIA 注意力内核:低精度训练提速最高 1.42 倍

针对 NVIDIA Blackwell GPU 的注意力计算,研究者提出 FLAMIA 内核方案,用 MXFP8 和 NVFP4/FP8 混合精度路径,在带因果掩码的 packed 文档训练负载上,把含量化的前向加反向注意力调用提速约 6% 至 42%。

一句话看懂:针对 NVIDIA Blackwell GPU 的注意力计算,研究者提出 FLAMIA 内核方案,用 MXFP8 和 NVFP4/FP8 混合精度路径,在带因果掩码的 packed 文档训练负载上,把含量化的前向加反向注意力调用提速约 6% 至 42%。

事件核心:发生了什么

根据 Hugging Face Papers 收录的论文摘要(发布时间为 2026 年 11 月 2 日),研究团队推出 FLAMIA,一组面向 NVIDIA Blackwell 架构的注意力内核,支持 MXFP8 以及 NVFP4 与 FP8 混合的训练路径。其设计针对三个具体痛点:量化带来的额外开销、因果注意力中共享量化 scale 可能导致的信息泄漏,以及前向与反向计算数值不一致。方案引入 token-block 缩放以在不同矩阵方向复用已量化操作数,并做针对性防护;同时用文档内局部缩放和工作负载感知调度,直接在 packed 布局中处理变长文档,无需重打包。

摘要给出了一组有条件的测速结果:在多种因果 packed 文档工作负载中,MXFP8 路径相对各负载下最快的对比基线,B200 上提速 1.10 至 1.40 倍,B300 上提速 1.06 至 1.42 倍,统计口径包含量化步骤。作者还在 Llama 风格模型的预训练上验证了训练稳定性和质量保持。

为什么重要

长上下文训练中,注意力是算力和显存的主要消耗项之一。Blackwell 提供了高吞吐低精度矩阵乘法能力,但要让这种硬件优势真正变成训练速度,难点不在矩阵乘本身,而在量化开销、因果安全性和前后向数值一致性这些工程细节。FLAMIA 的价值在于把量化、训练数值和文档调度放在一起设计,尝试把低比特硬件的理论吞吐转化为实际注意力加速。目前公开信息显示,这仍是一篇论文层面的工作,尚未核验全文实验,也不等于已上线的产品功能。

对用户/开发者/创作者的影响

对使用大模型训练的开发者而言,这类内核如果能稳定落地,意味着在 B200/B300 集群上处理长上下文、多文档打包训练时,有可能获得可观的端到端提速,同时减少为了低精度训练而重打包数据或额外修补数值一致性的工程负担。对以长文档处理为卖点的 AI 应用来说,训练侧的算力效率提升最终可能影响成本结构,但这取决于内核是否被主流训练框架吸收、是否开源可复现。创作者层面短期没有直接变化,训练成本下降传导到产品价格通常需要更长时间。

GamsGo AI

AI 工具推荐

想把多个 AI 模型放在一个入口?

GamsGo AI 集成 ChatGPT、DeepSeek、Gemini、Claude、Midjourney、Veo 等常用模型,适合写作、绘图、视频和日常 AI 工作流。

了解 GamsGo AI

推广链接:通过此链接购买,我可能获得佣金,不影响你的价格。

值得关注的后续

一是完整论文和实验细节是否公开,尤其是 1.10 至 1.42 倍这一区间在不同序列长度和批量下的稳定性;二是 FLAMIA 是否会以开源内核或框架集成的方式提供给开发者,进而被 Megatron、PyTorch 等训练栈采用;三是 NVIDIA 官方低精度注意力方案和竞品是否有对应跟进,以及混合 NVFP4/FP8 路径的实际训练质量在生产规模下能否保持。

来源:Hugging Face Papers

celebrityanime
celebrityanime
文章: 32182

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注