Neural Network Optimization Methods and Algorithms

8 minute read

For the seemingly small project I undertook of creating a machine learning neural network that could learn by itself to play tic-tac-toe, I bumped into the necesity of implementing at least one momentum algorithm for the optimization of the network during backpropagation.

And since my original post for the TicTacToe project is quite large already, I decided to post separately these optimization methods and how did I implement them in my code.

Adam

source

Adaptive Moment Estimation (Adam) is an optimization method that computes adaptive learning rates for each weight and bias. In addition to storing an exponentially decaying average of past squared gradients \(v_t\) and an exponentially decaying average of past gradients \(m_t\), similar to momentum. Whereas momentum can be seen as a ball running down a slope, Adam behaves like a heavy ball with friction, which thus prefers flat minima in the error surface. We compute the decaying averages of past and past squared gradients \(m_t\) and \(v_t\) respectively as follows:

\(
\begin{align}
\begin{split}
m_t &= \beta_1 m_{t-1} + (1 - \beta_1) g_t \\
v_t &= \beta_2 v_{t-1} + (1 - \beta_2) g_t^2
\end{split}
\end{align}
\)

\(m_t\) and \(v_t\) are estimates of the first moment (the mean) and the second moment (the uncentered variance) of the gradients respectively, hence the name of the method. As \(m_t\) and \(v_t\) are initialized as vectors of 0's, the authors of Adam observe that they are biased towards zero, especially during the initial time steps, and especially when the decay rates are small (i.e. \(\beta_1\) and \(\beta_2\) are close to 1).

They counteract these biases by computing bias-corrected first and second moment estimates:

\(
\begin{align}
\begin{split}
\hat{m}_t &= \dfrac{m_t}{1 - \beta^t_1} \\
\hat{v}_t &= \dfrac{v_t}{1 - \beta^t_2} \end{split}
\end{align}
\)

We then use these to update the weights and biases which yields the Adam update rule:

\(\theta_{t+1} = \theta_{t} - \dfrac{\eta}{\sqrt{\hat{v}_t} + \epsilon} \hat{m}_t\).

The authors propose defaults of 0.9 for \(\beta_1\), 0.999 for \(\beta_2\), and \(10^{-8}\) for \(\epsilon\).

view on github

# decaying averages of past gradients
self.v["dW" + str(i)] = ((c.BETA1
                        * self.v["dW" + str(i)])
                        + ((1 - c.BETA1)
                        * np.array(self.gradients[i])
                        ))
self.v["db" + str(i)] = ((c.BETA1
                        * self.v["db" + str(i)])
                        + ((1 - c.BETA1)
                        * np.array(self.bias_gradients[i])
                        ))

# decaying averages of past squared gradients
self.s["dW" + str(i)] = ((c.BETA2
                        * self.s["dW"+str(i)])
                        + ((1 - c.BETA2)
                        * (np.square(np.array(self.gradients[i])))
                         ))
self.s["db" + str(i)] = ((c.BETA2
                        * self.s["db" + str(i)])
                        + ((1 - c.BETA2)
                        * (np.square(np.array(
                                         self.bias_gradients[i])))
                         ))

if c.ADAM_BIAS_Correction:
    # bias-corrected first and second moment estimates
    self.v["dW" + str(i)] = self.v["dW" + str(i)]
                          / (1 - (c.BETA1 ** true_epoch))
    self.v["db" + str(i)] = self.v["db" + str(i)]
                          / (1 - (c.BETA1 ** true_epoch))
    self.s["dW" + str(i)] = self.s["dW" + str(i)]
                          / (1 - (c.BETA2 ** true_epoch))
    self.s["db" + str(i)] = self.s["db" + str(i)]
                          / (1 - (c.BETA2 ** true_epoch))

# apply to weights and biases
weight_col -= ((eta * (self.v["dW" + str(i)]
                      / (np.sqrt(self.s["dW" + str(i)])
                      + c.EPSILON))))
self.bias[i] -= ((eta * (self.v["db" + str(i)]
                        / (np.sqrt(self.s["db" + str(i)])
                        + c.EPSILON))))

SGD Momentum

source

Vanilla SGD has trouble navigating ravines, i.e. areas where the surface curves much more steeply in one dimension than in another, which are common around local optima. In these scenarios, SGD oscillates across the slopes of the ravine while only making hesitant progress along the bottom towards the local optimum.

Momentum is a method that helps accelerate SGD in the relevant direction and dampens oscillations. It does this by adding a fraction \(\gamma\) of the update vector of the past time step to the current update vector:

\(
\begin{align}
\begin{split}
v_t &= \beta_1 v_{t-1} + \eta \nabla_\theta J( \theta) \\
\theta &= \theta - v_t
\end{split}
\end{align}
\)

The momentum term \(\beta_1\) is usually set to 0.9 or a similar value.

Essentially, when using momentum, we push a ball down a hill. The ball accumulates momentum as it rolls downhill, becoming faster and faster on the way (until it reaches its terminal velocity if there is air resistance, i.e. \(\beta_1 < 1\)). The same thing happens to our weight and biases updates: The momentum term increases for dimensions whose gradients point in the same directions and reduces updates for dimensions whose gradients change directions. As a result, we gain faster convergence and reduced oscillation.

