飞桨自动混合精度技术让你的训练速度飞起来

新知榜官方账号

2023-09-25 00:34:50

1. 什么是自动混合精度技术

自动混合精度是一种自动将半精度和单精度混合使用,从而加速模型训练的技术。其中单精度(FloatPrecision32,FP32)是计算机常用的一种数据类型。半精度(FloatPrecision16,FP16)是一种相对较新的浮点类型,在计算机中使用2字节(16位)存储,更适于在精度要求不高的场景中使用。在深度学习领域,使用Float16代替Float32来存储数据,可以训练更大更复杂的模型,使用更大的batchsize。同时,混合精度计算的需求应运而生,可以将训练过程中一些对精度损失不敏感且能使用TensorCore进行加速的运算使用半精度处理,最大限度的提升访存和计算效率。

2. 飞桨AMP技术的实现原理

飞桨AMP技术仅仅通过一行代码即可帮助用户将单精度训练的模型修改为自动混合精度训练。同时通过黑白名单和动态LossScaling来保证训练的稳定性,避免出现INF或者NAN问题。飞桨AMP可以充分发挥新一代NVIDIAGPU中TensorCore的计算性能优势,ResNet50、Transformer等模型的训练速度与单精度训练相比可以提升到1.5~2.9倍。AMP技术在提升访存和计算效率的同时,伴随的副作用是由于半精度数据类型的精度范围与转换前的单精度相比过窄,容易产生INF和NAN问题。为避免此类问题,AMP技术实现了自动调整LossScaling功能,即在AMP训练过程中,为避免精度下溢,每训练一定数量批次的数据,就将Loss放大指定倍数。如果Loss在放大过程中发生上溢,则可以再缩小一定倍数,确保整个训练过程中,梯度可以正常收敛。

3. AMP技术的使用方法

用户仅需要使用飞桨提供的AMP函数fluid.contrib.mixed_precision.decorate将原来的优化器SGDOptimizer进行封装,然后使用封装后的优化器更新参数梯度即可。同时,AMP技术实现了自动调整LossScaling功能,无需手动指定。在使用AMP训练过程中,系统会自动读取黑白名单,从而感知哪些算子需要被转换为半精度计算,不需要手动指定。对于某些特殊场景,如果开发者希望使用自定义的黑白名单,则可以使用AutoMixedPrecisionLists类设置。

4. AMP技术的性能优势

飞桨AMP技术在ResNet50、Transformer等模型上训练速度相对于FP32训练来说有非常大的优势,单卡加速比可达2.9倍,八卡加速比可达2.8倍。AMP技术在深度学习训练中可以大幅度提升训练速度和访存效率。

5. 多卡GPU训练的优化

在新发布的飞桨核心框架1.7版本上,AMP技术深度优化了多卡GPU训练。通过将梯度传输这个过程提到Cast操作之前,每个GPU卡在得到对应的半精度梯度后再执行Cast操作,将其转变为单精度类型,可以降低GPU多卡之间的梯度传输带宽,对减少带宽占用方面非常有效。

本页网址:https://www.xinzhibang.net/article_detail-13485.html

寻求报道,请 点击这里 微信扫码咨询

关键词

飞桨 自动混合精度技术 训练速度

分享至微信: 微信扫码阅读

相关工具

相关文章

相关快讯