Line search methods

Authors: Lihe Cao, Zhengyi Sui, Jiaqi Zhang, Yuqing Yan, and Yuhui Gu (6800 Fall 2021).

Introduction

Line search method is an iterative approach to find a local minimum of a multidimensional nonlinear function using the function's gradients. It aims at finding an acceptable step length witch satisfies certain standard conditions. Line search and trust-region method are two fundamental strategies for locating the new iterate given the current point. With the ability to solve the unconstrained optimization problem, line search is widely used in many cases including machine learning, game theory and other fields.

Generic Line Search Method

Basic Algorithm

• Pick the starting point $x_{0}$
• Repeat the following steps until $f_{k}:=f(x_{k})$ coverges to a local minimum :
  • Choose a descent direction $p_{k}$ starting at $x_{k}$, defined as: $\nabla f_{k}^{\top }p_{k}<0$ for $\nabla f_{k}\not =0$
  • Find a step length $\alpha _{k}>0$ so that $f(x_{k}+\alpha _{k}p_{k})<f_{k}$
  • Set $x_{k+1}=x_{k}+\alpha _{k}p_{k}$

Search Direction for Line Search

The direction of the line search should be chosen to make $f$ decrease moving from point $x_{k}$ to $x_{k+1}$, and it is usually related to the gradient $\nabla f_{k}$. The most obvious direction is the $-\nabla f_{k}$ because it is the one to make $f$ decreases most rapidly. This claim can be verified by Taylor's theorem:

$f(x_{k}+\alpha )=f(x_{k})+\alpha p^{\top }\nabla f_{k}+{\frac {1}{2}}\alpha ^{2}p^{\top }f(x_{k}+tp)p$ where $t\in (0,\alpha )$

The rate of change in $f$ along the direction $p$ at $x_{k}$ is the coefficient of $\alpha$. Therefore, the unit direction $p$ of most rapid decrease is the solution to

$\min \ p^{\top }\nabla f_{k}$

${\text{s.t.}}\ \ ||p||=1$.

$p={\frac {-\nabla f_{k}}{||\nabla f_{k}||}}$ is the solution and this direction is orthogonal to the contours of the function. In the following sections, this will be used as the default direction of the line search.

However, the steepest descent direction is not the most effient, as the steepest descent method does not pass the Rosenbrock test (see Figure 1). Carefully desgined descent directions deviating from the steepest direction can be used in practice to produce faster convergence.

Step Length

The step length is a non-negative value such that $f(x_{k}+\alpha _{k}p_{k})<f(x_{k})$. When choosing the step length $\alpha _{k}$, there is a trade off between giving a substantial reduction of $f$ and not spending too much time finding the solution. If $\alpha _{k}$ is too large, then the step will overshoot, while if the step length is too small, it is time consuming to find the convergent point. An exact line search and inexact line search are needed to find the value of $\alpha$ and more detail about these approaches will be introduced in the next section.

Convergence

For a line search algorithm to be reliable, it should be globally convergent, that is the gradient norms, $||\nabla f(x_{k})||$, should converge to zero with each iteration, i.e., $\lim _{k\to \infty }||\nabla f(x_{k})||=0$.

It can be shown from Zoutendijk's theorem that if the line search algorithm satisfies (weak) Wolfe's conditions (similar results also hold for strong Wolfe and Goldstein conditions) and has a search direction that makes an angle with the steepest descent direction that is bounded away from 90°, the algorithm is globally convergent.

Zoutendijk's theorem states that, given an iteration where $p_{k}$ is the descent direction and $\alpha _{k}$ is the step length that satisfies (weak) Wolfe conditions, if the objective $f$ is bounded below on $\mathbb {R} ^{n}$ and is continuously differentiable in an open set ${\mathcal {N}}$ containing the level set ${\mathcal {L}}:=\{x\ |\ f(x)\leq f(x_{0})\}$, where $x_{0}$ is the starting point of the iteration, and the gradient $\nabla f$ is Lipschitz continuous on ${\mathcal {N}}$, then

$\sum _{k=0}^{\infty }\cos ^{2}\theta _{k}||\nabla f_{k}||^{2}<\infty$,

where $\theta _{k}$ is the angle between $p_{k}$ and the steepest descent direction $-\nabla f(x_{k})$.

The Zoutendijk condition above implies that

