Conjugate gradient methods

From Cornell University Computational Optimization Open Textbook - Optimization Wiki
Revision as of 15:47, 6 December 2021 by MiScott1601 (talk | contribs)
Jump to navigation Jump to search

Author: Alexandra Roberts, Anye Shi, Yue Sun (SYSEN 6800 Fall 2021)

Introduction

Figure 1. A comparison of the convergence of gradient descent (in green) and conjugate vector (in red) for minimizing a quadratic function. In theory, the conjugate gradient method converges in at most steps, where is the size of the matrix of the system (here ).[1]

The conjugate gradient method (CG) was originally invented to minimize a quadratic function:

where is an symmetric positive definite matrix, and are vectors.
The solution to the minimization problem is equivalent to solving the linear system, i.e. determining when , i.e.

The conjugate gradient method is often implemented as an iterative algorithm and can be considered as being between Newton’s method, a second-order method that incorporates Hessian and gradient, and the method of steepest descent, a first-order method that uses gradient [2]. Newton’s Method usually reduces the number of iterations needed, but the calculation of the Hessian matrix and its inverse increases the computation required for each iteration. Steepest descent takes repeated steps in the opposite direction of the gradient of the function at the current point. It often takes steps in the same direction as earlier ones, resulting in slow convergence (Figure 1). To avoid the high computational cost of Newton’s method and to accelerate the convergence rate of steepest descent, the conjugate gradient method was developed.

The idea of the CG method is to pick orthogonal search directions first and, in each search direction, take exactly one step such that the step size is to the proposed solution at that direction. The solution is reached after steps [3] as, theoretically, the number of iterations needed by the CG method is equal to the number of different eigenvalues of , i.e. at most . This makes it attractive for large and sparse problems. The method can be used to solve least-squares problems and can also be generalized to a minimization method for general smooth functions [3].

Theory

The definition of A-conjugate direction

Let be a symmetric positive definite matrix. are the vectors that orthogonal (conjugate) to each other with respect to if
.

Note that if , any two vectors will be conjugated to each other. If , conjugacy is equivalent to the conventional notion of orthogonality. If are -conjugated to each other, then the set of vectors are linearly independent.

The motivation of A-conjugacy[4]

As is a set of -conjugate vectors, then can be used as a basis and express the solution to is:


Then multiplying

Because and the A-conjugacy of , i.e. , the multiplication will cancel out all the terms except for term k


Then the solution x* will be

Because A is a symmetric and positive-definite matrix, so the term defines an inner product and, therefore, no need to calculate the inversion of matrix A.

Conjugate Direction Theorem

Let be a set of n A-conjugate vectors, be a random starting point. Then



After n steps, xn = x*.

Proof:
Given



Therefore






The conjugate gradient method

The conjugate gradient method is a conjugate direction method in which selected successive direction vectors are treated as a conjugate version of the successive gradients obtained while the method progresses. The conjugate directions are not specified beforehand but rather are determined sequentially at each step of the iteration [4]. If the conjugate vectors are carefully chosen, then not all the conjugate vectors may be needed to obtain the solution. Therefore, the conjugate gradient method is regarded as an iterative method. This also allows approximate solutions to systems where n is so large that the direct method requires too much time [3].

Algorithm[3]

Given be a set of n A-conjugate vectors, then can be minimized by stepping from along to the minimum , stepping from along to the minimum , etc. And let be randomly chosen, then the algorithm is the following:

Alg 1: Pick mutually A-conjugate, and from a random ,
For k = 1 to n
{

  1. ;
  2. ;

}
Return

Here Alg 1 is with a particular choice of . Let be the gradient at . A practical way to enforce this is by requiring that the next search direction be built out of the current gradient and all previous search directions. The CG method picks as the component of A-conjugate to :



As , for i = 1,...,k, giving the following CG algorithm:
Alg 2: From a random ,
For k = 1 to n
{

  1. ;
  2. if return ;
  3. if (k > 1) ;
  4. if (k = 1) ;
  5. else ;
  6. ;
  7. ;

}
Return

The formulas in the Alg 2 can be simplified as the following:



Then and can be simplified by multiplying the above gradient formula by and as the following:


As ,
so we have

Therefore

This gives the following simplified version of Alg 2:
Alg 3: From a random , and set ,
For k = 1 to n
{

  1. if return ;
  2. if (k > 1) ;
  3. if (k = 1) ;
  4. else ;
  5. ;
  6. Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \textbf{x}_k = \textbf{x}_{i-1} + \alpha_i\textbf{d}_i}
  7. Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \textbf{g}_{i}=\textbf{g}_{i-1}-\alpha_i\textbf{A}\textbf{d}_i} ;

}
Return Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \textbf{x}_n}

