Gradient descent: Difference between revisions

From Wikipedia, the free encyclopedia
Jump to navigation Jump to search
imported>Headbomb
Altered template type. Add: date, title, eprint, class, authors 1-2. Removed parameters. Some additions/deletions were parameter name changes. | Use this tool. Report bugs. | #UCB_Gadget
 
imported>Rambor12
 
(One intermediate revision by the same user not shown)
Line 15: Line 15:
[[File:Gradient descent.svg|thumb|350px|Illustration of gradient descent on a series of [[level set]]s]]
[[File:Gradient descent.svg|thumb|350px|Illustration of gradient descent on a series of [[level set]]s]]


Gradient descent is based on the observation that if the [[multi-variable function]] <math>F(\mathbf{x})</math> is [[Defined and undefined|defined]] and [[Differentiable function|differentiable]] in a neighborhood of a point <math>\mathbf{a}</math>, then <math>F(\mathbf{x})</math> decreases ''fastest'' if one goes from <math>\mathbf{a}</math> in the direction of the negative [[gradient]] of <math>F</math> at <math>\mathbf{a}, -\nabla F(\mathbf{a})</math>. It follows that, if
Gradient descent is based on the observation that if the [[multi-variable function]] <math>f(\mathbf{x})</math> is [[Defined and undefined|defined]] and [[Differentiable function|differentiable]] in a neighborhood of a point <math>\mathbf{a}</math>, then <math>f(\mathbf{x})</math> decreases ''fastest'' if one goes from <math>\mathbf{a}</math> in the direction of the negative [[gradient]] of <math>f</math> at <math>\mathbf{a}, -\nabla f(\mathbf{a})</math>. It follows that, if


:<math> \mathbf{a}_{n+1} = \mathbf{a}_n-\gamma\nabla F(\mathbf{a}_n)</math>
:<math> \mathbf{a}_{n+1} = \mathbf{a}_n-\eta \nabla f(\mathbf{a}_n)</math>


for a small enough step size or [[learning rate]] <math>\gamma \in \R_{+}</math>, then  <math>F(\mathbf{a_n})\geq F(\mathbf{a_{n+1}})</math>. In other words, the term <math>\gamma\nabla F(\mathbf{a})</math> is subtracted from <math>\mathbf{a}</math> because we want to move against the gradient, toward the local minimum. With this observation in mind, one starts with a guess <math>\mathbf{x}_0</math> for a local minimum of <math>F</math>, and considers the sequence <math>\mathbf{x}_0, \mathbf{x}_1, \mathbf{x}_2, \ldots</math> such that
for a small enough step size or [[learning rate]] <math>\eta\in \R_{+}</math>, then  <math>f(\mathbf{a_n})\geq f(\mathbf{a_{n+1}})</math>. In other words, the term <math>\eta\nabla f(\mathbf{a})</math> is subtracted from <math>\mathbf{a}</math> because we want to move against the gradient, toward the local minimum. With this observation in mind, one starts with a guess <math>\mathbf{x}_0</math> for a local minimum of <math>f</math>, and considers the sequence <math>\mathbf{x}_0, \mathbf{x}_1, \mathbf{x}_2, \ldots</math> such that


:<math>\mathbf{x}_{n+1}=\mathbf{x}_n-\gamma_n \nabla F(\mathbf{x}_n),\ n \ge 0.</math>
:<math>\mathbf{x}_{n+1}=\mathbf{x}_n-\eta_n\nabla f(\mathbf{x}_n),\ n \ge 0.</math>


We have a [[Monotonic function|monotonic]] sequence
We have a [[Monotonic function|monotonic]] sequence


:<math>F(\mathbf{x}_0)\ge F(\mathbf{x}_1)\ge F(\mathbf{x}_2)\ge \cdots,</math>
:<math>f(\mathbf{x}_0)\ge f(\mathbf{x}_1)\ge f(\mathbf{x}_2)\ge \cdots,</math>


so the sequence <math>(\mathbf{x}_n)</math> converges to the desired local minimum. Note that the value of the ''step size'' <math>\gamma</math> is allowed to change at every iteration.  
so the sequence <math>(\mathbf{x}_n)</math> converges to the desired local minimum. Note that the value of the ''step size'' <math>\eta</math> is allowed to change at every iteration.  


It is possible to guarantee the [[Convergent series|convergence]] to a local minimum under certain assumptions on the function <math>F</math> (for example, <math>F</math> [[Convex function|convex]] and <math>\nabla F</math> [[Lipschitz continuity|Lipschitz]]) and particular choices of <math>\gamma</math>. Those include the sequence
It is possible to guarantee the [[Convergent series|convergence]] to a local minimum under certain assumptions on the function <math>f</math> (for example, <math>f</math> [[Convex function|convex]] and <math>\nabla f</math> [[Lipschitz continuity|Lipschitz]]) and particular choices of <math>\eta</math>. Those include the sequence


<math>\gamma_{n} = \frac{ \left | \left (\mathbf x_{n} - \mathbf x_{n-1} \right )^T \left [\nabla F (\mathbf x_{n}) - \nabla F (\mathbf x_{n-1}) \right ] \right |}{\left \|\nabla F(\mathbf{x}_{n}) - \nabla F(\mathbf{x}_{n-1}) \right \|^2}</math>
<math>\eta_{n} = \frac{ \left | \left (\mathbf x_{n} - \mathbf x_{n-1} \right )^\top \left [\nabla f (\mathbf x_{n}) - \nabla f (\mathbf x_{n-1}) \right ] \right |}{\left \|\nabla f(\mathbf{x}_{n}) - \nabla f(\mathbf{x}_{n-1}) \right \|^2}</math>


as in the [[Barzilai-Borwein method]],<ref>{{cite journal |last1=Barzilai |first1=Jonathan |last2=Borwein |first2=Jonathan M. |year=1988 |title=Two-Point Step Size Gradient Methods |journal=IMA Journal of Numerical Analysis |volume=8 |issue=1 |pages=141–148 |doi=10.1093/imanum/8.1.141}}</ref><ref>{{cite book |last=Fletcher |first=R. |title=Optimization and Control with Applications |publisher=Springer |year=2005 |isbn=0-387-24254-6 |editor-last=Qi |editor-first=L. |series=Applied Optimization |volume=96 |location=Boston |pages=235–256 |chapter=On the Barzilai–Borwein Method |editor2-last=Teo |editor2-first=K. |editor3-last=Yang |editor3-first=X.}}</ref> or a sequence <math> \gamma_n</math> satisfying the [[Wolfe conditions]] (which can be found by using [[line search]]). When the function <math>F</math> is [[Convex function|convex]], all local minima are also global minima, so in this case gradient descent can converge to the global solution.
as in the [[Barzilai-Borwein method]],<ref>{{cite journal |last1=Barzilai |first1=Jonathan |last2=Borwein |first2=Jonathan M. |year=1988 |title=Two-Point Step Size Gradient Methods |journal=IMA Journal of Numerical Analysis |volume=8 |issue=1 |pages=141–148 |doi=10.1093/imanum/8.1.141}}</ref><ref>{{cite book |last=Fletcher |first=R. |title=Optimization and Control with Applications |publisher=Springer |year=2005 |isbn=0-387-24254-6 |editor-last=Qi |editor-first=L. |series=Applied Optimization |volume=96 |location=Boston |pages=235–256 |chapter=On the Barzilai–Borwein Method |editor2-last=Teo |editor2-first=K. |editor3-last=Yang |editor3-first=X.}}</ref> or a sequence <math> \eta_n</math> satisfying the [[Wolfe conditions]] (which can be found by using [[line search]]). When the function <math>f</math> is [[Convex function|convex]], all local minima are also global minima, so in this case gradient descent can converge to the global solution.


This process is illustrated in the adjacent picture. Here, <math>F</math> is assumed to be defined on the plane, and that its graph has a [[Bowl (vessel)|bowl]] shape.  The blue curves are the [[contour line]]s, that is, the regions on which the value of <math>F</math> is constant. A red arrow originating at a point shows the direction of the negative gradient at that point. Note that the (negative) gradient at a point is [[orthogonal]] to the contour line going through that point. We see that gradient ''descent'' leads us to the bottom of the bowl, that is, to the point where the value of the function <math>F</math> is minimal.
This process is illustrated in the adjacent picture. Here, <math>f</math> is assumed to be defined on the plane, and that its graph has a [[Bowl (vessel)|bowl]] shape.  The blue curves are the [[contour line]]s, that is, the regions on which the value of <math>f</math> is constant. A red arrow originating at a point shows the direction of the negative gradient at that point. Note that the (negative) gradient at a point is [[orthogonal]] to the contour line going through that point. We see that gradient ''descent'' leads us to the bottom of the bowl, that is, to the point where the value of the function <math>f</math> is minimal.


=== An analogy for understanding gradient descent ===
=== An analogy for understanding gradient descent ===
Line 44: Line 44:


