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 is a first-order method to solve unconstrained optimization problems. When applying line search methods, the user needs to supply a starting point for all algorithms. With the initial starting point, $x_{0}$, optimization algorithms generate a sequence of iterates $\{x_{k}\}_{k=0}^{\infty }$ which terminates when an approximated solution has been achieved or no more progress can be made. Line search and trust-region method are two fundamental strategies for locating the new $x_{k+1}$ given the current point.

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. We can verify the claim 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, we will use this 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}$, we need to 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. We have exact line search and inexact line search 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})$

One advantage of the steepest descent method is that it has a nice convergence theory. 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$

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

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

When we minimize the objective function using numeric methods, in each iteration, the updated objective is $\phi (\alpha )=f(x_{k}+\alpha p_{k})$, a function of $\alpha$ when we fix the direction. Our goal is to minimize the objective with respect to $\alpha$. However, sometimes if we want to solve for the exact minimum in each iteration, it might be computationally expensive and the algorithm will be time consuming. Therefore, in practice we just 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})$. A problem is, we cannot guarantee aconvergence to the function's minimum, so we often apply the Wolfe or Goldstein conditions to find 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.

(1) 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 10E-4. This condition ensures the computed step length can reduces the objective function $f(x_{k})$ sufficiently. Only using this condition, however, we cannot guarantee $x_{k}$ to converge in a reasonable number of iterations, since Armijo condition is always satisfied with step length that is small enough. Therefore, we need to pair it with the second condition below, in order to keep $\alpha _{k}$ from being too short.

(2) 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 )$.

(2*) Strong Wolfe curvature condition

The (weak) Wolfe conditions can result in an $\alpha$ value that is not close to the minimizer of $\phi (\alpha )$. We can modify the (weak) Wolfe conditions by using the following condition called Strong Wolfe condition which writes the curvature condition in $(2)$ 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.71.

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.

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, Newton's method and Quasi-Newton method.