AdaGrad

Author: Daniel Villarraga (SYSEN 6800 Fall 2021)

Introduction

AdaGrad is a family of sub-gradient algorithms for stochastic optimization. The algorithms belonging to that family are similar to second-order stochastic gradient descend with an approximation for the Hessian of the optimized function. AdaGrad's name comes from Adaptative Gradient. Intuitively, it adapts the learning rate for each feature depending on the estimated geometry of the problem; particularly, it tends to assign higher learning rates to infrequent features, which ensures that the parameter updates rely less on frequency and more on relevance.

AdaGrad was introduced by Duchi et al. in a highly cited paper published in the Journal of machine learning research in 2011. It is arguably one of the most popular algorithms for machine learning (particularly for training deep neural networks) and it influenced the development of the Adam algorithm.

Theory

The objective of AdaGrad is to minimize the expected value of a stochastic objective function, with respect to a set of parameters, given a sequence of realizations of the function. As with other sub-gradient-based methods, it achieves so by updating the parameters in the opposite direction of the sub-gradients. While standard sub-gradient methods use update rules with step-sizes that ignore the information from the past observations, AdaGrad adapts the learning rate for each parameter individually using the sequence of gradient estimates.

Definitions

$f(x)$: Stochastic objective function with parameters $x$.

$f_{t}(x)$: Realization of stochastic objective at time step $t$. For simplicity $f_{t}$.

$g_{t}(x)$: The gradient of $f_{t}(x)$ with respect to $x$, formally $\nabla _{x}f_{t}(x)$. For simplicity, $g_{t}$.

$x_{t}$: Parameters at time step $t$.

$G_{t}$: The outer product of all previous subgradients, given by ${\textstyle \sum _{\tau =1}^{t}g_{\tau }g_{\tau }^{\top }}$

Standard Sub-gradient Update

Standard sub-gradient algorithms update parameters $x$ according to the following rule:

$$x_{t+1}=x_{t}-\eta g_{t}$$

where $\eta$ denotes the step-size often refered as learning rate or step-size. Expanding each term on the previous equation, the vector of parameters is updated as follows:

$${\begin{bmatrix}x_{t+1}^{(2)}\\x_{t+1}^{(2)}\\\vdots \\x_{t+1}^{(m)}\end{bmatrix}}={\begin{bmatrix}x_{t}^{(2)}\\x_{t}^{(2)}\\\vdots \\x_{t}^{(m)}\end{bmatrix}}-\eta {\begin{bmatrix}g_{t}^{(2)}\\g_{t}^{(2)}\\\vdots \\g_{t}^{(m)}\end{bmatrix}}$$

AdaGrad Update

The general AdaGrad update rule is given by:

$$x_{t+1}=x_{t}-\eta G_{t}^{-1/2}g_{t}$$

where $G_{t}^{-1/2}$ is the inverse of the square root of $G_{t}$. A simplified version of the update rule takes the diagonal elements of $G_{t}$ instead of the whole matrix:

$$x_{t+1}=x_{t}-\eta {\text{diag}}(G_{t})^{-1/2}g_{t}$$

which can be computed in linear time. In practice, a small quantity $\epsilon$ is added to each diagonal element in $G_{t}$ to avoid singularity problems, the resulting update rule is given by:

$$x_{t+1}=x_{t}-\eta {\text{diag}}(\epsilon I+G_{t})^{-1/2}g_{t}$$

where $I$ denotes the identity matrix. An expanded form of the previous update is presented below,

$${\begin{bmatrix}x_{t+1}^{(2)}\\x_{t+1}^{(2)}\\\vdots \\x_{t+1}^{(m)}\end{bmatrix}}={\begin{bmatrix}x_{t}^{(2)}\\x_{t}^{(2)}\\\vdots \\x_{t}^{(m)}\end{bmatrix}}-{\begin{bmatrix}\eta {\frac {1}{\sqrt {\epsilon +G_{t}^{(1,1)}}}}\\\eta {\frac {1}{\sqrt {\epsilon +G_{t}^{(2,2)}}}}\\\vdots \\\eta {\frac {1}{\sqrt {\epsilon +G_{t}^{(m,m)}}}}\end{bmatrix}}\odot {\begin{bmatrix}g_{t}^{(2)}\\g_{t}^{(2)}\\\vdots \\g_{t}^{(m)}\end{bmatrix}}$$

where the operator $\odot$ denotes the Hadamard product between matrices of the same dimension, and $G_{t}^{(j,j)}$ is the $j$ element in the $G_{t}$ diagonal. From the last expression, it is clear that the update rule for AdaGrad adapts the step-size for each parameter $j$ accoding to ${\textstyle \eta (\epsilon +G_{t}^{(j,j)})^{-1/2}}$, while standard sub-gradient methods have fixed step-size $\eta$ for every parameter.

Adaptative Learning Rate Effect

An estimate for the uncentered second moment of the objective function's gradient is given by the following expression:

$v={\frac {1}{t}}\sum _{\tau =1}^{t}g_{t}g_{t}^{\top }$