=== Choosing the step size and descent direction ===
=== Choosing the step size and descent direction ===
Since using a step size <math>\gamma</math> that is too small would slow convergence, and a <math>\gamma</math> too large would lead to overshoot and divergence, finding a good setting of <math>\gamma</math> is an important practical problem. [[Philip Wolfe (mathematician)|Philip Wolfe]] also advocated using "clever choices of the [descent] direction" in practice.<ref>{{cite journal |last1=Wolfe |first1=Philip |title=Convergence Conditions for Ascent Methods |journal=SIAM Review |date=April 1969 |volume=11 |issue=2 |pages=226–235 |doi=10.1137/1011036 }}</ref> While using a direction that deviates from the steepest descent direction may seem counter-intuitive, the idea is that the smaller slope may be compensated for by being sustained over a much longer distance.
Since using a step size <math>\eta</math> that is too small would slow convergence, and a <math>\eta</math> too large would lead to overshoot and divergence, finding a good setting of <math>\eta</math> is an important practical problem. [[Philip Wolfe (mathematician)|Philip Wolfe]] also advocated using "clever choices of the [descent] direction" in practice.<ref>{{cite journal |last1=Wolfe |first1=Philip |title=Convergence Conditions for Ascent Methods |journal=SIAM Review |date=April 1969 |volume=11 |issue=2 |pages=226–235 |doi=10.1137/1011036 }}</ref> While using a direction that deviates from the steepest descent direction may seem counter-intuitive, the idea is that the smaller slope may be compensated for by being sustained over a much longer distance.


To reason about this mathematically, consider a direction <math> \mathbf{p}_n</math> and step size <math> \gamma_n</math> and consider the more general update:
To reason about this mathematically, consider a direction <math> \mathbf{p}_n</math> and step size <math> \eta_n</math> and consider the more general update:
:<math> \mathbf{a}_{n+1} = \mathbf{a}_n-\gamma_n\,\mathbf{p}_n</math>.
:<math> \mathbf{a}_{n+1} = \mathbf{a}_n-\eta_n\,\mathbf{p}_n</math>.
Finding good settings of <math> \mathbf{p}_n</math> and <math> \gamma_n</math> requires some thought. First of all, we would like the update direction to point downhill. Mathematically, letting <math> \theta_n</math> denote the angle between <math>-\nabla F(\mathbf{a_n})</math> and <math> \mathbf{p}_n</math>, this requires that <math> \cos \theta_n > 0.</math> To say more, we need more information about the objective function that we are optimising. Under the fairly weak assumption that <math>F</math> is continuously differentiable, we may prove that:<ref>{{cite arXiv|last1=Bernstein|first1=Jeremy|last2=Vahdat|first2=Arash|last3=Yue|first3=Yisong|last4=Liu|first4=Ming-Yu|date=2020-06-12|title=On the distance between two neural networks and the stability of learning|class=cs.LG|eprint=2002.03432}}</ref>
Finding good settings of <math> \mathbf{p}_n</math> and <math> \eta_n</math> requires some thought. First of all, we would like the update direction to point downhill. Mathematically, letting <math> \theta_n</math> denote the angle between <math>-\nabla f(\mathbf{a_n})</math> and <math> \mathbf{p}_n</math>, this requires that <math> \cos \theta_n > 0.</math> To say more, we need more information about the objective function that we are optimising. Under the fairly weak assumption that <math>f</math> is continuously differentiable, we may prove that:<ref>{{cite arXiv|last1=Bernstein|first1=Jeremy|last2=Vahdat|first2=Arash|last3=Yue|first3=Yisong|last4=Liu|first4=Ming-Yu|date=2020-06-12|title=On the distance between two neural networks and the stability of learning|class=cs.LG|eprint=2002.03432}}</ref>
{{NumBlk|:|<math> F(\mathbf{a}_{n+1}) \leq F(\mathbf{a}_n) - \gamma_n \|\nabla F(\mathbf{a}_n)\|_2 \|\mathbf{p}_n\|_2 \left[\cos \theta_n - \max_{t\in[0,1]} \frac{\|\nabla F(\mathbf{a}_n - t \gamma_n \mathbf{p}_n) - \nabla F(\mathbf{a}_n)\|_2}{\| \nabla F(\mathbf{a}_n) \|_2}\right]</math>|{{EquationRef|1}}}}
{{NumBlk|:|<math> f(\mathbf{a}_{n+1}) \leq f(\mathbf{a}_n) - \eta_n \|\nabla f(\mathbf{a}_n)\|_2 \|\mathbf{p}_n\|_2 \left(\cos \theta_n - \max_{t\in[0,1]} \frac{\|\nabla f(\mathbf{a}_n - t \eta_n \mathbf{p}_n) - \nabla f(\mathbf{a}_n)\|_2}{\| \nabla f(\mathbf{a}_n) \|_2}\right)</math>|{{EquationRef|1}}}}
This inequality implies that the amount by which we can be sure the function <math>F</math> is decreased depends on a trade off between the two terms in square brackets. The first term in square brackets measures the angle between the descent direction and the negative gradient. The second term measures how quickly the gradient changes along the descent direction.
This inequality implies that the amount by which we can be sure the function <math>f</math> is decreased depends on a trade off between the two terms in square brackets. The first term in square brackets measures the angle between the descent direction and the negative gradient. The second term measures how quickly the gradient changes along the descent direction.


In principle inequality ({{EquationNote|1}}) could be optimized over <math> \mathbf{p}_n</math> and <math> \gamma_n</math> to choose an optimal step size and direction. The problem is that evaluating the second term in square brackets requires evaluating <math> \nabla F(\mathbf{a}_n - t \gamma_n \mathbf{p}_n)</math>, and extra gradient evaluations are generally expensive and undesirable. Some ways around this problem are:
In principle inequality ({{EquationNote|1}}) could be optimized over <math> \mathbf{p}_n</math> and <math> \eta_n</math> to choose an optimal step size and direction. The problem is that evaluating the second term in square brackets requires evaluating <math> \nabla f(\mathbf{a}_n - t \eta_n \mathbf{p}_n)</math>, and extra gradient evaluations are generally expensive and undesirable. Some ways around this problem are:


* Forgo the benefits of a clever descent direction by setting <math>\mathbf{p}_n = \nabla F(\mathbf{a_n})</math>, and use [[line search]] to find a suitable step-size <math> \gamma_n</math>, such as one that satisfies the [[Wolfe conditions]]. A more economic way of choosing learning rates is [[backtracking line search]], a method that has both good theoretical guarantees and experimental results. Note that one does not need to choose  <math>\mathbf{p}_n </math> to be the gradient; any direction that has positive inner product with the gradient will result in a reduction of the function value (for a sufficiently small value of <math> \gamma_n</math>).
* Forgo the benefits of a clever descent direction by setting <math>\mathbf{p}_n = \nabla f(\mathbf{a_n})</math>, and use [[line search]] to find a suitable step-size <math> \gamma_n</math>, such as one that satisfies the [[Wolfe conditions]]. A more economic way of choosing learning rates is [[backtracking line search]], a method that has both good theoretical guarantees and experimental results. Note that one does not need to choose  <math>\mathbf{p}_n </math> to be the gradient; any direction that has positive inner product with the gradient will result in a reduction of the function value (for a sufficiently small value of <math> \eta_n</math>).
* Assuming that <math>F</math> is twice-differentiable, use its Hessian <math>\nabla^2 F</math> to estimate <math> \|\nabla F(\mathbf{a}_n - t \gamma_n \mathbf{p}_n) - \nabla F(\mathbf{a}_n)\|_2 \approx \| t \gamma_n \nabla^2 F(\mathbf{a}_n) \mathbf{p}_n\|.</math>Then choose <math> \mathbf{p}_n</math> and <math> \gamma_n</math> by optimising inequality ({{EquationNote|1}}).
* Assuming that <math>f</math> is twice-differentiable, use its Hessian <math>\nabla^2 f</math> to estimate <math> \|\nabla f(\mathbf{a}_n - t \eta_n \mathbf{p}_n) - \nabla f(\mathbf{a}_n)\|_2 \approx \| t \eta_n \nabla^2 f(\mathbf{a}_n) \mathbf{p}_n\|.</math>Then choose <math> \mathbf{p}_n</math> and <math> \eta_n</math> by optimising inequality ({{EquationNote|1}}).
* Assuming that <math>\nabla F</math> is [[Lipschitz continuity|Lipschitz]], use its Lipschitz constant <math> L</math> to bound <math> \|\nabla F(\mathbf{a}_n - t \gamma_n \mathbf{p}_n) - \nabla F(\mathbf{a}_n)\|_2 \leq L t \gamma_n \|\mathbf{p}_n\|.</math> Then choose <math> \mathbf{p}_n</math> and <math> \gamma_n</math> by optimising inequality ({{EquationNote|1}}).
* Assuming that <math>\nabla f</math> is [[Lipschitz continuity|Lipschitz]], use its Lipschitz constant <math> L</math> to bound <math> \|\nabla f(\mathbf{a}_n - t \eta_n \mathbf{p}_n) - \nabla f(\mathbf{a}_n)\|_2 \leq L t \eta_n \|\mathbf{p}_n\|.</math> Then choose <math> \mathbf{p}_n</math> and <math> \eta_n</math> by optimising inequality ({{EquationNote|1}}).
* Build a custom model of <math> \max_{t\in[0,1]} \frac{\|\nabla F(\mathbf{a}_n - t \gamma_n \mathbf{p}_n) - \nabla F(\mathbf{a}_n)\|_2}{\| \nabla F(\mathbf{a}_n) \|_2}</math> for <math>F</math>. Then choose <math> \mathbf{p}_n</math> and <math> \gamma_n</math> by optimising inequality ({{EquationNote|1}}).
* Build a custom model of <math> \max_{t\in[0,1]} \frac{\|\nabla f(\mathbf{a}_n - t \eta_n \mathbf{p}_n) - \nabla f(\mathbf{a}_n)\|_2}{\| \nabla f(\mathbf{a}_n) \|_2}</math> for <math>f</math>. Then choose <math> \mathbf{p}_n</math> and <math> \eta_n</math> by optimising inequality ({{EquationNote|1}}).
* Under stronger assumptions on the function <math>F</math> such as [[Convex function|convexity]], more [[#Fast gradient methods|advanced techniques]] may be possible.
* Under stronger assumptions on the function <math>f</math> such as [[Convex function|convexity]], more [[#Fast gradient methods|advanced techniques]] may be possible.


Usually by following one of the recipes above, [[convergent series|convergence]] to a local minimum can be guaranteed. When the function <math>F</math> is [[Convex function|convex]], all local minima are also global minima, so in this case gradient descent can converge to the global solution.
Usually by following one of the recipes above, [[convergent series|convergence]] to a local minimum can be guaranteed. When the function <math>f</math> is [[Convex function|convex]], all local minima are also global minima, so in this case gradient descent can converge to the global solution.


==Solution of a linear system==
==Solution of a linear system==
Line 68: Line 68:
Gradient descent can be used to solve a [[system of linear equations]]
Gradient descent can be used to solve a [[system of linear equations]]


:<math>A\mathbf{x}-\mathbf{b}=0</math>
:<math>\mathbf{A}\mathbf{x}-\mathbf{b}=0</math>


reformulated as a quadratic minimization problem.
reformulated as a quadratic minimization problem.
If the system matrix <math>A</math> is real [[Symmetric matrix|symmetric]] and [[positive-definite matrix|positive-definite]], an objective function is defined as the quadratic function, with minimization of
If the system matrix <math>\mathbf{A}</math> is real [[Symmetric matrix|symmetric]] and [[positive-definite matrix|positive-definite]], an objective function is defined as the quadratic function, with minimization of


:<math>F(\mathbf{x})=\mathbf{x}^T A\mathbf{x}-2\mathbf{x}^T\mathbf{b},</math>
:<math>f(\mathbf{x})=\mathbf{x}^\top \mathbf{A}\mathbf{x}-2\mathbf{x}^\top \mathbf{b},</math>


so that
so that


:<math>\nabla F(\mathbf{x})=2(A\mathbf{x}-\mathbf{b}).</math>
:<math>\nabla f(\mathbf{x})=2(\mathbf{A}\mathbf{x}-\mathbf{b}).</math>


For a general real matrix <math>A</math>, [[linear least squares]] define
For a general real matrix <math>\mathbf{A}</math>, [[linear least squares]] define


:<math>F(\mathbf{x})=\left\|A\mathbf{x}-\mathbf{b}\right\|^2.</math>
:<math>f(\mathbf{x})=\left\|\mathbf{A}\mathbf{x}-\mathbf{b}\right\|^2.</math>


In traditional linear least squares for real <math>A</math> and <math>\mathbf{b}</math> the [[Euclidean norm]] is used, in which case
In traditional linear least squares for real <math>\mathbf{A}</math> and <math>\mathbf{b}</math> the [[Euclidean norm]] is used, in which case


:<math>\nabla F(\mathbf{x})=2A^T(A\mathbf{x}-\mathbf{b}).</math>
:<math>\nabla f(\mathbf{x})=2\mathbf{A}^\top(\mathbf{A}\mathbf{x}-\mathbf{b}).</math>


The [[line search]] minimization, finding the locally optimal step size <math>\gamma</math> on every iteration, can be performed analytically for quadratic functions, and explicit formulas for the locally optimal <math>\gamma</math> are known.<ref name="BP">{{cite book |author-link=Boris T. Polyak |last=Polyak |first=Boris |title=Introduction to Optimization |year=1987 |language=en |url=https://www.researchgate.net/publication/342978480 }}</ref><ref name=saad1996iterative>{{cite book|last=Saad|first=Yousef|title=Iterative methods for sparse linear systems|year=2003|publisher=Society for Industrial and Applied Mathematics|location=Philadelphia, Pa.|isbn=978-0-89871-534-7|pages=[https://archive.org/details/iterativemethods0000saad/page/195 195]|edition=2nd|url=https://archive.org/details/iterativemethods0000saad/page/195}}</ref>
The [[line search]] minimization, finding the locally optimal step size <math>\eta</math> on every iteration, can be performed analytically for quadratic functions, and explicit formulas for the locally optimal <math>\eta</math> are known.<ref name="BP">{{cite book |author-link=Boris T. Polyak |last=Polyak |first=Boris |title=Introduction to Optimization |year=1987 |language=en |url=https://www.researchgate.net/publication/342978480 }}</ref><ref name=saad1996iterative>{{cite book|last=Saad|first=Yousef|title=Iterative methods for sparse linear systems|year=2003|publisher=Society for Industrial and Applied Mathematics|location=Philadelphia, Pa.|isbn=978-0-89871-534-7|pages=[https://archive.org/details/iterativemethods0000saad/page/195 195]|edition=2nd|url=https://archive.org/details/iterativemethods0000saad/page/195}}</ref>


For example, for real [[Symmetric matrix|symmetric]] and [[positive-definite matrix|positive-definite]] matrix <math>A</math>, a simple algorithm can be as follows,<ref name="BP" />
For example, for real [[Symmetric matrix|symmetric]] and [[positive-definite matrix|positive-definite]] matrix <math>\mathbf{A}</math>, a simple algorithm can be as follows,<ref name="BP" />
:<math>\begin{align}
:<math>\begin{align}
& \text{repeat in the loop:} \\
& \text{repeat in the loop:} \\
& \qquad \mathbf{r} := \mathbf{b} - \mathbf{A x} \\
& \qquad \mathbf{r} := \mathbf{b} - \mathbf{A x} \\
& \qquad \gamma := {\mathbf{r}^\mathsf{T} \mathbf{r}}/{\mathbf{r}^\mathsf{T} \mathbf{A r}}  \\
& \qquad \eta := {\mathbf{r}^\top \mathbf{r}}/{\mathbf{r}^\top \mathbf{A r}}  \\
& \qquad \mathbf{x} := \mathbf{x} + \gamma \mathbf{r} \\
& \qquad \mathbf{x} := \mathbf{x} + \eta\mathbf{r} \\
& \qquad \hbox{if } \mathbf{r}^\mathsf{T} \mathbf{r} \text{ is sufficiently small, then exit loop} \\
& \qquad \hbox{if } \mathbf{r}^\top \mathbf{r} \text{ is sufficiently small, then exit loop} \\
& \text{end repeat loop} \\
& \text{end repeat loop} \\
& \text{return } \mathbf{x} \text{ as the result}
& \text{return } \mathbf{x} \text{ as the result}
\end{align}</math>
\end{align}</math>


To avoid multiplying by <math>A</math> twice per iteration,
To avoid multiplying by <math>\mathbf{A}</math> twice per iteration,
we note that <math>\mathbf{x} := \mathbf{x} + \gamma \mathbf{r}</math> implies <math>\mathbf{r} := \mathbf{r} - \gamma \mathbf{A r}</math>, which gives the traditional algorithm,<ref name=":0">{{cite journal |first1=Henricus |last1=Bouwmeester |first2=Andrew |last2=Dougherty |first3=Andrew V. |last3=Knyazev |title=Nonsymmetric Preconditioning for Conjugate Gradient and Steepest Descent Methods |journal=Procedia Computer Science |volume=51 |pages=276–285 |year=2015 |doi=10.1016/j.procs.2015.05.241 |doi-access=free |arxiv=1212.6680 }}</ref>
we note that <math>\mathbf{x} := \mathbf{x} + \eta\mathbf{r}</math> implies <math>\mathbf{r} := \mathbf{r} - \eta\mathbf{A r}</math>, which gives the traditional algorithm,<ref name=":0">{{cite journal |first1=Henricus |last1=Bouwmeester |first2=Andrew |last2=Dougherty |first3=Andrew V. |last3=Knyazev |title=Nonsymmetric Preconditioning for Conjugate Gradient and Steepest Descent Methods |journal=Procedia Computer Science |volume=51 |pages=276–285 |year=2015 |doi=10.1016/j.procs.2015.05.241 |doi-access=free |arxiv=1212.6680 }}</ref>
:<math>\begin{align}
:<math>\begin{align}
& \mathbf{r} := \mathbf{b} - \mathbf{A x} \\
& \mathbf{r} := \mathbf{b} - \mathbf{A x} \\
& \text{repeat in the loop:} \\
& \text{repeat in the loop:} \\
& \qquad \gamma := {\mathbf{r}^\mathsf{T} \mathbf{r}}/{\mathbf{r}^\mathsf{T} \mathbf{A r}}  \\
& \qquad \eta := {\mathbf{r}^\top \mathbf{r}}/{\mathbf{r}^\top \mathbf{A r}}  \\
& \qquad \mathbf{x} := \mathbf{x} + \gamma \mathbf{r} \\
& \qquad \mathbf{x} := \mathbf{x} + \eta\mathbf{r} \\
& \qquad \hbox{if } \mathbf{r}^\mathsf{T} \mathbf{r} \text{ is sufficiently small, then exit loop} \\
& \qquad \hbox{if } \mathbf{r}^\top \mathbf{r} \text{ is sufficiently small, then exit loop} \\
& \qquad \mathbf{r} := \mathbf{r} - \gamma \mathbf{A r} \\
& \qquad \mathbf{r} := \mathbf{r} - \eta\mathbf{A r} \\
& \text{end repeat loop} \\
& \text{end repeat loop} \\
& \text{return } \mathbf{x} \text{ as the result}
& \text{return } \mathbf{x} \text{ as the result}
\end{align}</math>
\end{align}</math>


The method is rarely used for solving linear equations, with the [[conjugate gradient method]] being one of the most popular alternatives. The number of gradient descent iterations is commonly proportional to the spectral [[condition number]] <math>\kappa(A)</math> of the system matrix <math>A</math> (the ratio of the maximum to minimum [[eigenvalues]] of {{nowrap|<math>A^TA</math>)}}, while the convergence of [[conjugate gradient method]] is typically determined by a square root of the condition number, i.e., is much faster. Both methods can benefit from [[Preconditioner|preconditioning]], where gradient descent may require less assumptions on the preconditioner.<ref name=":0" />
The method is rarely used for solving linear equations, with the [[conjugate gradient method]] being one of the most popular alternatives. The number of gradient descent iterations is commonly proportional to the spectral [[condition number]] <math>\kappa(\mathbf{A})</math> of the system matrix <math>\mathbf{A}</math> (the ratio of the maximum to minimum [[eigenvalues]] of {{nowrap|<math>\mathbf{A}^\top \mathbf{A}</math>)}}, while the convergence of [[conjugate gradient method]] is typically determined by a square root of the condition number, i.e., is much faster. Both methods can benefit from [[Preconditioner|preconditioning]], where gradient descent may require less assumptions on the preconditioner.<ref name=":0" />


=== Geometric behavior and residual orthogonality ===
=== Geometric behavior and residual orthogonality ===
In steepest descent applied to solving <math> A \vec{x} = \vec{b} </math>, where <math> A </math> is symmetric positive-definite, the residual vectors <math> \vec{r}_k = \vec{b} - A\vec{x}_k </math> are orthogonal across iterations:
In steepest descent applied to solving <math> \mathbf{A x} = \mathbf{b} </math>, where <math> \mathbf{A} </math> is symmetric positive-definite, the residual vectors <math> \mathbf{r}_k = \mathbf{b} - \mathbf{A}\mathbf{x}_k </math> are orthogonal across iterations:


:<math>
:<math>
\vec{r}_{k+1} \cdot \vec{r}_k = 0.
\langle \mathbf{r}_{k+1}, \mathbf{r}_k\rangle = 0.
</math>
</math>


Because each step is taken in the steepest direction, steepest-descent steps
Because each step is taken in the steepest direction, steepest-descent steps
alternate between directions aligned with the extreme axes of the elongated
alternate between directions aligned with the extreme axes of the elongated
level sets.  When <math>\kappa(A)</math> is large, this produces a
level sets.  When <math>\kappa(\mathbf{A})</math> is large, this produces a
characteristic zig-zag path. The poor conditioning of <math> A </math> is the primary cause of the slow convergence, and orthogonality of successive residuals reinforces this alternation.
characteristic zig-zag path. The poor conditioning of <math> \mathbf{A} </math> is the primary cause of the slow convergence, and orthogonality of successive residuals reinforces this alternation.


[[File:Steepest descent convergence path for A = 2 2, 2 3.png|thumb|Convergence path of steepest descent method for A = [[2, 2], [2, 3]]]]
[[File:Steepest descent convergence path for A = 2 2, 2 3.png|thumb|Convergence path of steepest descent method for A = [[2, 2], [2, 3]]]]


As shown in the image on the right, steepest descent converges slowly due to the high condition number of <math> A </math>, and the orthogonality of residuals forces each new direction to undo the overshoot from the previous step. The result is a path that zigzags toward the solution. This inefficiency is one reason conjugate gradient or preconditioning methods are preferred.<ref>{{Cite book | author1=Holmes, M. | title=Introduction to Scientific Computing and Data Analysis, 2nd Ed | year=2023 | publisher=Springer | isbn=978-3-031-22429-4 }}</ref>
As shown in the image on the right, steepest descent converges slowly due to the high condition number of <math> \mathbf{A} </math>, and the orthogonality of residuals forces each new direction to undo the overshoot from the previous step. The result is a path that zigzags toward the solution. This inefficiency is one reason conjugate gradient or preconditioning methods are preferred.<ref>{{Cite book | author1=Holmes, M. | title=Introduction to Scientific Computing and Data Analysis, 2nd Ed | year=2023 | publisher=Springer | isbn=978-3-031-22429-4 }}</ref>


==Solution of a non-linear system==
==Solution of a non-linear system==
Line 161: Line 161:
One might now define the objective function
One might now define the objective function


:<math>\begin{align}F(\mathbf{x}) &= \frac{1}{2} G^\mathrm{T}(\mathbf{x}) G(\mathbf{x}) \\&=\frac{1}{2} \left[ \left (3x_1-\cos(x_2x_3)-\frac{3}{2} \right)^2  + \left(4x_1^2-625x_2^2+2x_2-1 \right)^2 +\right.\\
:<math>\begin{align}f(\mathbf{x}) &= \frac{1}{2} G^\top(\mathbf{x}) G(\mathbf{x}) \\&=\frac{1}{2} \left[ \left (3x_1-\cos(x_2x_3)-\frac{3}{2} \right)^2  + \left(4x_1^2-625x_2^2+2x_2-1 \right)^2 +\right.\\
&{}\qquad\left. \left(\exp(-x_1x_2) + 20x_3 + \frac{10\pi-3}{3} \right)^2 \right],\end{align}</math>
&{}\qquad\left. \left(\exp(-x_1x_2) + 20x_3 + \frac{10\pi-3}{3} \right)^2 \right],\end{align}</math>


Line 173: Line 173:
We know that
We know that


:<math>\mathbf{x}^{(1)}=\mathbf{0}-\gamma_0 \nabla F(\mathbf{0}) = \mathbf{0}-\gamma_0 J_G(\mathbf{0})^\mathrm{T} G(\mathbf{0}),</math>
:<math>\mathbf{x}^{(1)}=\mathbf{0}-\eta_0 \nabla f(\mathbf{0}) = \mathbf{0}-\eta_0 J_G(\mathbf{0})^\top G(\mathbf{0}),</math>


where the [[Jacobian matrix]] <math>J_G</math> is given by
where the [[Jacobian matrix]] <math>J_G</math> is given by
Line 197: Line 197:
Thus
Thus


:<math>\mathbf{x}^{(1)}= \mathbf{0}-\gamma_0 \begin{bmatrix}
:<math>\mathbf{x}^{(1)}= \mathbf{0}-\eta_0 \begin{bmatrix}
   -7.5\\
   -7.5\\
   -2\\
   -2\\
Line 205: Line 205:
and
and


:<math>F(\mathbf{0}) = 0.5 \left( (-2.5)^2 + (-1)^2 + (10.472)^2 \right) = 58.456.</math>
:<math>f(\mathbf{0}) = 0.5 \left( (-2.5)^2 + (-1)^2 + (10.472)^2 \right) = 58.456.</math>


[[File:Gradient Descent Example Nonlinear Equations.gif|thumb|right|350px|An animation showing the first 83 iterations of gradient descent applied to this example. Surfaces are [[isosurface]]s of <math>F(\mathbf{x}^{(n)})</math> at current guess <math>\mathbf{x}^{(n)}</math>, and arrows show the direction of descent. Due to a small and constant step size, the convergence is slow.]]
[[File:Gradient Descent Example Nonlinear Equations.gif|thumb|right|350px|An animation showing the first 83 iterations of gradient descent applied to this example. Surfaces are [[isosurface]]s of <math>f(\mathbf{x}^{(n)})</math> at current guess <math>\mathbf{x}^{(n)}</math>, and arrows show the direction of descent. Due to a small and constant step size, the convergence is slow.]]


Now, a suitable <math>\gamma_0</math> must be found such that
Now, a suitable <math>\eta_0</math> must be found such that


:<math>F\left (\mathbf{x}^{(1)}\right ) \le F\left (\mathbf{x}^{(0)}\right ) = F(\mathbf{0}).</math>
:<math>f\left (\mathbf{x}^{(1)}\right ) \le f\left (\mathbf{x}^{(0)}\right ) = f(\mathbf{0}).</math>


This can be done with any of a variety of [[line search]] algorithms. One might also simply guess <math>\gamma_0=0.001,</math> which gives
This can be done with any of a variety of [[line search]] algorithms. One might also simply guess <math>\eta_0=0.001,</math> which gives


:<math> \mathbf{x}^{(1)}=\begin{bmatrix}
:<math> \mathbf{x}^{(1)}=\begin{bmatrix}
Line 223: Line 223:
Evaluating the objective function at this value, yields
Evaluating the objective function at this value, yields


:<math>F \left (\mathbf{x}^{(1)}\right ) = 0.5 \left ((-2.48)^2 + (-1.00)^2 + (6.28)^2 \right ) = 23.306.</math>
:<math>f \left (\mathbf{x}^{(1)}\right ) = 0.5 \left ((-2.48)^2 + (-1.00)^2 + (6.28)^2 \right ) = 23.306.</math>


The decrease from <math>F(\mathbf{0})=58.456</math> to the next step's value of
The decrease from <math>f(\mathbf{0})=58.456</math> to the next step's value of


:<math> F\left (\mathbf{x}^{(1)}\right ) =23.306 </math>
:<math> f \left (\mathbf{x}^{(1)}\right ) =23.306 </math>


is a sizable decrease in the objective function. Further steps would reduce its value further until an approximate solution to the system was found.
is a sizable decrease in the objective function. Further steps would reduce its value further until an approximate solution to the system was found.
Line 239: Line 239:
The gradient descent can take many iterations to compute a local minimum with a required [[accuracy]], if the [[curvature]] in different directions is very different for the given function. For such functions, [[preconditioning]], which changes the geometry of the space to shape the function level sets like [[concentric circles]], cures the slow convergence. Constructing and applying preconditioning can be computationally expensive, however.
The gradient descent can take many iterations to compute a local minimum with a required [[accuracy]], if the [[curvature]] in different directions is very different for the given function. For such functions, [[preconditioning]], which changes the geometry of the space to shape the function level sets like [[concentric circles]], cures the slow convergence. Constructing and applying preconditioning can be computationally expensive, however.


The gradient descent can be modified via momentums<ref>{{Cite journal |last1=Abdulkadirov |first1=Ruslan |last2=Lyakhov |first2=Pavel |last3=Nagornov |first3=Nikolay |date=January 2023 |title=Survey of Optimization Algorithms in Modern Neural Networks |journal=Mathematics |language=en |volume=11 |issue=11 |pages=2466 |doi=10.3390/math11112466 |doi-access=free |issn=2227-7390}}</ref> ([[Nesterov]], Polyak,<ref>{{Cite journal |last1=Diakonikolas |first1=Jelena |last2=Jordan |first2=Michael I. |date=January 2021 |title=Generalized Momentum-Based Methods: A Hamiltonian Perspective |url=https://epubs.siam.org/doi/10.1137/20M1322716 |journal=SIAM Journal on Optimization |language=en |volume=31 |issue=1 |pages=915–944 |doi=10.1137/20M1322716 |arxiv=1906.00436 |issn=1052-6234}}</ref> and Frank-Wolfe<ref>{{Cite journal |last=Meyer |first=Gerard G. L. |date=November 1974 |title=Accelerated Frank–Wolfe Algorithms |url=http://epubs.siam.org/doi/10.1137/0312050 |journal=SIAM Journal on Control |language=en |volume=12 |issue=4 |pages=655–663 |doi=10.1137/0312050 |issn=0036-1402}}</ref>) and heavy-ball parameters (exponential moving averages<ref>{{Citation |last1=Kingma |first1=Diederik P. |title=Adam: A Method for Stochastic Optimization |date=2017-01-29 |last2=Ba |first2=Jimmy|arxiv=1412.6980 }}</ref> and positive-negative momentum<ref>{{Cite journal |last1=Xie |first1=Zeke |last2=Yuan |first2=Li |last3=Zhu |first3=Zhanxing |last4=Sugiyama |first4=Masashi |date=2021-07-01 |title=Positive-Negative Momentum: Manipulating Stochastic Gradient Noise to Improve Generalization |url=https://proceedings.mlr.press/v139/xie21h.html |journal=Proceedings of the 38th International Conference on Machine Learning |language=en |publisher=PMLR |pages=11448–11458|arxiv=2103.17182 }}</ref>). The main examples of such optimizers are Adam, DiffGrad, Yogi, AdaBelief, etc.
The gradient descent can be modified via momentums<ref>{{Cite journal |last1=Abdulkadirov |first1=Ruslan |last2=Lyakhov |first2=Pavel |last3=Nagornov |first3=Nikolay |date=January 2023 |title=Survey of Optimization Algorithms in Modern Neural Networks |journal=Mathematics |language=en |volume=11 |issue=11 |pages=2466 |doi=10.3390/math11112466 |doi-access=free |issn=2227-7390}}</ref> ([[Nesterov]], Polyak,<ref>{{Cite journal |last1=Diakonikolas |first1=Jelena |last2=Jordan |first2=Michael I. |date=January 2021 |title=Generalized Momentum-Based Methods: A Hamiltonian Perspective |url=https://epubs.siam.org/doi/10.1137/20M1322716 |journal=SIAM Journal on Optimization |language=en |volume=31 |issue=1 |pages=915–944 |doi=10.1137/20M1322716 |arxiv=1906.00436 |issn=1052-6234}}</ref> and Frank-Wolfe<ref>{{Cite journal |last=Meyer |first=Gerard G. L. |date=November 1974 |title=Accelerated Frank–Wolfe Algorithms |url=http://epubs.siam.org/doi/10.1137/0312050 |journal=SIAM Journal on Control |language=en |volume=12 |issue=4 |pages=655–663 |doi=10.1137/0312050 |issn=0036-1402|url-access=subscription }}</ref>) and heavy-ball parameters (exponential moving averages<ref>{{Citation |last1=Kingma |first1=Diederik P. |title=Adam: A Method for Stochastic Optimization |date=2017-01-29 |last2=Ba |first2=Jimmy|arxiv=1412.6980 }}</ref> and positive-negative momentum<ref>{{Cite journal |last1=Xie |first1=Zeke |last2=Yuan |first2=Li |last3=Zhu |first3=Zhanxing |last4=Sugiyama |first4=Masashi |date=2021-07-01 |title=Positive-Negative Momentum: Manipulating Stochastic Gradient Noise to Improve Generalization |url=https://proceedings.mlr.press/v139/xie21h.html |journal=Proceedings of the 38th International Conference on Machine Learning |language=en |publisher=PMLR |pages=11448–11458|arxiv=2103.17182 }}</ref>). The main examples of such optimizers are Adam, DiffGrad, Yogi, AdaBelief, etc.


Methods based on [[Newton's method in optimization|Newton's method]] and inversion of the [[Hessian matrix|Hessian]] using [[conjugate gradient]] techniques can be better alternatives.<ref>{{cite book |first1=W. H. |last1=Press |author-link1 = William H. Press |first2=S. A. |last2=Teukolsky |author-link2 = Saul Teukolsky |first3=W. T. |last3=Vetterling |first4=B. P. |last4=Flannery |author-link4 = Brian P. Flannery |title=Numerical Recipes in C: The Art of Scientific Computing |url=https://archive.org/details/numericalrecipes00pres_0 |url-access=registration |edition=2nd |publisher=[[Cambridge University Press]] |location=New York |year=1992 |isbn=0-521-43108-5 }}</ref><ref>{{cite book |first=T. |last=Strutz |title=Data Fitting and Uncertainty: A Practical Introduction to Weighted Least Squares and Beyond |edition=2nd |publisher=Springer Vieweg |year=2016 |isbn=978-3-658-11455-8 }}</ref> Generally, such methods converge in fewer iterations, but the cost of each iteration is higher. An example is the [[Broyden–Fletcher–Goldfarb–Shanno algorithm|BFGS method]] which consists in calculating on every step a matrix by which the gradient vector is multiplied to go into a "better" direction, combined with a more sophisticated [[line search]] algorithm, to find the "best" value of <math>\gamma.</math> For extremely large problems, where the computer-memory issues dominate, a limited-memory method such as [[Limited-memory BFGS|L-BFGS]] should be used instead of BFGS or the steepest descent.  
Methods based on [[Newton's method in optimization|Newton's method]] and inversion of the [[Hessian matrix|Hessian]] using [[conjugate gradient]] techniques can be better alternatives.<ref>{{cite book |first1=W. H. |last1=Press |author-link1 = William H. Press |first2=S. A. |last2=Teukolsky |author-link2 = Saul Teukolsky |first3=W. T. |last3=Vetterling |first4=B. P. |last4=Flannery |author-link4 = Brian P. Flannery |title=Numerical Recipes in C: The Art of Scientific Computing |url=https://archive.org/details/numericalrecipes00pres_0 |url-access=registration |edition=2nd |publisher=[[Cambridge University Press]] |location=New York |year=1992 |isbn=0-521-43108-5 }}</ref><ref>{{cite book |first=T. |last=Strutz |title=Data Fitting and Uncertainty: A Practical Introduction to Weighted Least Squares and Beyond |edition=2nd |publisher=Springer Vieweg |year=2016 |isbn=978-3-658-11455-8 }}</ref> Generally, such methods converge in fewer iterations, but the cost of each iteration is higher. An example is the [[Broyden–Fletcher–Goldfarb–Shanno algorithm|BFGS method]] which consists in calculating on every step a matrix by which the gradient vector is multiplied to go into a "better" direction, combined with a more sophisticated [[line search]] algorithm, to find the "best" value of <math>\eta.</math> For extremely large problems, where the computer-memory issues dominate, a limited-memory method such as [[Limited-memory BFGS|L-BFGS]] should be used instead of BFGS or the steepest descent.  


While it is sometimes possible to substitute gradient descent for a [[Local search (optimization)|local search]] algorithm, gradient descent is not in the same family: although it is an [[iterative method]] for [[Global optimization|local optimization]], it relies on an [[loss function|objective function’s gradient]] rather than an explicit exploration of a [[Feasible region|solution space]].
While it is sometimes possible to substitute gradient descent for a [[Local search (optimization)|local search]] algorithm, gradient descent is not in the same family: although it is an [[iterative method]] for [[Global optimization|local optimization]], it relies on an [[loss function|objective function’s gradient]] rather than an explicit exploration of a [[Feasible region|solution space]].
Line 252: Line 252:


===Fast gradient methods===
===Fast gradient methods===
[[Yurii Nesterov]] has proposed<ref>{{cite book |first=Yurii |last=Nesterov |author-link=Yurii Nesterov |title=Introductory Lectures on Convex Optimization : A Basic Course |publisher=Springer |year=2004 |isbn=1-4020-7553-7 }}</ref> a simple modification that enables faster convergence for convex problems and has been since further generalized. For unconstrained smooth problems, the method is called the [[fast gradient method]] (FGM) or the [[accelerated gradient method]] (AGM). Specifically, if the differentiable function <math>F</math> is convex and <math>\nabla F</math> is [[Lipschitz continuity|Lipschitz]], and it is not assumed that <math>F</math> is [[Convex function#Strongly convex functions|strongly convex]], then the error in the objective value generated at each step <math>k</math> by the gradient descent method will be [[Big O notation|bounded by]] <math display="inline">\mathcal{O}\left({k^{-1}}\right)</math>. Using the Nesterov acceleration technique, the error decreases at <math display="inline">\mathcal{O}\left({k^{-2}}\right)</math>.<ref>{{cite web |url=https://www.seas.ucla.edu/~vandenbe/236C/lectures/fgrad.pdf |title=Fast Gradient Methods |work=Lecture notes for EE236C at UCLA |first=Lieven |last=Vandenberghe |date=2019 }}</ref><ref>{{Cite journal |last=Walkington |first=Noel J. |date=2023 |title=Nesterov's Method for Convex Optimization |url=https://epubs.siam.org/doi/10.1137/21M1390037 |journal=SIAM Review |language=en |volume=65 |issue=2 |pages=539–562 |doi=10.1137/21M1390037 |issn=0036-1445}}</ref> It is known that the rate <math>\mathcal{O}\left({k^{-2}}\right)</math> for the decrease of the [[loss function|cost function]] is optimal for first-order optimization methods. Nevertheless, there is the opportunity to improve the algorithm by reducing the constant factor. The [[optimized gradient method]] (OGM)<ref>{{cite journal |first1=D. |last1=Kim |first2=J. A. |last2=Fessler |title=Optimized First-order Methods for Smooth Convex Minimization |journal=[[Mathematical Programming]] |volume=151 |issue=1–2 |pages=81–107 |year=2016 |doi=10.1007/s10107-015-0949-3 |pmid=27765996 |pmc=5067109 |arxiv=1406.5468 |s2cid=207055414 }}</ref> reduces that constant by a factor of two and is an optimal first-order method for large-scale problems.<ref>{{cite journal |first=Yoel |last=Drori |date=2017 |title=The Exact Information-based Complexity of Smooth Convex Minimization |journal=Journal of Complexity |volume=39 |pages=1–16 |doi=10.1016/j.jco.2016.11.001 |arxiv=1606.01424 |s2cid=205861966 }}</ref>
[[Yurii Nesterov]] has proposed<ref>{{cite book |first=Yurii |last=Nesterov |author-link=Yurii Nesterov |title=Introductory Lectures on Convex Optimization : A Basic Course |publisher=Springer |year=2004 |isbn=1-4020-7553-7 }}</ref> a simple modification that enables faster convergence for convex problems and has been since further generalized. For unconstrained smooth problems, the method is called the [[fast gradient method]] (FGM) or the [[accelerated gradient method]] (AGM). Specifically, if the differentiable function <math>f</math> is convex and <math>\nabla f</math> is [[Lipschitz continuity|Lipschitz]], and it is not assumed that <math>f</math> is [[Convex function#Strongly convex functions|strongly convex]], then the error in the objective value generated at each step <math>k</math> by the gradient descent method will be [[Big O notation|bounded by]] <math display="inline">\mathcal{O}\left({k^{-1}}\right)</math>. Using the Nesterov acceleration technique, the error decreases at <math display="inline">\mathcal{O}\left({k^{-2}}\right)</math>.<ref>{{cite web |url=https://www.seas.ucla.edu/~vandenbe/236C/lectures/fgrad.pdf |title=Fast Gradient Methods |work=Lecture notes for EE236C at UCLA |first=Lieven |last=Vandenberghe |date=2019 }}</ref><ref>{{Cite journal |last=Walkington |first=Noel J. |date=2023 |title=Nesterov's Method for Convex Optimization |url=https://epubs.siam.org/doi/10.1137/21M1390037 |journal=SIAM Review |language=en |volume=65 |issue=2 |pages=539–562 |doi=10.1137/21M1390037 |issn=0036-1445|url-access=subscription }}</ref> It is known that the rate <math>\mathcal{O}\left({k^{-2}}\right)</math> for the decrease of the [[loss function|cost function]] is optimal for first-order optimization methods. Nevertheless, there is the opportunity to improve the algorithm by reducing the constant factor. The [[optimized gradient method]] (OGM)<ref>{{cite journal |first1=D. |last1=Kim |first2=J. A. |last2=Fessler |title=Optimized First-order Methods for Smooth Convex Minimization |journal=[[Mathematical Programming]] |volume=151 |issue=1–2 |pages=81–107 |year=2016 |doi=10.1007/s10107-015-0949-3 |pmid=27765996 |pmc=5067109 |arxiv=1406.5468 |s2cid=207055414 }}</ref> reduces that constant by a factor of two and is an optimal first-order method for large-scale problems.<ref>{{cite journal |first=Yoel |last=Drori |date=2017 |title=The Exact Information-based Complexity of Smooth Convex Minimization |journal=Journal of Complexity |volume=39 |pages=1–16 |doi=10.1016/j.jco.2016.11.001 |arxiv=1606.01424 |s2cid=205861966 }}</ref>


For constrained or non-smooth problems, Nesterov's FGM is called the [[fast proximal gradient method]] (FPGM), an acceleration of the [[proximal gradient method]].
For constrained or non-smooth problems, Nesterov's FGM is called the [[fast proximal gradient method]] (FPGM), an acceleration of the [[proximal gradient method]].

Latest revision as of 14:21, 20 June 2025

Template:Short description Script error: No such module "about". Template:Machine learning

File:Gradient Descent in 2D.webm
Gradient Descent in 2D

Gradient descent is a method for unconstrained mathematical optimization. It is a first-order iterative algorithm for minimizing a differentiable multivariate function.

The idea is to take repeated steps in the opposite direction of the gradient (or approximate gradient) of the function at the current point, because this is the direction of steepest descent. Conversely, stepping in the direction of the gradient will lead to a trajectory that maximizes that function; the procedure is then known as gradient ascent. It is particularly useful in machine learning for minimizing the cost or loss function.[1] Gradient descent should not be confused with local search algorithms, although both are iterative methods for optimization.

Gradient descent is generally attributed to Augustin-Louis Cauchy, who first suggested it in 1847.[2] Jacques Hadamard independently proposed a similar method in 1907.[3][4] Its convergence properties for non-linear optimization problems were first studied by Haskell Curry in 1944,[5] with the method becoming increasingly well-studied and used in the following decades.[6][7]

A simple extension of gradient descent, stochastic gradient descent, serves as the most basic algorithm used for training most deep networks today.

Description

File:Gradient descent.svg
Illustration of gradient descent on a series of level sets

Gradient descent is based on the observation that if the multi-variable function f(𝐱) is defined and differentiable in a neighborhood of a point 𝐚, then f(𝐱) decreases fastest if one goes from 𝐚 in the direction of the negative gradient of f at 𝐚,f(𝐚). It follows that, if

𝐚n+1=𝐚nηf(𝐚n)

for a small enough step size or learning rate η+, then f(a𝐧)f(a𝐧+𝟏). In other words, the term ηf(𝐚) is subtracted from 𝐚 because we want to move against the gradient, toward the local minimum. With this observation in mind, one starts with a guess 𝐱0 for a local minimum of f, and considers the sequence 𝐱0,𝐱1,𝐱2, such that

𝐱n+1=𝐱nηnf(𝐱n), n0.

We have a monotonic sequence

f(𝐱0)f(𝐱1)f(𝐱2),

so the sequence (𝐱n) converges to the desired local minimum. Note that the value of the step size η is allowed to change at every iteration.

It is possible to guarantee the convergence to a local minimum under certain assumptions on the function f (for example, f convex and f Lipschitz) and particular choices of η. Those include the sequence

ηn=|(𝐱n𝐱n1)[f(𝐱n)f(𝐱n1)]|f(𝐱n)f(𝐱n1)2

as in the Barzilai-Borwein method,[8][9] or a sequence ηn satisfying the Wolfe conditions (which can be found by using line search). When the function f is convex, all local minima are also global minima, so in this case gradient descent can converge to the global solution.

This process is illustrated in the adjacent picture. Here, f is assumed to be defined on the plane, and that its graph has a bowl shape. The blue curves are the contour lines, that is, the regions on which the value of f is constant. A red arrow originating at a point shows the direction of the negative gradient at that point. Note that the (negative) gradient at a point is orthogonal to the contour line going through that point. We see that gradient descent leads us to the bottom of the bowl, that is, to the point where the value of the function f is minimal.

An analogy for understanding gradient descent

File:Okanogan-Wenatchee National Forest, morning fog shrouds trees (37171636495).jpg
Fog in the mountains

The basic intuition behind gradient descent can be illustrated by a hypothetical scenario. People are stuck in the mountains and are trying to get down (i.e., trying to find the global minimum). There is heavy fog such that visibility is extremely low. Therefore, the path down the mountain is not visible, so they must use local information to find the minimum. They can use the method of gradient descent, which involves looking at the steepness of the hill at their current position, then proceeding in the direction with the steepest descent (i.e., downhill). If they were trying to find the top of the mountain (i.e., the maximum), then they would proceed in the direction of steepest ascent (i.e., uphill). Using this method, they would eventually find their way down the mountain or possibly get stuck in some hole (i.e., local minimum or saddle point), like a mountain lake. However, assume also that the steepness of the hill is not immediately obvious with simple observation, but rather it requires a sophisticated instrument to measure, which the persons happen to have at the moment. It takes quite some time to measure the steepness of the hill with the instrument, thus they should minimize their use of the instrument if they wanted to get down the mountain before sunset. The difficulty then is choosing the frequency at which they should measure the steepness of the hill so not to go off track.

In this analogy, the persons represent the algorithm, and the path taken down the mountain represents the sequence of parameter settings that the algorithm will explore. The steepness of the hill represents the slope of the function at that point. The instrument used to measure steepness is differentiation. The direction they choose to travel in aligns with the gradient of the function at that point. The amount of time they travel before taking another measurement is the step size.

Choosing the step size and descent direction

Since using a step size η that is too small would slow convergence, and a η too large would lead to overshoot and divergence, finding a good setting of η is an important practical problem. Philip Wolfe also advocated using "clever choices of the [descent] direction" in practice.[10] While using a direction that deviates from the steepest descent direction may seem counter-intuitive, the idea is that the smaller slope may be compensated for by being sustained over a much longer distance.

To reason about this mathematically, consider a direction 𝐩n and step size ηn and consider the more general update:

𝐚n+1=𝐚nηn𝐩n.

Finding good settings of 𝐩n and ηn requires some thought. First of all, we would like the update direction to point downhill. Mathematically, letting θn denote the angle between f(a𝐧) and 𝐩n, this requires that cosθn>0. To say more, we need more information about the objective function that we are optimising. Under the fairly weak assumption that f is continuously differentiable, we may prove that:[11] Template:NumBlk This inequality implies that the amount by which we can be sure the function f is decreased depends on a trade off between the two terms in square brackets. The first term in square brackets measures the angle between the descent direction and the negative gradient. The second term measures how quickly the gradient changes along the descent direction.

In principle inequality (Template:EquationNote) could be optimized over 𝐩n and ηn to choose an optimal step size and direction. The problem is that evaluating the second term in square brackets requires evaluating f(𝐚ntηn𝐩n), and extra gradient evaluations are generally expensive and undesirable. Some ways around this problem are:

  • Forgo the benefits of a clever descent direction by setting 𝐩n=f(a𝐧), and use line search to find a suitable step-size γn, such as one that satisfies the Wolfe conditions. A more economic way of choosing learning rates is backtracking line search, a method that has both good theoretical guarantees and experimental results. Note that one does not need to choose 𝐩n to be the gradient; any direction that has positive inner product with the gradient will result in a reduction of the function value (for a sufficiently small value of ηn).
  • Assuming that f is twice-differentiable, use its Hessian 2f to estimate f(𝐚ntηn𝐩n)f(𝐚n)2tηn2f(𝐚n)𝐩n.Then choose 𝐩n and ηn by optimising inequality (Template:EquationNote).
  • Assuming that f is Lipschitz, use its Lipschitz constant L to bound f(𝐚ntηn𝐩n)f(𝐚n)2Ltηn𝐩n. Then choose 𝐩n and ηn by optimising inequality (Template:EquationNote).
  • Build a custom model of maxt[0,1]f(𝐚ntηn𝐩n)f(𝐚n)2f(𝐚n)2 for f. Then choose 𝐩n and ηn by optimising inequality (Template:EquationNote).
  • Under stronger assumptions on the function f such as convexity, more advanced techniques may be possible.

Usually by following one of the recipes above, convergence to a local minimum can be guaranteed. When the function f is convex, all local minima are also global minima, so in this case gradient descent can converge to the global solution.

Solution of a linear system

File:Steepest descent.png
The steepest descent algorithm applied to the Wiener filter[12]

Gradient descent can be used to solve a system of linear equations

𝐀𝐱𝐛=0

reformulated as a quadratic minimization problem. If the system matrix 𝐀 is real symmetric and positive-definite, an objective function is defined as the quadratic function, with minimization of

f(𝐱)=𝐱𝐀𝐱2𝐱𝐛,

so that

f(𝐱)=2(𝐀𝐱𝐛).

For a general real matrix 𝐀, linear least squares define

f(𝐱)=𝐀𝐱𝐛2.

In traditional linear least squares for real 𝐀 and 𝐛 the Euclidean norm is used, in which case

f(𝐱)=2𝐀(𝐀𝐱𝐛).

The line search minimization, finding the locally optimal step size η on every iteration, can be performed analytically for quadratic functions, and explicit formulas for the locally optimal η are known.[6][13]

For example, for real symmetric and positive-definite matrix 𝐀, a simple algorithm can be as follows,[6]

repeat in the loop:𝐫:=𝐛𝐀𝐱η:=𝐫𝐫/𝐫𝐀𝐫𝐱:=𝐱+η𝐫if 𝐫𝐫 is sufficiently small, then exit loopend repeat loopreturn 𝐱 as the result

To avoid multiplying by 𝐀 twice per iteration, we note that 𝐱:=𝐱+η𝐫 implies 𝐫:=𝐫η𝐀𝐫, which gives the traditional algorithm,[14]

𝐫:=𝐛𝐀𝐱repeat in the loop:η:=𝐫𝐫/𝐫𝐀𝐫𝐱:=𝐱+η𝐫if 𝐫𝐫 is sufficiently small, then exit loop𝐫:=𝐫η𝐀𝐫end repeat loopreturn 𝐱 as the result

The method is rarely used for solving linear equations, with the conjugate gradient method being one of the most popular alternatives. The number of gradient descent iterations is commonly proportional to the spectral condition number κ(𝐀) of the system matrix 𝐀 (the ratio of the maximum to minimum eigenvalues of 𝐀𝐀), while the convergence of conjugate gradient method is typically determined by a square root of the condition number, i.e., is much faster. Both methods can benefit from preconditioning, where gradient descent may require less assumptions on the preconditioner.[14]

Geometric behavior and residual orthogonality

In steepest descent applied to solving 𝐀𝐱=𝐛, where 𝐀 is symmetric positive-definite, the residual vectors 𝐫k=𝐛𝐀𝐱k are orthogonal across iterations:

𝐫k+1,𝐫k=0.

Because each step is taken in the steepest direction, steepest-descent steps alternate between directions aligned with the extreme axes of the elongated level sets. When κ(𝐀) is large, this produces a characteristic zig-zag path. The poor conditioning of 𝐀 is the primary cause of the slow convergence, and orthogonality of successive residuals reinforces this alternation.

File:Steepest descent convergence path for A = 2 2, 2 3.png
Convergence path of steepest descent method for A = [[2, 2], [2, 3]]

As shown in the image on the right, steepest descent converges slowly due to the high condition number of 𝐀, and the orthogonality of residuals forces each new direction to undo the overshoot from the previous step. The result is a path that zigzags toward the solution. This inefficiency is one reason conjugate gradient or preconditioning methods are preferred.[15]

Solution of a non-linear system

Gradient descent can also be used to solve a system of nonlinear equations. Below is an example that shows how to use the gradient descent to solve for three unknown variables, x1, x2, and x3. This example shows one iteration of the gradient descent.

Consider the nonlinear system of equations

{3x1cos(x2x3)32=04x12625x22+2x21=0exp(x1x2)+20x3+10π33=0

Let us introduce the associated function

G(𝐱)=[3x1cos(x2x3)324x12625x22+2x21exp(x1x2)+20x3+10π33],

where

𝐱=[x1x2x3].

One might now define the objective function

f(𝐱)=12G(𝐱)G(𝐱)=12[(3x1cos(x2x3)32)2+(4x12625x22+2x21)2+(exp(x1x2)+20x3+10π33)2],

which we will attempt to minimize. As an initial guess, let us use

𝐱(0)=𝟎=[000].

We know that

𝐱(1)=𝟎η0f(𝟎)=𝟎η0JG(𝟎)G(𝟎),

where the Jacobian matrix JG is given by

JG(𝐱)=[3sin(x2x3)x3sin(x2x3)x28x11250x2+20x2exp(x1x2)x1exp(x1x2)20].

We calculate:

JG(𝟎)=[3000200020],G(𝟎)=[2.5110.472].

Thus

𝐱(1)=𝟎η0[7.52209.44],

and

f(𝟎)=0.5((2.5)2+(1)2+(10.472)2)=58.456.
File:Gradient Descent Example Nonlinear Equations.gif
An animation showing the first 83 iterations of gradient descent applied to this example. Surfaces are isosurfaces of f(𝐱(n)) at current guess 𝐱(n), and arrows show the direction of descent. Due to a small and constant step size, the convergence is slow.

Now, a suitable η0 must be found such that

f(𝐱(1))f(𝐱(0))=f(𝟎).

This can be done with any of a variety of line search algorithms. One might also simply guess η0=0.001, which gives

𝐱(1)=[0.00750.0020.20944].

Evaluating the objective function at this value, yields

f(𝐱(1))=0.5((2.48)2+(1.00)2+(6.28)2)=23.306.

The decrease from f(𝟎)=58.456 to the next step's value of

f(𝐱(1))=23.306

is a sizable decrease in the objective function. Further steps would reduce its value further until an approximate solution to the system was found.

Comments

Gradient descent works in spaces of any number of dimensions, even in infinite-dimensional ones. In the latter case, the search space is typically a function space, and one calculates the Fréchet derivative of the functional to be minimized to determine the descent direction.[7]

That gradient descent works in any number of dimensions (finite number at least) can be seen as a consequence of the Cauchy-Schwarz inequality, i.e. the magnitude of the inner (dot) product of two vectors of any dimension is maximized when they are colinear. In the case of gradient descent, that would be when the vector of independent variable adjustments is proportional to the gradient vector of partial derivatives.

The gradient descent can take many iterations to compute a local minimum with a required accuracy, if the curvature in different directions is very different for the given function. For such functions, preconditioning, which changes the geometry of the space to shape the function level sets like concentric circles, cures the slow convergence. Constructing and applying preconditioning can be computationally expensive, however.

The gradient descent can be modified via momentums[16] (Nesterov, Polyak,[17] and Frank-Wolfe[18]) and heavy-ball parameters (exponential moving averages[19] and positive-negative momentum[20]). The main examples of such optimizers are Adam, DiffGrad, Yogi, AdaBelief, etc.

Methods based on Newton's method and inversion of the Hessian using conjugate gradient techniques can be better alternatives.[21][22] Generally, such methods converge in fewer iterations, but the cost of each iteration is higher. An example is the BFGS method which consists in calculating on every step a matrix by which the gradient vector is multiplied to go into a "better" direction, combined with a more sophisticated line search algorithm, to find the "best" value of η. For extremely large problems, where the computer-memory issues dominate, a limited-memory method such as L-BFGS should be used instead of BFGS or the steepest descent.

While it is sometimes possible to substitute gradient descent for a local search algorithm, gradient descent is not in the same family: although it is an iterative method for local optimization, it relies on an objective function’s gradient rather than an explicit exploration of a solution space.

Gradient descent can be viewed as applying Euler's method for solving ordinary differential equations x(t)=f(x(t)) to a gradient flow. In turn, this equation may be derived as an optimal controller[23] for the control system x(t)=u(t) with u(t) given in feedback form u(t)=f(x(t)).


Modifications

Gradient descent can converge to a local minimum and slow down in a neighborhood of a saddle point. Even for unconstrained quadratic minimization, gradient descent develops a zig-zag pattern of subsequent iterates as iterations progress, resulting in slow convergence. Multiple modifications of gradient descent have been proposed to address these deficiencies.

Fast gradient methods

Yurii Nesterov has proposed[24] a simple modification that enables faster convergence for convex problems and has been since further generalized. For unconstrained smooth problems, the method is called the fast gradient method (FGM) or the accelerated gradient method (AGM). Specifically, if the differentiable function f is convex and f is Lipschitz, and it is not assumed that f is strongly convex, then the error in the objective value generated at each step k by the gradient descent method will be bounded by 𝒪(k1). Using the Nesterov acceleration technique, the error decreases at 𝒪(k2).[25][26] It is known that the rate 𝒪(k2) for the decrease of the cost function is optimal for first-order optimization methods. Nevertheless, there is the opportunity to improve the algorithm by reducing the constant factor. The optimized gradient method (OGM)[27] reduces that constant by a factor of two and is an optimal first-order method for large-scale problems.[28]

For constrained or non-smooth problems, Nesterov's FGM is called the fast proximal gradient method (FPGM), an acceleration of the proximal gradient method.

Momentum or heavy ball method

Trying to break the zig-zag pattern of gradient descent, the momentum or heavy ball method uses a momentum term in analogy to a heavy ball sliding on the surface of values of the function being minimized,[6] or to mass movement in Newtonian dynamics through a viscous medium in a conservative force field.[29] Gradient descent with momentum remembers the solution update at each iteration, and determines the next update as a linear combination of the gradient and the previous update. For unconstrained quadratic minimization, a theoretical convergence rate bound of the heavy ball method is asymptotically the same as that for the optimal conjugate gradient method.[6]

This technique is used in stochastic gradient descent and as an extension to the backpropagation algorithms used to train artificial neural networks.[30][31] In the direction of updating, stochastic gradient descent adds a stochastic property. The weights can be used to calculate the derivatives.

Extensions

Gradient descent can be extended to handle constraints by including a projection onto the set of constraints. This method is only feasible when the projection is efficiently computable on a computer. Under suitable assumptions, this method converges. This method is a specific case of the forward-backward algorithm for monotone inclusions (which includes convex programming and variational inequalities).[32]

Gradient descent is a special case of mirror descent using the squared Euclidean distance as the given Bregman divergence.[33]

Theoretical properties

The properties of gradient descent depend on the properties of the objective function and the variant of gradient descent used (for example, if a line search step is used). The assumptions made affect the convergence rate, and other properties, that can be proven for gradient descent.[34] For example, if the objective is assumed to be strongly convex and lipschitz smooth, then gradient descent converges linearly with a fixed step size.[1] Looser assumptions lead to either weaker convergence guarantees or require a more sophisticated step size selection.[34]

See also

Template:Div col

Template:Div col end

References

Template:Reflist

Further reading

  • Script error: No such module "citation/CS1".
  • Script error: No such module "citation/CS1".
  • Script error: No such module "citation/CS1".

External links

Template:Sister project

Template:Artificial intelligence navbox Template:Optimization algorithms

  1. a b Script error: No such module "citation/CS1".
  2. Script error: No such module "citation/CS1".
  3. Script error: No such module "Citation/CS1".
  4. Script error: No such module "Citation/CS1".
  5. Script error: No such module "Citation/CS1".
  6. a b c d e Script error: No such module "citation/CS1".
  7. a b Script error: No such module "citation/CS1".
  8. Script error: No such module "Citation/CS1".
  9. Script error: No such module "citation/CS1".
  10. Script error: No such module "Citation/CS1".
  11. Script error: No such module "citation/CS1".
  12. Haykin, Simon S. Adaptive filter theory. Pearson Education India, 2008. - p. 108-142, 217-242
  13. Script error: No such module "citation/CS1".
  14. a b Script error: No such module "Citation/CS1".
  15. Script error: No such module "citation/CS1".
  16. Script error: No such module "Citation/CS1".
  17. Script error: No such module "Citation/CS1".
  18. Script error: No such module "Citation/CS1".
  19. Script error: No such module "citation/CS1".
  20. Script error: No such module "Citation/CS1".
  21. Script error: No such module "citation/CS1".
  22. Script error: No such module "citation/CS1".
  23. Script error: No such module "Citation/CS1".
  24. Script error: No such module "citation/CS1".
  25. Script error: No such module "citation/CS1".
  26. Script error: No such module "Citation/CS1".
  27. Script error: No such module "Citation/CS1".
  28. Script error: No such module "Citation/CS1".
  29. Script error: No such module "Citation/CS1".
  30. Script error: No such module "citation/CS1".
  31. Script error: No such module "citation/CS1". Part of a lecture series for the Coursera online course Neural Networks for Machine Learning Template:Webarchive.
  32. Script error: No such module "citation/CS1".
  33. Script error: No such module "citation/CS1".
  34. a b Script error: No such module "citation/CS1".