English

Backtracking Gradient Descent allowing unbounded learning rates

Optimization and Control 2020-01-09 v2 Machine Learning Machine Learning

Abstract

In unconstrained optimisation on an Euclidean space, to prove convergence in Gradient Descent processes (GD) xn+1=xnδnf(xn)x_{n+1}=x_n-\delta _n \nabla f(x_n) it usually is required that the learning rates δn\delta _n's are bounded: δnδ\delta _n\leq \delta for some positive δ\delta . Under this assumption, if the sequence xnx_n converges to a critical point zz, then with large values of nn the update will be small because xn+1xnf(xn)||x_{n+1}-x_n||\lesssim ||\nabla f(x_n)||. This may also force the sequence to converge to a bad minimum. If we can allow, at least theoretically, that the learning rates δn\delta _n's are not bounded, then we may have better convergence to better minima. A previous joint paper by the author showed convergence for the usual version of Backtracking GD under very general assumptions on the cost function ff. In this paper, we allow the learning rates δn\delta _n to be unbounded, in the sense that there is a function h:(0,)(0,)h:(0,\infty)\rightarrow (0,\infty ) such that limt0th(t)=0\lim _{t\rightarrow 0}th(t)=0 and δnmax{h(xn),δ}\delta _n\lesssim \max \{h(x_n),\delta \} satisfies Armijo's condition for all nn, and prove convergence under the same assumptions as in the mentioned paper. It will be shown that this growth rate of hh is best possible if one wants convergence of the sequence {xn}\{x_n\}. A specific way for choosing δn\delta _n in a discrete way connects to Two-way Backtracking GD defined in the mentioned paper. We provide some results which either improve or are implicitly contained in those in the mentioned paper and another recent paper on avoidance of saddle points.

Keywords

Cite

@article{arxiv.2001.02005,
  title  = {Backtracking Gradient Descent allowing unbounded learning rates},
  author = {Tuyen Trung Truong},
  journal= {arXiv preprint arXiv:2001.02005},
  year   = {2020}
}

Comments

Convergence for Two-way Backtracking GD can be proven under more general assumptions, in particular valid for C^2 functions. In statement of Theorem 0.3, need to add the assumption that {f(x_n}) is non-increasing. Some typos corrected. 5 pages

R2 v1 2026-06-23T13:04:52.940Z