$\lim _{k\to \infty }\cos ^{2}\theta _{k}||\nabla f_{k}||^{2}=0$,

by the n-th term divergence test. Hence, if the algorithm chooses a search direction that is bounded away from 90° relative to the gradient, i.e., given $\epsilon >0$,

$\cos \theta _{k}\geq \epsilon >0,\ \forall k$,

it follows that

$\lim _{k\to \infty }||\nabla f_{k}||=0$.

However, the Zoutendijk condition doesn't guarantee convergence to a local minimum but only stationary points. Hence, additional conditions on the search direction is necessary, such as finding a direction of negative curvature whenever possible, to avoid converging to a nonminimizing stationary point.

Exact Search

Steepest Descent Method

Given the intuition that the negative gradient $-\nabla f_{k}$ can be an effective search direction, steepest descent follows the idea and establishes a systematic method for minimizing the objective function. Setting $-\nabla f_{k}$ as the direction, steepest descent computes the step-length $\alpha _{k}$ by minimizing a single-variable objective function. More specifically, the steps of Steepest Descent Method are as follows:

Steepest Descent Algorithm

Set a starting point $x_{0}$

Set a convergence criterium $\epsilon >0$

Set $k=0$

Set the maximum iteration $N$

While $k\leq N$:

$\nabla f(x_{k})=\left.{\frac {\partial f(x)}{\partial x}}\right\vert _{x=x_{k}}$

If $\nabla f(x_{k})\leq \epsilon$:

Break

End if

$\alpha _{k}={\underset {\alpha }{\arg \min }}f(x_{k}-\alpha \nabla f(x_{k}))$

$x_{k+1}=x_{k}-\alpha _{k}\nabla f(x_{k})$

$k=k+1$

End while

Return $x_{k}$, $f(x_{k})$

Figure 1. Steepest descent does not produce convergence on the Rosenbrock function.

One advantage of the steepest descent method is the convergency. For a steepest descent method, it converges to a local minimum from any starting point.

Theorem: Global Convergence of Steepest Descent

Let the gradient of $f\in C^{1}$ be uniformly Lipschitz continuous on $\mathbb {R} ^{n}$. Then, for the iterates with steepest-descent search directions, one of the following situations occurs:

• $\nabla f(x_{k})=0$ for some finite $k$

• $\lim _{k\to \infty }f(x_{k})=-\infty$

• $\lim _{k\to \infty }\nabla f(x_{k})=0$

However, steepest descent has disadvantages in that the convergence is always slow and numerically may be not convergent. (See Figure 1.)

Steepest descent method is a special case of gradient descent in that the step-length is analytically defined. However, step-lengths cannot always be computed analytically; in this case, inexact methods can be used to optimize $\alpha$ at each iteration.

Inexact Search

While minimizing an objective function using numeric methods, in each iteration, the updated objective is $\phi (\alpha )=f(x_{k}+\alpha p_{k})$, which is a function of $\alpha$ after fixing the search direction. The goal is to minimize this objective with respect to $\alpha$. However, if one wants to solve for the exact minimum in each iteration, it could be computationally expensive and the algorithm will be time-consuming. Therefore, in practice, it is easier solve the subproblem

${\underset {\alpha }{min}}\quad \phi (\alpha )=f(x_{k}+\alpha p_{k})$

numerically and find a reasonable step length $\alpha$ instead, which will decrease the objective function. That is, $\alpha$ satisfies $f(x_{k}+\alpha p_{k})\leq f(x_{k})$. However, convergence to the function's minimum cannot be guaranteed, so Wolfe or Goldstein conditions need to be applied when searching for an acceptable step length.

Wolfe Conditions

This condition is proposed by Phillip Wolfe in 1969 . It provide an efficient way of choosing a step length that decreases the objective function sufficiently. It consists of two conditions: Armijo (sufficient decrease) condition and the curvature condition.

Figure 2. Armijo Condition.

Armijo (Sufficient Decrease) Condition

$f(x_{k}+\alpha p_{k})\leq f(x_{k})+c_{1}\alpha _{k}p_{k}^{\top }\nabla {f(x_{k})}$,

where $c_{1}\in (0,1)$ and is often chosen to be of a small order of magnitude around $10^{-4}$. This condition ensures the computed step length can reduces the objective function $f(x_{k})$ sufficiently. Only using this condition, however, it cannot be guaranteed that $x_{k}$ will converge in a reasonable number of iterations, since Armijo condition is always satisfied with step length being small enough. Therefore, the second condition below needs to be paired with the sufficient decrease condition to keep $\alpha _{k}$ from being too short.