which is similar to the definition of matrix $G_{t}$, used in AdaGrad's update rule. Noting that, AdaGrad adapts the learning rate for each parameter proportionally to the inverse of the gradient's variance for every parameter. This leads to the main advantages of AdaGrad:

1. Parameters associated with low frequency features tend to have larger learning rates than parameters associated with high frequency features.
2. Step-sizes in directions with high gradient variance are lower than the step-sizes in directions with low gradient variance. Geometrically, the step-sizes tend to decrease proportionally to the curvature of the stochastic objective function.

which favor the convergence rate of the algorithm.

Algorithm

The general version of the AdaGrad algorithm is presented in the pseudocode below. The update step within the for loop can be modified with the version that uses the diagonal of $G_{t}$.

Regret Bound

The regret is defined as:

$R(t)=\sum _{t=1}^{T}f_{t}(x_{t})-\sum _{t=1}^{t}f_{t}(x^{*})$

where $x^{*}$is the set of optimal parameters. AdaGrad has a regret bound of order $O({\sqrt {T}})$, which leads to the convergence rate of order $O(1/{\sqrt {T}})$, and the convergence guarantee (${\textstyle \lim _{T\to \infty }R(T)/T=0}$). The detailed proof and assumptions for this bound can be found in the original journal paper.

Numerical Example

Figure 1. This figure shows a trajectory of parameter updates for the example presented in this section using AdaGrad.

To illustrate how the parameter updates work in AdaGrad take the following numerical example. The dataset consists of random generated obsevations $x,y$ that follow the linear relationship:

$$y=20\cdot x+\epsilon$$

where $\epsilon$ is random noise. The first 5 obsevations are shown in the table below.

$x$	$y$
0.39	9.83
0.10	2.27
0.30	5.10
0.35	6.32
0.85	15.50

The cost function is defined as ${\textstyle f(a,b)={\frac {1}{N}}\sum _{i=1}^{n}([a+b\cdot x_{i}]-y_{i})^{2}}$. And an observation of the cost function at time step $t$ is given by $f_{t}(a,b)=([a+b\cdot x_{t}]-y_{t})^{2}$, where $x_{t},y_{t}$ are sampled from the obsevations. Finally, the subgradient is determined by:

$$g_{t}={\begin{bmatrix}2([a_{t}+b_{t}\cdot x_{t}]-y_{t})\\2x_{t}([a_{t}+b_{t}\cdot x_{t}]-y_{t})\end{bmatrix}}$$

Take a learning rate $\eta =5$, initial parameters $a_{1},b_{1}=0$, and $x_{1}=0.39,y_{1}=9.84$. For the first iteration of AdaGrad the subgradient is equal to:

$$g_{1}={\begin{bmatrix}2([0+0\cdot 0.39]-9.84)\\2\cdot 0.39([0+0\cdot 0.39]-9.84)\end{bmatrix}}={\begin{bmatrix}-19.68\\-7.68\end{bmatrix}}$$

and $G_{t}$ is:

$$G_{1}=\sum _{\tau =1}^{1}g_{1}g_{1}^{\top }={\begin{bmatrix}-19.68\\-7.68\end{bmatrix}}{\begin{bmatrix}-19.68&-7.68\end{bmatrix}}={\begin{bmatrix}387.30&151.14\\151.14&58.98\end{bmatrix}}$$

So the first parameter's update is calculated as follows:

$${\begin{bmatrix}a_{2}\\b_{2}\end{bmatrix}}={\begin{bmatrix}0\\0\end{bmatrix}}-{\begin{bmatrix}5\cdot {\frac {1}{\sqrt {387.30}}}\\5\cdot {\frac {1}{\sqrt {58.98}}}\end{bmatrix}}\odot {\begin{bmatrix}-19.68\\-7.68\end{bmatrix}}={\begin{bmatrix}5\\5\end{bmatrix}}$$

This process is repetated until convergence or for a fixed number of iterations $T$. An example AdaGrad update trajectory for this example is presented in Figure 1. Note that in Figure 1, the algorithm converges to a region close to $(a,b)=(0,20)$, which are the set of parameters that originated the obsevations in this example.

Applications

The AdaGrad family of algorithms is typically used in machine learning applications. Mainly, it is a good choice for deep learning models with sparse input features. However, it can be applied to any optimization problem with a differentiable cost function. Given its popularity and proven performance, different versions of AdaGrad are implemented in the leading deep learning frameworks like TensorFlow and PyTorch. Nevertheless, in practice, AdaGrad tends to be substituted by the use of the Adam algorithm; since, for a given choice of hyperparameters, Adam is equivalent to AdaGrad .  

Conclusion

AdaGrad is a family of algorithms for stochastic optimization that uses a Hessian approximation of the cost function for the update rule. It uses that information to adapt different learning rates for the parameters associated with each feature. Their main two advantages are that: it assigns more significant learning rates to parameters related to low-frequency features, and the updates in directions of high curvature tend to be lower than those in low-curvature directions. Therefore, it works especially well for problems where the input features are sparse. AdaGrad is reasonably popular in the machine learning community, and it is implemented in the primary deep learning frameworks. In practice, the Adam algorithm is usually preferred over AdaGrad since, for a given choice of hyperparameters, Adam is equivalent to AdaGrad.