Numerical example

Consider the linear system Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \textbf{A}\textbf{x} = \textbf{b}}
Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \textbf{A}\textbf{x} = \begin{bmatrix}5 & 1 \\1 & 8 \\\end{bmatrix}\begin{bmatrix}x_{1} \\x_{2}\end{bmatrix} = \begin{bmatrix}3 \\2\end{bmatrix}} .
The initial starting point is set to be
Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \textbf{x}_{0} = \begin{bmatrix}2 \\1\end{bmatrix}} .
Implement the conjugate gradient method to approximate the solution to the system.

Solution:
The exact solution is given below for later reference:
Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \textbf{x}_{*} = \begin{bmatrix}22/39 \\7/39\end{bmatrix}\approx \begin{bmatrix}0.5641\\0.1794\end{bmatrix}} .

Step 1:
Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \textbf{g}_{0}=\textbf{b}-\textbf{A}\textbf{x}_0 = \begin{bmatrix}3 \\2\end{bmatrix}-\begin{bmatrix}5 &1 \\1& 8\\\end{bmatrix}\begin{bmatrix}2\\1\end{bmatrix} =\begin{bmatrix}-8 \\-8\end{bmatrix}= \textbf{d}_1}

Step 2:
Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \alpha_1 = \frac{\textbf{g}_{0}^{T}\textbf{g}_{0}}{\textbf{d}_{1}^{T}\textbf{A}\textbf{d}_1}=\frac{\begin{bmatrix}-8 &-8 \\\end{bmatrix}\begin{bmatrix}-8 \\-8\end{bmatrix}}{\begin{bmatrix}-8 &-8 \\\end{bmatrix}\begin{bmatrix}5 &1 \\1& 8 \\\end{bmatrix}\begin{bmatrix}-8 \\-8\end{bmatrix}}=\frac{2}{15}}

Step 3:
Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \textbf{x}_1 = \textbf{x}_{0} + \alpha_{1}\textbf{d}_{1}=\begin{bmatrix}2\\1\end{bmatrix}+\frac{2}{15}\begin{bmatrix}-8\\-8\end{bmatrix}=\begin{bmatrix}0.9333 \\-0.0667\end{bmatrix}}

Step 4:
Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \textbf{g}_{1}=\textbf{g}_{0}-\alpha_1\textbf{A}\textbf{d}_1 = \begin{bmatrix}-8\\-8 \end{bmatrix}-\frac{2}{15}\begin{bmatrix}5&1\\1&8\end{bmatrix}\begin{bmatrix}-8\\-8\end{bmatrix} = \begin{bmatrix}-1.6\\1.6\end{bmatrix}}

Step 5:
Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \beta_2 =-\frac{\textbf{g}_{1}^T\textbf{g}_{1}}{{\textbf{g}_{0}^T\textbf{g}_{0}}}=- \frac{\begin{bmatrix}-1.6&1.6\\\end{bmatrix}\begin{bmatrix}-1.6 \\1.6\end{bmatrix}}{\begin{bmatrix}-8 &-8\\\end{bmatrix}\begin{bmatrix}-8 \\-8\end{bmatrix}}= -0.04}

Step 6:
Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \textbf{d}_2 = \textbf{g}_1 - \beta_2\textbf{d}_1=\begin{bmatrix}-1.6\\1.6\end{bmatrix}+0.04\begin{bmatrix}-8\\-8\end{bmatrix}=\begin{bmatrix}-1.92 \\1.28\end{bmatrix}}

Step 7:
Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \alpha_2 =\frac{\textbf{g}_{1}^T\textbf{g}_{1}}{{\textbf{d}_{2}^T\textbf{A}\textbf{d}_{2}}}=\frac{\begin{bmatrix}-1.6 &1.6\end{bmatrix}\begin{bmatrix}-1.6\\1.6\end{bmatrix}}{\begin{bmatrix}-1.92&1.28\\\end{bmatrix}\begin{bmatrix}5 &1 \\1& 8 \\\end{bmatrix}\begin{bmatrix}-1.92\\1.28\end{bmatrix}}=0.1923}

Step 8:
Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \textbf{x}_2 = \textbf{x}_{1} +\alpha_{2}\textbf{d}_{2}=\begin{bmatrix}0.9333\\-0.0667\end{bmatrix}+0.1923\begin{bmatrix}-1.92\\1.28\end{bmatrix}=\begin{bmatrix}0.5641 \\0.1794\end{bmatrix}}

Therefore, Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \textbf{x}_2} is the approximation result of the system.

Application