Figure 3. Curvature Condition.

Curvature Condition

$\nabla {f(x_{k}+\alpha p_{k})}^{\top }p_{k}\geq c_{2}\nabla {f(x_{k})}^{\top }p_{k}$,

where $c_{2}\in (c_{1},1)$ is much greater than $c_{1}$ and is typically on the order of 0.1. This condition ensures a sufficient increase of the gradient.

This left hand side of the curvature condition is simply the derivative of $\phi (\alpha )$, thus ensuring $\alpha _{k}$ to be in the vicinity of a stationary point of $\phi (\alpha )$.

Figure 4. Goldstein Condition.

Strong Wolfe curvature condition

The (weak) Wolfe conditions can result in an $\alpha$ value that is not close to the minimizer of $\phi (\alpha )$. The (weak) Wolfe conditions can be modified by using the following condition called Strong Wolfe condition, which writes the curvature condition in absolute values

$|p_{k}\nabla {f(x_{k}+\alpha p_{k})|\leq c_{2}|p_{k}^{\top }f(x_{k})}|$.

The strong Wolfe curvature condition restricts the slope of $\phi (\alpha )$ from getting too positive, hence excluding points far away from the stationary point of $\phi$.

Goldstein Conditions

Another condition to find an appropriate step length is called Goldstein conditions.

$f(x_{k})+(1-c)\alpha _{k}\nabla {f_{k}^{\top }}p_{k}\leq f(x_{k}+\alpha p_{k})\leq f(x_{k})+c\alpha _{k}\nabla {f_{k}^{\top }}p_{k}$

where $0\leq c\leq 1/2$. The Goldstein condition is quite similar with the Wolfe condition in that, its second inequality ensures that the step length $\alpha$ will decrease the objective function sufficiently and its first inequality keep $\alpha$ from being too short. In comparison with Wolfe condition, one disadvantage of Goldstein condition is that the first inequality of the condition might exclude all minimizers of $\phi$ function. However, usually it is not a fatal problem as long as the objective decrease in the direction of convergence. As a short conclusion, the Goldstein and Wolfe conditions have quite similar convergence theories. Compared to the Wolfe conditions, the Goldstein conditions are often used in Newton-type methods but are not well suited for quasi-Newton methods that maintain a positive definite Hessian approximation.

Backtracking Line Search

The backtracking method is often used to find the appropriate step length and terminate line search based. The backtracking method starts with a relatively large initial step length (e.g., 1 for Newton method), then iteratively shrinking it by a contraction factor until the Armijo (sufficient decrease) condition is satisfied. The advantage of this approach is that the curvature condition needs not be considered, and the step length found at each line search iterate is ensured to be short enough to satisfy sufficient decrease but large enough to still allow the algorithm to make reasonable progress towards convergence.

The backtracking algorithm involves control parameters $\rho \in (0,1)$ and $c\in (0,1)$, and it is roughly as follows:

Choose $\alpha _{0}>0,\rho \in (0,1),c\in (0,1)$

Set $\alpha \leftarrow \alpha _{0}$

While $f(x_{k}+\alpha p_{k})>f(x_{k})+c\alpha \nabla f_{k}^{\top }p_{k}$

$\alpha \leftarrow \rho \alpha$

End while

Return $\alpha _{k}=\alpha$

Numeric Example

For example, we can use line search to solve the unconstrained optimization problem

$\min \ f(x)=x_{1}+x_{2}+2x_{1}x_{2}+2x_{1}^{2}+x_{2}^{2}$

First iteration:

Starting from $x=[0,0]^{T}$, we have $\nabla f(x)=[1+2x_{2}+4x_{1},-1+2x_{1}+2x_{2}]$

$\alpha _{0}=-\nabla f(x_{0})=[-1,1]$

$x_{1}=[0,0]+\alpha [-1,1]=[-\alpha ,\alpha ]$

$f(x_{1})=\alpha _{0}^{2}-2\alpha _{0}$ Taking partial derivative with respect to $\alpha _{0}$ and set it to zero ${\frac {\partial f(x_{1})}{\partial \alpha _{0}}}=\alpha _{0}=1$ Therefore, $x_{1}=[-1,1]$

