在Keras中Adagrad的工作原理是什么?在Keras优化器中,self.weights是什么意思?

5
例如,Keras中Adagrad的实现如下:
class Adagrad(Optimizer):
"""Adagrad optimizer.
It is recommended to leave the parameters of this optimizer
at their default values.
# Arguments
    lr: float >= 0. Learning rate.
    epsilon: float >= 0.
    decay: float >= 0. Learning rate decay over each update.
# References
    - [Adaptive Subgradient Methods for Online Learning and Stochastic Optimization](http://www.jmlr.org/papers/volume12/duchi11a/duchi11a.pdf)
"""

def __init__(self, lr=0.01, epsilon=1e-8, decay=0., **kwargs):
    super(Adagrad, self).__init__(**kwargs)
    self.lr = K.variable(lr)
    self.epsilon = epsilon
    self.decay = K.variable(decay)
    self.initial_decay = decay
    self.iterations = K.variable(0.)

def get_updates(self, params, constraints, loss):
    grads = self.get_gradients(loss, params)
    shapes = [K.get_variable_shape(p) for p in params]
    accumulators = [K.zeros(shape) for shape in shapes]
    self.weights = accumulators
    self.updates = []

    lr = self.lr
    if self.initial_decay > 0:
        lr *= (1. / (1. + self.decay * self.iterations))
        self.updates.append(K.update_add(self.iterations, 1))

    for p, g, a in zip(params, grads, accumulators):
        new_a = a + K.square(g)  # update accumulator
        self.updates.append(K.update(a, new_a))
        new_p = p - lr * g / (K.sqrt(new_a) + self.epsilon)
        # apply constraints
        if p in constraints:
            c = constraints[p]
            new_p = c(new_p)
        self.updates.append(K.update(p, new_p))
    return self.updates

函数'get_update()'似乎是一步更新。但是累加器应该存储历史信息吗?为什么每一步都将其初始化为零?如何使其成为整个训练过程中的累加器?

这行代码是做什么用的?

self.weights = accumulators

看起来 self.weights 不再被调用了。

1个回答

3
您是正确的。在Keras中,所有优化器的get_updates()实现了一步更新的张量逻辑。每次调用此函数都会为model.fit()_make_train_function() 这里创建一个张量函数,通过将更新规则作为update= 这里传递。此更新规则在迭代到迭代时用于更新模型参数和其他参数。
优化器类的self.weights是其内部参数。它不用于训练。它只是用于保持优化器的状态(指向参数/累加器张量的指针列表),并在调用model.save时通过调用get_weights() 这里保存,在调用model.load时通过set_weights() 这里加载回来。

1
我明白了。非常感谢。 - Chris

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接