当前位置:网站首页 > 论文专利 >专利
【论文专利】

专利名称:Adam算法加速系统及方法

专利国别:中国

专利号:202210601059.0

法律状态:实审

发明人:张春,张惟宜,牛丽婷,姜汉钧

申请人:清华大学

地址:100084 北京市海淀区100084信箱82分箱清华大学专利办公室

申请日期:2022-05-30

授权日期:

摘要:

本发明公开了一种Adam算法加速系统及方法,该系统包括:一阶矩估计子系统,用于根据上轮一阶矩与本轮梯度,计算本轮一阶矩;二阶矩平方根估计子系统,用于根据上轮二阶矩平方根与本轮梯度,计算本轮二阶矩平方根;更新步计算子系统包括:平方根倒数计算模块,用于采用快速平方根倒数算法,根据本轮二阶矩平方根,计算平方根倒数;第一乘法器,用于对本轮一阶矩与平方根倒数进行乘法计算;第二乘法器,用于将第一乘法器的输出与平方根倒数进行乘法计算,获得本轮更新步,所述本轮更新步用于在硬件中训练神经网络,更新网络参数。本发明可以实现在硬件中训练神经网络时进行Adam算法加速,提高吞吐率。

专利证书 PDF/Jpg