Second iteration:

Given $x_{1}=[-1,1]$, we have $\alpha _{1}=-\nabla f(x_{1})=[1,1]$ Then $x_{2}=[-1,1]+\alpha _{1}[1,1]=[-1+\alpha _{1},1+\alpha _{1}]$

$f(x_{2})=5\alpha _{1}^{2}-2\alpha _{1}-1$ Taking partial derivative with respect to $\alpha _{1}$ and set it to zero

${\frac {\partial f(x_{2})}{\partial \alpha _{1}}}=0$ Then we can get $\alpha _{1}=0.2$

Therefore, $x_{2}=[-0.8,1.2]$

Third iteration:

Given $x_{2}=-0.8,1.2]$, we have have $\alpha _{2}=-\nabla f(x_{2})=[-0.2,0.2]$

Then $x_{3}=[-0.8,1.2]+\alpha _{2}[-0.2,0.2]=[0.8-0.2\alpha _{2},1.2+0.2\alpha _{2}]$

$f(x_{3})=0.04\alpha _{2}^{2}-1.2-0.08\alpha _{2}$

Taking partial derivative with respect to $\alpha _{2}$

${\frac {\partial f(x_{3})}{\partial \alpha _{2}}}=0.08\alpha _{2}-0.08=0$ Then we can get $\alpha _{2}=1$

Therefore, $x_{3}=[-1,1.4]$

Fourth iteration:

Given $x_{3}=[-1,1.4]$, we have $\alpha _{3}=-\nabla f(x_{3})=[0.2,0.2]$

Then $x_{4}=[-1,1.4]+\alpha _{2}[0.2,0.2]=[0.2\alpha _{3}-1,0.2\alpha _{3}+1.4]$

$f(x_{4})=-0.08\alpha _{3}+0.2\alpha _{3}^{2}-1.24$

Taking partial derivative with respect to $\alpha _{3}$,

${\frac {\partial f(x_{4})}{\partial \alpha _{3}}}=0.4\alpha _{3}-0.08=0$Then we can get $\alpha _{3}=0.2$

Therefore, $x_{4}=[-0.96,1.44]$

Fifth iteration:

Given $x_{4}=[-0.96,1.44]$, we have $\alpha _{4}=-\nabla f(x_{3})=[-0.04,0.04]$

Then $x_{5}=[-0.96,1.4]+\alpha _{3}[-0.04,0.04]=[-0.96-0.04\alpha _{4},0.04\alpha _{4}+1.44]$

$f(x_{5})=0.0016\alpha _{4}^{2}-0.0032\alpha 4-1.248$

Taking partial derivative with respect to $\alpha _{4}$,

${\frac {\partial f(x_{5})}{\partial \alpha _{4}}}=0.00324\alpha _{4}-0.0032=0$ Then we can get $\alpha _{4}=1$

Therefore, $x_{5}=[-1,1.48]$ and $\nabla f(x_{5})=[-0.04,-0.04]$

Check to see if the convergence satisfied evaluated $||\nabla f(x_{5})||$:

$||\nabla f(x_{5})||={\sqrt {(-0.04)^{2}+(-0.04)^{2}}}=0.0565$.

Since 0.0565 is relatively small and is close enough to zero, the line search is converged. The derived optimal solution is $x_{5}=[-1,1.48]$ and the optimal value is -1.25.

Applications

A common application of line search method is in minimizing the loss function in training machine learning models. For example, when training a classification model with logistic regression, gradient descent algorithm, which is a classic method of line search, can be used to minimize the logistic loss and compute the coefficients by iteration till the loss function reaches converges to a local minimum.

In addition, Newton's method and quasi-Newton's method are applied in various industries when finding the minimizer of nonlinear objective functions. These inexact line search methods achieve faster calculation when applied in algorithms.

Conclusion

Line Search is a useful strategy to solve unconstrained optimization problems. The success of the line search algorithm depends on careful consideration of the choice of both the direction $p_{k}$ and the step size $\alpha _{k}$.This page has introduced the basic algorithm firstly, and then includes the exact search and inexact search. The exact search contains the steepest descent, and the inexact search covers the Wolfe and Goldstein conditions, backtracking, and Zoutendijk's theorem. More approaches to solve unconstrained optimization problems can be found in trust-region methods, conjugate gradient methods, Newton's method and Quasi-Newton method.