买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!
申请/专利权人:清华大学
摘要:本发明公开了一种Adam算法加速系统及方法,该系统包括:一阶矩估计子系统,用于根据上轮一阶矩与本轮梯度,计算本轮一阶矩;二阶矩平方根估计子系统,用于根据上轮二阶矩平方根与本轮梯度,计算本轮二阶矩平方根;更新步计算子系统包括:平方根倒数计算模块,用于采用快速平方根倒数算法,根据本轮二阶矩平方根,计算平方根倒数;第一乘法器,用于对本轮一阶矩与平方根倒数进行乘法计算;第二乘法器,用于将第一乘法器的输出与平方根倒数进行乘法计算,获得本轮更新步,所述本轮更新步用于在硬件中训练神经网络,更新网络参数。本发明可以实现在硬件中训练神经网络时进行Adam算法加速,提高吞吐率。
主权项:1.一种Adam算法加速系统,其特征在于,包括:一阶矩估计子系统、二阶矩平方根估计子系统和更新步计算子系统,其中,一阶矩估计子系统,用于根据上轮一阶矩与本轮梯度,计算本轮一阶矩;二阶矩平方根估计子系统,用于根据上轮二阶矩平方根与本轮梯度,计算本轮二阶矩平方根;更新步计算子系统包括:平方根倒数计算模块,用于采用快速平方根倒数算法,根据本轮二阶矩平方根,计算平方根倒数;第一乘法器,用于对本轮一阶矩与平方根倒数进行乘法计算;第二乘法器,用于将第一乘法器的输出与平方根倒数进行乘法计算,获得本轮更新步,所述本轮更新步用于在硬件中训练神经网络,更新网络参数;还包括参数更新模块,所述参数更新模块包括:超参数处理模块,用于根据一阶指数衰减因子β1和二阶指数衰减因子β2,对第一参数第二参数取近似;对第一参数和第二参数进行近似处理;采用公式确定第二右移位数,其中,y为第二右移位,lr为学习率;第三移位寄存器,用于根据第二右移位数,对第二乘法器的输出右移操作,获得缩放后的本轮更新步;网络参数更新模块,用于根据第一参数、第二参数、本轮更新步和学习率,更新网络参数。
全文数据:
权利要求:
百度查询: 清华大学 Adam算法加速系统及方法
免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。