Barzilai-Borwein method

From HandWiki
Short description: Mathematical optimization method

The Barzilai-Borwein method[1] is an iterative gradient descent method for unconstrained optimization using either of two step sizes derived from the linear trend of the most recent two iterates.  This method, and modifications, are globally convergent under mild conditions,[2][3] and perform competitively with conjugate gradient methods for many problems.[4] Not depending on the objective itself, it can also solve some systems of linear and non-linear equations.

Method

To minimize a convex function [math]\displaystyle{ f:\mathbb{R}^n\rightarrow\mathbb{R} }[/math] with gradient vector [math]\displaystyle{ g }[/math] at point [math]\displaystyle{ x }[/math], let there be two prior iterates, [math]\displaystyle{ g_{k-1}(x_{k-1}) }[/math] and [math]\displaystyle{ g_{k}(x_{k}) }[/math], in which [math]\displaystyle{ x_{k}=x_{k-1}-\alpha_{k-1} g_{k-1} }[/math] where [math]\displaystyle{ \alpha_{k-1} }[/math] is the previous iteration's step size (not necessarily a Barzilai-Borwein step size), and for brevity, let [math]\displaystyle{ \Delta x=x_k-x_{k-1} }[/math] and [math]\displaystyle{ \Delta g=g_k-g_{k-1} }[/math].

A Barzilai-Borwein (BB) iteration is [math]\displaystyle{ x_{k+1}=x_k-\alpha _kg_k }[/math] where the step size [math]\displaystyle{ \alpha _k }[/math] is either

[long BB step] [math]\displaystyle{ \alpha _k^{LONG}=\frac{\Delta x\cdot\Delta x}{\Delta x\cdot\Delta g} }[/math], or

[short BB step] [math]\displaystyle{ \alpha _k^{SHORT}=\frac{\Delta x \cdot\Delta g}{\Delta g \cdot\Delta g} }[/math].

Barzilai-Borwein also applies to systems of equations [math]\displaystyle{ g(x)=0 }[/math] for [math]\displaystyle{ g:\mathbb{R}^n\rightarrow \mathbb{R}^n }[/math] in which the Jacobian of [math]\displaystyle{ g }[/math] is positive-definite in the symmetric part, that is, [math]\displaystyle{ \Delta x \cdot\Delta g }[/math] is necessarily positive.

Derivation

Despite its simplicity and optimality properties, Cauchy's classical steepest-descent method[5] for unconstrained optimization often performs poorly.[6] This has motivated many to propose alternate search directions, such as the conjugate gradient method. Jonathan Barzilai and Jonathan Borwein instead proposed new step sizes for the gradient by approximating the quasi-Newton method, creating a scalar approximation of the Hessian estimated from the finite differences between two evaluation points of the gradient, these being the most recent two iterates.

In a quasi-Newton iteration,

[math]\displaystyle{ x_{k+1}=x_k-B^{-1}g(x_k) }[/math]

where [math]\displaystyle{ B }[/math] is some approximation of the Jacobian matrix of [math]\displaystyle{ g }[/math] (i.e. Hessian of the objective function) which satisfies the secant equation [math]\displaystyle{ B_k \Delta x_k = \Delta g_k }[/math]. Barzilai and Borwein simplify [math]\displaystyle{ B }[/math] with a scalar [math]\displaystyle{ 1/\alpha }[/math], which usually cannot exactly satisfy the secant equation, but approximate it as [math]\displaystyle{ \frac{1}{\alpha}\Delta x\approx \Delta g }[/math]. Approximations by two least-squares criteria are:

[1] Minimize [math]\displaystyle{ \|\Delta x/\alpha-\Delta g\|^2 }[/math] with respect to [math]\displaystyle{ \alpha }[/math], yielding the long BB step, or

[2] Minimize [math]\displaystyle{ \|\Delta x-\alpha\Delta g\|^2 }[/math] with respect to [math]\displaystyle{ \alpha }[/math], yielding the short BB step.

Properties

In one dimension, both BB step sizes are equal and same as the classical secant method.

The long BB step size is the same as a linearized Cauchy step, i.e. the first estimate using a secant-method for the line search (also, for linear problems).  The short BB step size is same as a linearized minimum-residual step.  BB applies the step sizes upon the forward direction vector for the next iterate, instead of the prior direction vector as if for another line-search step.