Conjugate gradient methods have often been used to solve a wide variety of numerical problems, including linear and nonlinear algebraic equations, eigenvalue problems and minimization problems. These applications have been similar in that they involve large numbers of variables or dimensions. In these circumstances any method of solution which involves storing a full matrix of this large order, becomes inapplicable. Thus recourse to the conjugate gradient method may be the only alternative [5]. Here we show an example of image reconstruction.

Iterative image reconstruction

The conjugate gradient method is used to solve for the update in iterative image reconstruction problems. For example, in the magnetic resonance imaging (MRI) contrast known as quantitative susceptibility mapping (QSM), the reconstructed image Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \chi} is iteratively solved for from magnetic field data Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \textbf{b}} by the relation[6]
Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \textbf{b}=\textbf{D}\chi}

Where Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle D} is the matrix expressing convolution with the dipole kernel in the Fourier domain. Given that the problem is ill-posed, a physical prior is used in the reconstruction, which is framed as a constrained L1 norm minimization

Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle min_{\chi}\left\| f(\chi)\right\|_1}
Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle s.t. \left\| g(\chi)-c\right\|_{2}^{2}}

A detailed treatment of the function Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle f(\chi)} and Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle g(\chi)} can be found at [6]. This problem can be expressed as an unconstrained minimization problem via the Lagrange Multiplier Method


Where

Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle E(\chi,\lambda)\equiv \left\| f(\chi)\right\|_1+\lambda (\left\| g(\chi)-c\right\|_{2}^{2}-\varepsilon)}

The first-order conditions require Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \nabla_{\chi}E(\chi,\lambda)=0} and Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \nabla_{\lambda}E(\chi,\lambda)=0} . These conditions result in Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \nabla_{\chi}f(\chi)+\nabla_{\chi}g(\chi) - \tilde{c}=0} and Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \left\| g(\chi)-c\right\|_{2}^{2}\approx \varepsilon } , respectively. The update can be solved for via fixed point iteration [6].

Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \chi_{n+1}=L^{-1}(\chi^n)\tilde{c}}

And expressed as the quasi-Newton problem, more robust to round-off error [6]

Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \chi_{n+1}=\chi_n - L^{-1}(\chi_n)\nabla E(\chi_n,\lambda)}

Which is solved with the CG method until the residual Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \left\|\chi_{n+1}-\chi\right\|_2/\left\|\chi_n\right\|_2\leq \theta } where Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \theta} is a specified tolerance, such as Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle 10^{-2}} .

Conclusion

The conjugate gradient method was invented to avoid the high computational cost of Newton’s method and to accelerate the convergence rate of steepest descent. As an iterative method, each step only requires Failed to parse (SVG (MathML can be enabled via browser plugin): Invalid response ("Math extension cannot connect to Restbase.") from server "https://wikimedia.org/api/rest_v1/":): {\displaystyle \textbf{A}\textbf{d}_i} multiplication free from the storage of matrix . And selected direction vectors are treated as a conjugate version of the successive gradients obtained while the method progresses. So it monotonically improves approximations to the exact solution and may reach the required tolerance after a relatively small (compared to the problem size) number of iterations in the absence of round-off error, which makes it widely used for solving large and sparse problems.

Reference

  1. Jonathan Shewchuk, “An Introduction to the Conjugate Gradient Method Without the Agonizing Pain,” 1994.
  2. “Conjugate gradient method,” Wikipedia. Nov. 25, 2021. Accessed: Nov. 26, 2021. [Online]. Available: https://en.wikipedia.org/w/index.php?title=Conjugate_gradient_method&oldid=1057033318
  3. 3.0 3.1 3.2 3.3 W. Stuetzle, “The Conjugate Gradient Method.” 2001. [Online]. Available: https://sites.stat.washington.edu/wxs/Stat538-w03/conjugate-gradients.pdf
  4. 4.0 4.1 A. Singh and P. Ravikumar, “Conjugate Gradient Descent.” 2012. [Online]. Available: http://www.cs.cmu.edu/~pradeepr/convexopt/Lecture_Slides/conjugate_direction_methods.pdf
  5. R. Fletcher, “Conjugate gradient methods for indefinite systems,” in Numerical Analysis, Berlin, Heidelberg, 1976, pp. 73–89. doi: 10.1007/BFb0080116.
  6. 6.0 6.1 6.2 6.3 J. Liu et al., “Morphology enabled dipole inversion for quantitative susceptibility mapping using structural consistency between the magnitude image and the susceptibility map,” NeuroImage, vol. 59, no. 3, pp. 2560–2568, Feb. 2012, doi: 10.1016/j.neuroimage.2011.08.082.