在 6 月 30 日最新发布的 MLPerf Training v2.0 榜单里，百度使用飞桨框架（PaddlePaddle）和百度智能云百舸计算平台提交的 BERT Large 模型 GPU 训练性能结果，在同等 GPU 配置下的所有提交结果里排名第一，超越了高度定制优化且长期处于榜单领先位置的 NGC PyTorch 框架，向全世界展现了飞桨框架的性能优势。
图 1 MLPerf Training v2.0 BERT 模型效能前五名训练成果。
图 1 展示了 MLPerf Training v2.0 BERT 模型在 8 卡 NVIDIA GPU A100（400W 功耗，80G 显存）下前五名的训练性能结果，百度飞桨方案比其他提交结果快 5%-11% 不等。飞桨在 BERT 模型 8 卡 GPU 训练上创造了世界最优的训练性能，这来自于飞桨框架基础性能与分布式技术的领先性，以及飞桨与 NVIDIA GPU 的深度协同优化。对于深度学习模型训练任务，从数据读取到模型计算，从底层算子到上层分布式策略，从多设备负载均衡到全流程调度机制，都会影响最终训练性能。飞桨基于领先的架构设计和长期的实践打磨，在高性能训练方面做出了系统性的优化工作，主要体现在如下几个方面：针对分布式训练经常出现的负载不均衡问题，将模型训练和数据读取、预处理分配到不同设备上进行，确保异构算力物尽其用，实现数据 IO 和计算的平衡。针对变长序列输入模型大多采用的 padding 填充对齐方式带来的冗余计算问题，提供对变长输入和对应模型结构的高效支持，让 GPU 算力资源专注于有效计算，尤其是对 Transformer 类模型计算效率提升明显。针对框架基础性能优化的极致需求，研发了高性能算子库 PHI，充分优化 GPU 内核实现，提升算子内部计算的并行度，并通过算子融合降低仿存开销，发挥 GPU 的极限性能。针对传统数据并行性能、显存瓶颈受限的问题，实现了融合数据并行、模型并行、分组参数切片并行等策略的混合并行分布式训练策略，部分场景下可实现超线性加速的分布式训练性能。针对模型训练过程各环节存在的同步频率高、时间重叠度低等问题，设计异步调度机制，保证模型收敛的同时去除大部分同步操作，实现数据处理、训练和集合通信等各环节近乎全异步调度，提升端到端极致性能。百度一直重视大模型的技术研发，并致力于推动大模型的产业落地。大模型训练需要深度学习框架在高性能分布式训练方面提供强有力支撑。飞桨分布式训练从产业实践出发，不断强化领先优势，相继发布了业界首个通用异构参数服务器架构、4D 混合并行训练策略、端到端自适应分布式训练架构等多项亮点技术，并根据不同的模型结构和稀疏稠密等特性充分打磨，可支撑包括计算机视觉、自然语言处理、个性化推荐、科学计算在内的广泛领域的不同算法在异构硬件上实现高性能训练，有效助力大模型技术创新探索的快速迭代。飞桨领先的分布式技术和高性能训练特色，支持了基于飞桨的软硬件方案在 MLPerf 上持续获得优异表现，支撑了多个业界领先的文心大模型发布，如全球首个知识增强千亿大模型「鹏城 - 百度 · 文心」，知识增强的电力行业 NLP 大模型「国网 - 百度 · 文心」，知识增强的金融行业 NLP 大模型「浦发 - 百度 · 文心」，以及国产硬件集群上 AlphaFold2 千万级别蛋白质结构分析模型。飞桨在 MLPerf Training v2.0 榜单中获得了 BERT 模型训练性能世界第一的瞩目成绩。这不仅得益于飞桨框架在性能优化领域的长期耕耘，更离不开硬件生态的助力。近年来，飞桨的技术实力深受广大硬件厂商认可，合作日趋紧密，软硬一体协同发展，生态共创硕果累累。前不久（5 月 26 日），NVIDIA 与飞桨合作推出的 NGC-Paddle 正式上线。同时在本次 MLPerf 榜单中，Graphcore 也通过使用飞桨框架取得了优异成绩。未来，飞桨将继续打造性能优势，在软硬协同性能优化和大规模分布式训练方面持续技术创新，为广大用户提供更加便捷、易用、性能优异的深度学习框架。MLPerf 是由 AI 领域世界知名的学术研究者和产业专家发起的人工智能领域基准测试标杆。MLPerf 旨在提供一个公平、实用的基准测试平台，展示业界领先的 AI 软硬件系统的最佳性能，其测试结果已获得 AI 领域的普遍认可。世界上几乎所有主流的硬件生产商和软件服务提供商都会参考 MLPerf 发布的结果构建自己的基准测试系统，以测试其开发的新的 AI 加速芯片和深度学习框架在 MLPerf 模型上的性能表现。7 月 6 日（星期三）20:00，飞桨总架构师于佃海和飞桨资深研发工程师曾锦乐将通过直播，为大家揭秘同等 GPU 配置下，百度飞桨性能「世界第一」背后的关键技术。扫描下方二维码，后台回复【学习】进行报名，直播间还有更多好礼等你来拿！
© THE END
转载请联系本公众号获得授权
投稿或寻求报道：content@jiqizhixin.com
关注公众号：拾黑（shiheibook）了解更多
[广告]赞助链接：
四季很好，只要有你，文娱排行榜：https://www.yaopaiming.com/
让资讯触达的更精准有趣：https://www.0xu.cn/
机器之心
关注网络尖刀微信公众号
