微软推ZeRO++新系统:减少大模型训练时间和成本


(资料图)

编程客栈()6月26日 消息:微软研究人员推出了名为ZeRO++的新系统,用于优化android训练大型AI模型,以应对高数据传输开销和有限带宽的难题。

ZeRO++ 建立在现有的 ZeRO 优化的基础上,并提供增强的通信策略,以提高训练效率,减少训练时间和成本。

像Turing-NLG、ChatGPT和GPT-4这样的大型模型的训练需android要跨多个GPU设备占用大量内存和计算资源。DeepSpeed开发的ZeRO++引入了通信优化策略,以克服在每个GPU上批量大小较小时或在低带宽集群上进行训练时ZeRO的限制。

ZeRO优化系列包括ZeRO-Inference,编程客栈它使用GPU的集体内存和计算能力,将模型状态分割在多个GPU之间。然而,在训练过程中,ZeRO可能会产生较高的通信开销。ZeRO++通过整合三组通信优化来解决这个问题:量化权重通信(qwZ)、分层权重分割(hpZ)和量化梯度通信(qgZ)。

为了减少参数通信量,ZeRO++对权重进行量化,利用基于块的量化方法来保持训练精度。这种优化的量化过程比基本量化更快更准确。为了在反向传播过程中尽量减少通信开销,ZeRO++通过在每台机器上保持完整的模型副本,以 GPU 内存换取通信。梯度通信方面,ZeRO++引入了一种新的量化梯度通信范式 qgZ,可以减少跨节点的流量和延迟。

这些通信优化大大减少了通信量。与ZeRO相比,ZeRO++实现了减少高达4倍的通信量,提高了训练吞吐量和效率。当在每个GPU上使用小批量大小时,在高带宽集群中,ZeRO++相比ZeRO-3的吞吐量编程客栈提高了28%至3javascript6%。在低带宽集群中,与ZeRO-3相比,ZeRO++实现了平均2倍的加速,使得大模型训练在更多种类的集群上更为可行。

ZeRO++不仅限于训练场景,还可应用于对话模型中使用的人类反馈强化学习(RLHF)训练。通过将ZeRO++与DeepSpeed-Chat集成,RLHF训练可以获得更好的生成和训练吞吐量,生成吞吐量提高了高达2.25倍,训练吞吐量提高了1.26倍,超过了ZeRO。

DeepSpeed已经发布了ZeRO++,以使大型模型的训练在AI社区更加高效和可访问。该系统旨在加快训练速度,减少通信开销,并实现更大的批量大小,从而节省时间和资源。研究人员和实践者可以利用ZeRO++更有效地训练像ChatGPT这样的模型,并在AI领域探索新的可能性。

关键词:

郑州3个区仓储供不应求 仓储超2000万平方

仓储是现代物流建设中不可缺少的一个重要环节。为了鼓励仓储行业发展,今年7月底,河南出台的17条政策支持现代物流强省建设,第一条就提到

网易新闻 2022-09-16

美诺华2022年中报 上半年实现营收8.97亿元 同比增长29.74%

2022年8月15日,A股上市公司美诺华(代码:603538 SH)发布2022年半年度业绩报告。2022年1月1日-2022年6月30日,公司实现营业收入8 97亿元,

资本邦 2022-08-17

各行业工资单出炉!IT类最赚钱,还有这些钱景喜人

  中新经纬11月24日电 (张澍楠)虽说三百六十行,行行出状元,但行业之间的差距,仍然很大。究竟什么行业“最香”?被视为“高富帅”的金

2021-11-24

“狗咬人”事件当事人被撤职 多名干部被问责

  新华社郑州11月23日电(记者冯大鹏)在“狗咬人”舆情发酵后,23日晚,河南安阳通报了对涉“狗咬人”事件责任单位和责任人的处理决定。 

2021-11-24

北京朝阳区来广营华贸城7号院6号楼解除管控

  11月23日晚,朝阳区来广营地区清苑路第五社区华贸城7号院6号楼正式解除管控。  11月23日,华贸城7号院6号楼583户管控居民进行了第四

2021-11-24

大连市将4个中风险地区调整为低风险地区

  11月23日大连市新冠肺炎疫情防控总指挥部发布,大连市严格落实新冠肺炎疫情防控各项措施,至2021年11月23日24时,大连市庄河市城关街道

2021-11-24

云南哀牢山4名遇难地质队员遗体已移交其所在单位

  根据云南省普洱市哀牢山 "11·15 "联合指挥部通报,2021年11月23日21时50分,4名遇难人员遗体已移交其所在单位。 【编辑:叶攀】

2021-11-24

郑州3个区仓储供不应求 仓储超2000万平方

仓储是现代物流建设中不可缺少的一个重要环节。为了鼓励仓储行业发展,今年7月底,河南出台的17条政策支持现代物流强省建设,第一条就提到

网易新闻 2022-09-16

美诺华2022年中报 上半年实现营收8.97亿元 同比增长29.74%

2022年8月15日,A股上市公司美诺华(代码:603538 SH)发布2022年半年度业绩报告。2022年1月1日-2022年6月30日,公司实现营业收入8 97亿元,

资本邦 2022-08-17

各行业工资单出炉!IT类最赚钱,还有这些钱景喜人

  中新经纬11月24日电 (张澍楠)虽说三百六十行,行行出状元,但行业之间的差距,仍然很大。究竟什么行业“最香”?被视为“高富帅”的金

2021-11-24

第三届拉萨市旅游行业服务技能大赛决赛举行

  中新网拉萨11月23日电 (记者 冉文娟)第三届拉萨市旅游行业服务技能大赛决赛11月23日精彩举行。百余名选手经过层层选拔,经历初赛、网

2021-11-24

聚焦解决人兽冲突 东北虎豹国家公园启动专项行动

  中新网长春11月23日电 (郭佳 吴林锡)东北虎豹国家公园23日全面启动2021-2022年今冬明春清山清套·打击乱捕滥猎专项行动。该行动旨在

2021-11-24

呼吸治疗专家浦其斌逝世 曾多次参与公共卫生事件救治

2021-11-24

穗港澳职工交流基地挂牌 将带动更多穗港澳青年建设湾区

  中新网广州11月23日电 (蔡敏婕 罗瑞雄)“穗港澳职工交流服务基地”23日在广州挂牌。广州市总工会主席唐航浩称,设立穗港澳职工交流服

2021-11-24
x 广告
x 广告
x 广告

Copyright   2015-2022 中公仓储网版权所有  备案号:沪ICP备2022005074号-18   联系邮箱:5855973@qq.com