view on github

self.v["dW"+str(i)] = ((c.BETA1*self.v["dW" + str(i)])
                       +(eta*np.array(self.gradients[i])
                       ))
self.v["db"+str(i)] = ((c.BETA1*self.v["db" + str(i)])
                       +(eta*np.array(self.bias_gradients[i])
                       ))

weight_col -= self.v["dW" + str(i)]
self.bias[i] -= self.v["db" + str(i)]

Nesterov accelerated gradient (NAG)

source

However, a ball that rolls down a hill, blindly following the slope, is highly unsatisfactory. We'd like to have a smarter ball, a ball that has a notion of where it is going so that it knows to slow down before the hill slopes up again.

Nesterov accelerated gradient (NAG) is a way to give our momentum term this kind of prescience. We know that we will use our momentum term \(\beta_1 v_{t-1}\) to move the weights and biases \(\theta\). Computing \( \theta - \beta_1 v_{t-1} \) thus gives us an approximation of the next position of the weights and biases (the gradient is missing for the full update), a rough idea where our weights and biases are going to be. We can now effectively look ahead by calculating the gradient not w.r.t. to our current weights and biases \(\theta\) but w.r.t. the approximate future position of our weights and biases:

\(
\begin{align}
\begin{split}
v_t &= \beta_1 v_{t-1} + \eta \nabla_\theta J( \theta - \beta_1 v_{t-1} ) \\
\theta &= \theta - v_t
\end{split}
\end{align}
\)

Again, we set the momentum term \(\beta_1\) to a value of around 0.9. While Momentum first computes the current gradient and then takes a big jump in the direction of the updated accumulated gradient, NAG first makes a big jump in the direction of the previous accumulated gradient, measures the gradient and then makes a correction, which results in the complete NAG update. This anticipatory update prevents us from going too fast and results in increased responsiveness, which has significantly increased the performance of Neural Networks on a number of tasks.

Now that we are able to adapt our updates to the slope of our error function and speed up SGD in turn, we would also like to adapt our updates to each individual weight and bias to perform larger or smaller updates depending on their importance.

view on github

v_prev = {"dW" + str(i): self.v["dW" + str(i)],
          "db" + str(i): self.v["db" + str(i)]}

self.v["dW" + str(i)] =
            (c.NAG_COEFF * self.v["dW" + str(i)]
           - eta * np.array(self.gradients[i]))
self.v["db" + str(i)] =
            (c.NAG_COEFF * self.v["db" + str(i)]
           - eta * np.array(self.bias_gradients[i]))

weight_col += ((-1 * c.BETA1 * v_prev["dW" + str(i)])
               + (1 + c.BETA1) * self.v["dW" + str(i)])
self.bias[i] += ((-1 * c.BETA1 * v_prev["db" + str(i)])
               + (1 + c.BETA1) * self.v["db" + str(i)])

RMSprop

source

RMSprop is an unpublished, adaptive learning rate method proposed by Geoff Hinton in Lecture 6e of his Coursera Class.

RMSprop was developed stemming from the need to resolve other method's radically diminishing learning rates.

\(
\begin{align}
\begin{split}
E[\theta^2]_t &= \beta_1 E[\theta^2]_{t-1} + (1-\beta_1) \theta^2_t \\
\theta_{t+1} &= \theta_{t} - \dfrac{\eta}{\sqrt{E[\theta^2]_t + \epsilon}} \theta_{t}
\end{split}
\end{align}
\)

RMSprop divides the learning rate by an exponentially decaying average of squared gradients. Hinton suggests \(\beta_1\) to be set to 0.9, while a good default value for the learning rate \(\eta\) is 0.001.

view on github

self.s["dW" + str(i)] = ((c.BETA1
                      * self.s["dW" + str(i)])
                      + ((1-c.BETA1)
                      * (np.square(np.array(self.gradients[i])))
                        ))
self.s["db" + str(i)] = ((c.BETA1
                      * self.s["db" + str(i)])
                      + ((1-c.BETA1)
                      * (np.square(np.array(self.bias_gradients[i])))
                        ))

weight_col -= (eta * (np.array(self.gradients[i])
              / (np.sqrt(self.s["dW"+str(i)]+c.EPSILON)))
              )
self.bias[i] -= (eta * (np.array(self.bias_gradients[i])
               / (np.sqrt(self.s["db"+str(i)]+c.EPSILON)))
                )

Complete code

All in all the code ended up like this: view on github

@staticmethod
def cyclic_learning_rate(learning_rate, epoch):
    max_lr = learning_rate * c.MAX_LR_FACTOR
    cycle = np.floor(1 + (epoch / (2
                    * c.LR_STEP_SIZE))
                    )
    x = np.abs((epoch / c.LR_STEP_SIZE)
        - (2 * cycle) + 1)
    return learning_rate
        + (max_lr - learning_rate)
        * np.maximum(0, (1 - x))