Barzilai and Borwein proved their method converges R-superlinearly for quadratic minimization in two dimensions. Raydan[2] demonstrates convergence in general for quadratic problems. Convergence is usually non-monotone, that is, neither the objective function nor the residual or gradient magnitude necessarily decrease with each iteration along a successful convergence toward the solution.

If [math]\displaystyle{ f }[/math] is a quadratic function with Hessian [math]\displaystyle{ A }[/math], [math]\displaystyle{ 1/\alpha^{LONG} }[/math] is the Rayleigh quotient of [math]\displaystyle{ A }[/math] by vector [math]\displaystyle{ \Delta x }[/math], and [math]\displaystyle{ 1/\alpha^{SHORT} }[/math] is the Rayleigh quotient of [math]\displaystyle{ A }[/math] by vector [math]\displaystyle{ \sqrt{A}\Delta x }[/math] (here taking [math]\displaystyle{ \sqrt A }[/math] as a solution to [math]\displaystyle{ (\sqrt{A})^T\sqrt{A}=A }[/math], more at Definite matrix).

Fletcher[4] compared its computational performance to conjugate gradient (CG) methods, finding CG tending faster for linear problems, but BB often faster for non-linear problems versus applicable CG-based methods.

BB has low storage requirements, suitable for large systems with millions of elements in [math]\displaystyle{ x }[/math].

[math]\displaystyle{ \frac{\alpha^{SHORT}}{\alpha^{LONG}}=cos^2( }[/math]angle between [math]\displaystyle{ \Delta x }[/math] and [math]\displaystyle{ \Delta g) }[/math].

Modifications and related methods

Since being demonstrated by Raydan,[3] BB is often applied with the non-monotone safeguarding strategy of Grippo, Lampariello, and Lucidi.[7] This tolerates some rise of the objective, but excessive rise initiates a backtracking line search using smaller step sizes, to assure global convergence. Fletcher[4] finds that allowing wider limits for non-monotonicity tend to result in more efficient convergence.

Others[8][9][10][11] have identified a step size being the geometric mean between the long and short BB step sizes, which exhibits similar properties.

References

  1. Barzilai, Jonathan; Borwein, Jonathan M. (1988). "Two-Point Step Size Gradient Methods". IMA Journal of Numerical Analysis 8: 141–148. doi:10.1093/imanum/8.1.141. 
  2. 2.0 2.1 Raydan, Marcos (1993). "On the Barzilai and Borwein choice of steplength for the gradient method". IMA Journal of Numerical Analysis 13 (3): 321–326. doi:10.1093/imanum/13.3.321. 
  3. 3.0 3.1 Raydan, M. The Barzilai and Borwein gradient method for the large scale unconstrained minimization problem. SIAM Journal of Optimization 7, pp 26-33. 1997
  4. 4.0 4.1 4.2 Fletcher, R. (2005). "On the Barzilai–Borwein Method". In Qi, L.; Teo, K.; Yang, X. (eds.). Optimization and Control with Applications. Applied Optimization. Vol. 96. Boston: Springer. pp. 235–256. ISBN 0-387-24254-6
  5. A. Cauchy. Méthode générale pour la résolution des systèmes d’équations simultanées. C. R. Acad. Sci. Paris, 25:536–538, 1847.
  6. H. Akaike, On a successive transformation of probability distribution and its application to the analysis of the optimum gradient method, Ann. Inst. Statist. Math Tokyo, 11 (1959), pp. 1–17
  7. L. Grippo, F. Lampariello, and S. Lucidi, “A nonmonotone line search technique for Newton’s method,” SIAM J. Numer. Anal., vol. 23, pp. 707–716, 1986
  8. Varadhan R, Roland C (2008). Simple and Globally Convergent Methods for Accelerating the Convergence of Any EM Algorithm. Scandinavian Journal of Statistics, 35(2), 335-353.
  9. Y. H. Dai, M. Al-Baali, and X. Yang, “A positive Barzilai-Borwein-like stepsize and an extension for symmetric linear systems,” in Numerical Analysis and Optimization. Cham, Switzerland: Springer, 2015, pp. 59-75.
  10. Dai, Yu-Hong; Huang, Yakui; Liu, Xin-Wei (2018). "A family of spectral gradient methods for optimization". arXiv:1812.02974 [math.OC].
  11. Shuai Huang, Zhong Wan, A new nonmonotone spectral residual method for nonsmooth nonlinear equations, Journal of Computational and Applied Mathematics 313, pp 82-101, Elsevier, 2017

External links