Adam优化器简单理解和实现

时间:2024-04-10 21:42:47

前言:因为要实现GAN,简单理解一下训练过程需要用到的Adam优化器。

一、简单认识

Adam(自适应矩估计 Adaptive moment estimation)

Adam优化器简单理解和实现

翻译:

参数设定:Adam优化器简单理解和实现 (元素乘操作),Adam优化器简单理解和实现

Adam优化器简单理解和实现:步长或称学习率(stepsize);Adam优化器简单理解和实现:矩估计的指数衰减率;Adam优化器简单理解和实现:参数Adam优化器简单理解和实现的随机目标函数值;Adam优化器简单理解和实现:初始参数向量;

Adam优化器简单理解和实现:初始第一矩向量;Adam优化器简单理解和实现:第二矩向量;Adam优化器简单理解和实现:时间步长;

优化过程(更新参数向量过程):

while Adam优化器简单理解和实现 未收敛:  {

Adam优化器简单理解和实现

Adam优化器简单理解和实现  获得新一轮的梯度值(参数是上轮刚更新的)

Adam优化器简单理解和实现  更新第一矩向量

Adam优化器简单理解和实现   更新第二矩向量

Adam优化器简单理解和实现   计算偏差校正的第一矩向量

Adam优化器简单理解和实现   计算偏差校正的第二矩向量

Adam优化器简单理解和实现   更新参数

}

优化参数过程结束

 

二、代码实现

https://github.com/dream-catcher/learning_blogs/blob/master/Adam_Optimizer/adam_optimizer.py

 

参考链接:https://www.jianshu.com/p/aebcaf8af76e