字节跳动开源大模型训练框架veGiantModel ,性能最高提升6.9倍

新闻 自然语言处理
自然语言处理是人工智能研究的一个重要领域,旨在帮助计算机理解、解释和运用人类语言。

近日,字节跳动应用机器学习团队开源了一款名为 veGiantModel 的大模型训练框架,该训练框架主要应用于自然语言处理领域的大模型训练,最高可将大模型训练的性能提升6.9倍,大幅降低训练系统的压力。目前,字节跳动旗下的企业级技术服务平台火山引擎已在其机器学习平台上原生支持了 veGiantModel,该平台正在公测中 。

自然语言处理是人工智能研究的一个重要领域,旨在帮助计算机理解、解释和运用人类语言。近些年,自然语言处理在应用方面取得了较为显著的突破,主要是归功于 Bert、GPT、GPT-3 等预训练语言模型的普及。可以说,预训练语言模型已经成为人工智能领域的基础设施。由于大模型的算法表现更加出众,预训练语言模型在近几年呈现出迅速向大模型发展的趋势。然而,模型体积的快速增长也对现有的训练系统带来了不小的挑战,主要体现为显存压力、计算压力和通信压力。

针对现有训练系统在大模型训练场景下的上述挑战, 字节跳动应用机器学习团队提出了大模型训练框架 veGiantModel。

veGiantModel的中文名称叫做火山引擎大模型训练框架,是基于开源深度学习框架 PyTorch 、并在 Megatron 和 DeepSpeed 两大开源主流训练框架的基础之上构建的高性能大模型训练框架。

veGiantModel 可同时支持数据并行、算子切分、流水线并行 3 种分布式并行策略,并支持自动化和定制化的并行策略;基于字节自研的高性能异步通讯库 ByteCCL,veGiantModel 的训练任务吞吐相比其他主流开源框架有 1.2 倍到 3.5倍的提升,并且提供了更友好、灵活的流水线支持,降低了模型开发迭代所需要的人力;另外,veGiantModel 可在 GPU 上高效地支持数十亿至上千亿参数量的大模型,对网络带宽要求也更低,在私有化部署时无 RDMA 强依赖。

与 Megtraon 和 DeepSpeed 的对比测试显示,veGiantModel 的性能表现最好、受网络带宽影响最小,在Tesla V100上较上述两者有1.2 倍到3.5倍的提升,在 Ampere A100上最高可提升6.9倍。


veGiantModel开源 地址:

​https://github.com/volcengine/veGiantModel​​​

火山引擎机器学习平台 公测地址:

​https://www.volcengine.com/product/ml-platform​


责任编辑:黄显东 来源: 51CTO
相关推荐

2022-03-21 15:06:10

模型字节跳动框架

2022-04-26 15:09:14

优化模型训练

2021-09-17 13:05:55

模型人工智能框架

2022-07-18 17:37:27

字节跳动人工智能AI模型

2022-10-31 15:35:16

开源引擎

2023-04-19 16:51:54

分布式Primus开源

2022-06-22 06:49:39

Hertz开源HTTP 框架

2023-04-14 14:42:07

腾讯人工智能

2022-05-11 14:45:48

模型人工智能

2022-08-25 18:48:29

字节跳动CSS开源

2023-09-12 13:43:00

智能技术

2023-10-18 11:56:17

开源AI

2023-04-07 12:30:04

开源ShmipcIPC

2021-12-29 11:06:25

Java代码技巧

2023-01-03 16:54:27

字节跳动深度学习

2020-01-18 16:09:31

AI 数据人工智能

2023-02-17 10:32:20

2022-05-09 09:30:06

框架开源模型
点赞
收藏

51CTO技术栈公众号