【发布时间】:2019-01-12 03:52:50
【问题描述】:
我可能只是误解了 Adam 的工作原理,但为什么会这样:
x = tf.Variable([0.0, 0.0]) # variable
y = tf.constant([5.0, 1.0]) # target
cost = tf.abs(x-y)**2
由于y 的第一个维度大于第二个维度,因此第一个维度中的梯度大于第二个维度(应该如此),x 的每个维度都以自己的速率接近其目标值:
sgd = tf.train.GradientDescentOptimizer(0.001)
train = sgd.minimize(cost)
with tf.Session() as sess:
sess.run(tf.global_variables_initializer())
for _ in range(5):
sess.run(train)
grad,variable = sess.run(opt.compute_gradients(cost))[0]
print(grad,variable)
#[-9.98 -1.996] [0.01 0.002]
#[-9.96004 -1.992008] [0.01998 0.003996]
#[-9.94012 -1.988024] [0.02994004 0.00598801]
#[-9.920239 -1.9840479] [0.03988016 0.00797603]
#[-9.900399 -1.9800799] [0.0498004 0.00996008]
如果我们使用 Adam,为什么即使梯度具有完全不同的值,速率也基本相等?
adam = tf.train.AdamOptimizer(0.001)
train = adam.minimize(cost)
with tf.Session() as sess:
sess.run(tf.global_variables_initializer())
for _ in range(5):
sess.run(train)
grad,variable = sess.run(opt.compute_gradients(cost))[0]
print(grad,variable)
#[-9.998 -1.998] [0.001 0.001]
#[-9.996 -1.996] [0.00199999 0.00199997]
#[-9.994 -1.9940002] [0.00299997 0.00299989]
#[-9.992001 -1.9920005] [0.00399994 0.00399976]
#[-9.99 -1.990001] [0.0049999 0.00499955]
【问题讨论】:
-
我的意思是,对于 SGD,
x的第一个维度以大于第二个维度的速率接近 5.0 的目标值(比较 0.01 -> 0.01998 -> 0.2994004 -> 等等... 0.002 -> 0.003996 -> 0.00598801 等。)在亚当的情况下,这两个比率基本相等。 -
啊,我明白了。我会尝试在我的答案中加入更多信息。谢谢,我真的很喜欢这个问题 - 我之前没有这么清楚地看到亚当的这个特征(+1)。
标签: python tensorflow machine-learning optimization deep-learning