def apply_gradients(self, epoch):
    true_epoch = epoch - c.BATCH_SIZE
    eta = self.learning_rate
            * (1 / (1 + c.DECAY_RATE * true_epoch))

    if c.CLR_ON:
        eta = self.cyclic_learning_rate(eta, true_epoch)

    for i, weight_col in enumerate(self.weights):

        if c.OPTIMIZATION == 'vanilla':
            weight_col -= eta
                        * np.array(self.gradients[i])
                        / c.BATCH_SIZE
            self.bias[i] -= eta
                        * np.array(self.bias_gradients[i])
                        / c.BATCH_SIZE

        elif c.OPTIMIZATION == 'SGD_momentum':
            self.v["dW"+str(i)] = ((c.BETA1
                                   *self.v["dW" + str(i)])
                                   +(eta
                                   *np.array(self.gradients[i])
                                   ))
            self.v["db"+str(i)] = ((c.BETA1
                                   *self.v["db" + str(i)])
                                   +(eta
                                   *np.array(self.bias_gradients[i])
                                   ))

            weight_col -= self.v["dW" + str(i)]
            self.bias[i] -= self.v["db" + str(i)]

        elif c.OPTIMIZATION == 'NAG':
            v_prev = {"dW" + str(i): self.v["dW" + str(i)],
                      "db" + str(i): self.v["db" + str(i)]}

            self.v["dW" + str(i)] =
                        (c.NAG_COEFF * self.v["dW" + str(i)]
                       - eta * np.array(self.gradients[i]))
            self.v["db" + str(i)] =
                        (c.NAG_COEFF * self.v["db" + str(i)]
                       - eta * np.array(self.bias_gradients[i]))

            weight_col += ((-1 * c.BETA1 * v_prev["dW" + str(i)])
                           + (1 + c.BETA1) * self.v["dW" + str(i)])
            self.bias[i] += ((-1 * c.BETA1 * v_prev["db" + str(i)])
                           + (1 + c.BETA1) * self.v["db" + str(i)])

        elif c.OPTIMIZATION == 'RMSProp':
            self.s["dW" + str(i)] =
                            ((c.BETA1
                            *self.s["dW" + str(i)])
                            +((1-c.BETA1)
                            *(np.square(np.array(self.gradients[i])))
                            ))
            self.s["db" + str(i)] =
                            ((c.BETA1
                            *self.s["db" + str(i)])
                            +((1-c.BETA1)
                            *(np.square(np.array(self.bias_gradients[i])))
                            ))

            weight_col -= (eta
                          *(np.array(self.gradients[i])
                          /(np.sqrt(self.s["dW"+str(i)]+c.EPSILON)))
                          )
            self.bias[i] -= (eta
                          *(np.array(self.bias_gradients[i])
                          /(np.sqrt(self.s["db"+str(i)]+c.EPSILON)))
                            )

        if c.OPTIMIZATION == "ADAM":
            # decaying averages of past gradients
            self.v["dW" + str(i)] = ((
                                c.BETA1
                              * self.v["dW" + str(i)])
                              + ((1 - c.BETA1)
                              * np.array(self.gradients[i])
                                    ))
            self.v["db" + str(i)] = ((
                                c.BETA1
                              * self.v["db" + str(i)])
                              + ((1 - c.BETA1)
                              * np.array(self.bias_gradients[i])
                                    ))

            # decaying averages of past squared gradients
            self.s["dW" + str(i)] = ((c.BETA2
                                    * self.s["dW"+str(i)])
                                    + ((1 - c.BETA2)
                                    * (np.square(
                                            np.array(
                                                self.gradients[i])))
                                     ))
            self.s["db" + str(i)] = ((c.BETA2
                                    * self.s["db" + str(i)])
                                    + ((1 - c.BETA2)
                                    * (np.square(
                                            np.array(
                                                self.bias_gradients[i])))
                                     ))

            if c.ADAM_BIAS_Correction:
                # bias-corrected first and second moment estimates
                self.v["dW" + str(i)] =
                                self.v["dW" + str(i)]
                              / (1 - (c.BETA1 ** true_epoch))
                self.v["db" + str(i)] =
                                self.v["db" + str(i)]
                              / (1 - (c.BETA1 ** true_epoch))
                self.s["dW" + str(i)] =
                                self.s["dW" + str(i)]
                              / (1 - (c.BETA2 ** true_epoch))
                self.s["db" + str(i)] =
                                self.s["db" + str(i)]
                              / (1 - (c.BETA2 ** true_epoch))

            # apply to weights and biases
            weight_col -= ((eta
                            * (self.v["dW" + str(i)]
                            / (np.sqrt(self.s["dW" + str(i)])
                            + c.EPSILON))))
            self.bias[i] -= ((eta
                            * (self.v["db" + str(i)]
                            / (np.sqrt(self.s["db" + str(i)])
                            + c.EPSILON))))

    self.gradient_zeros()
Armando Maynez
Armando Maynez

Engineer, industry executive, research enthusiast. Avid learner with diverse interests in coding, machine learning, artificial intelligence and reinforcement learning. 17+ years of experience working in multinational corporations.

Comments

  Write a comment ...