Non-linear least squares

From HandWiki
Short description: Approximation method in statistics

Non-linear least squares is the form of least squares analysis used to fit a set of m observations with a model that is non-linear in n unknown parameters (m ≥ n). It is used in some forms of nonlinear regression. The basis of the method is to approximate the model by a linear one and to refine the parameters by successive iterations. There are many similarities to linear least squares, but also some significant differences. In economic theory, the non-linear least squares method is applied in (i) the probit regression, (ii) threshold regression, (iii) smooth regression, (iv) logistic link regression, (v) Box–Cox transformed regressors ([math]\displaystyle{ m(x,\theta_i) = \theta_1 + \theta_2 x^{(\theta_{3})} }[/math]).

Theory

Consider a set of [math]\displaystyle{ m }[/math] data points, [math]\displaystyle{ (x_1, y_1), (x_2, y_2), \dots, (x_m, y_m), }[/math] and a curve (model function) [math]\displaystyle{ \hat{y} = f(x, \boldsymbol \beta), }[/math] that in addition to the variable [math]\displaystyle{ x }[/math] also depends on [math]\displaystyle{ n }[/math] parameters, [math]\displaystyle{ \boldsymbol \beta = (\beta_1, \beta_2, \dots, \beta_n), }[/math] with [math]\displaystyle{ m\ge n. }[/math] It is desired to find the vector [math]\displaystyle{ \boldsymbol \beta }[/math] of parameters such that the curve fits best the given data in the least squares sense, that is, the sum of squares [math]\displaystyle{ S = \sum_{i=1}^{m} r_i^2 }[/math] is minimized, where the residuals (in-sample prediction errors) ri are given by [math]\displaystyle{ r_i = y_i - f(x_i, \boldsymbol \beta) }[/math] for [math]\displaystyle{ i=1, 2,\dots, m. }[/math]

The minimum value of S occurs when the gradient is zero. Since the model contains n parameters there are n gradient equations: [math]\displaystyle{ \frac{\partial S}{\partial \beta_j} = 2 \sum_i r_i\frac{\partial r_i}{\partial \beta_j} = 0 \quad (j=1,\ldots,n). }[/math]

In a nonlinear system, the derivatives [math]\displaystyle{ \frac{\partial r_i}{\partial \beta_j} }[/math] are functions of both the independent variable and the parameters, so in general these gradient equations do not have a closed solution. Instead, initial values must be chosen for the parameters. Then, the parameters are refined iteratively, that is, the values are obtained by successive approximation, [math]\displaystyle{ \beta_j \approx \beta_j^{k+1} =\beta^k_j+\Delta \beta_j. }[/math]

Here, k is an iteration number and the vector of increments, [math]\displaystyle{ \Delta \boldsymbol \beta }[/math] is known as the shift vector. At each iteration the model is linearized by approximation to a first-order Taylor polynomial expansion about [math]\displaystyle{ \boldsymbol \beta^k }[/math] [math]\displaystyle{ f(x_i,\boldsymbol \beta)\approx f(x_i,\boldsymbol \beta^k) +\sum_j \frac{\partial f(x_i,\boldsymbol \beta^k)}{\partial \beta_j} \left(\beta_j -\beta^{k}_j \right) = f(x_i,\boldsymbol \beta^k) +\sum_j J_{ij} \,\Delta\beta_j. }[/math] The Jacobian matrix, J, is a function of constants, the independent variable and the parameters, so it changes from one iteration to the next. Thus, in terms of the linearized model, [math]\displaystyle{ \frac{\partial r_i}{\partial \beta_j} = -J_{ij} }[/math] and the residuals are given by [math]\displaystyle{ \Delta y_i = y_i- f(x_i,\boldsymbol \beta^k), }[/math] [math]\displaystyle{ r_i = y_i - f(x_i, \boldsymbol \beta) = \left(y_i- f(x_i,\boldsymbol \beta^k)\right)+ \left(f(x_i,\boldsymbol \beta^k) - f(x_i, \boldsymbol \beta)\right)\approx\Delta y_i- \sum_{s=1}^{n} J_{is} \Delta \beta_s . }[/math]

Substituting these expressions into the gradient equations, they become [math]\displaystyle{ -2\sum_{i=1}^{m} J_{ij} \left( \Delta y_i - \sum_{s=1}^{n} J_{is}\ \Delta \beta_s \right) = 0, }[/math] which, on rearrangement, become n simultaneous linear equations, the normal equations [math]\displaystyle{ \sum_{i=1}^{m} \sum_{s=1}^{n} J_{ij}J_{is}\ \Delta \beta_s=\sum_{i=1}^{m} J_{ij}\ \Delta y_i \qquad (j=1,\dots,n). }[/math]

The normal equations are written in matrix notation as [math]\displaystyle{ \left(\mathbf{J}^\mathsf{T}\mathbf{J}\right) \Delta \boldsymbol \beta = \mathbf{J}^\mathsf{T}\ \Delta \mathbf{y}. }[/math]

These equations form the basis for the Gauss–Newton algorithm for a non-linear least squares problem.

Note the sign convention in the definition of the Jacobian matrix in terms of the derivatives. Formulas linear in [math]\displaystyle{ J }[/math] may appear with factor of [math]\displaystyle{ -1 }[/math] in other articles or the literature.

Extension by weights

When the observations are not equally reliable, a weighted sum of squares may be minimized, [math]\displaystyle{ S = \sum_{i=1}^m W_{ii} r_i^2. }[/math]

Each element of the diagonal weight matrix W should, ideally, be equal to the reciprocal of the error variance of the measurement.[1] The normal equations are then, more generally, [math]\displaystyle{ \left(\mathbf{J}^\mathsf{T}\mathbf{WJ}\right) \Delta \boldsymbol \beta = \mathbf{J}^\mathsf{T}\mathbf{W}\ \Delta \mathbf{y}. }[/math]

Geometrical interpretation

In linear least squares the objective function, S, is a quadratic function of the parameters. [math]\displaystyle{ S = \sum_i W_{ii} \left(y_i - \sum_j X_{ij}\beta_j \right)^2 }[/math] When there is only one parameter the graph of S with respect to that parameter will be a parabola. With two or more parameters the contours of S with respect to any pair of parameters will be concentric ellipses (assuming that the normal equations matrix [math]\displaystyle{ \mathbf{X}^\mathsf{T}\mathbf{WX} }[/math] is positive definite). The minimum parameter values are to be found at the centre of the ellipses. The geometry of the general objective function can be described as paraboloid elliptical. In NLLSQ the objective function is quadratic with respect to the parameters only in a region close to its minimum value, where the truncated Taylor series is a good approximation to the model. [math]\displaystyle{ S \approx \sum_i W_{ii} \left(y_i - \sum_j J_{ij}\beta_j \right)^2 }[/math] The more the parameter values differ from their optimal values, the more the contours deviate from elliptical shape. A consequence of this is that initial parameter estimates should be as close as practicable to their (unknown!) optimal values. It also explains how divergence can come about as the Gauss–Newton algorithm is convergent only when the objective function is approximately quadratic in the parameters.

Computation

Initial parameter estimates

Some problems of ill-conditioning and divergence can be corrected by finding initial parameter estimates that are near to the optimal values. A good way to do this is by computer simulation. Both the observed and calculated data are displayed on a screen. The parameters of the model are adjusted by hand until the agreement between observed and calculated data is reasonably good. Although this will be a subjective judgment, it is sufficient to find a good starting point for the non-linear refinement. Initial parameter estimates can be created using transformations or linearizations. Better still evolutionary algorithms such as the Stochastic Funnel Algorithm can lead to the convex basin of attraction that surrounds the optimal parameter estimates.[citation needed] Hybrid algorithms that use randomization and elitism, followed by Newton methods have been shown to be useful and computationally efficient[citation needed].

Solution

Any method among the ones described below can be applied to find a solution.

Convergence criteria

The common sense criterion for convergence is that the sum of squares does not increase from one iteration to the next. However this criterion is often difficult to implement in practice, for various reasons. A useful convergence criterion is [math]\displaystyle{ \left|\frac{S^k-S^{k+1}}{S^k}\right| \lt 0.0001. }[/math] The value 0.0001 is somewhat arbitrary and may need to be changed. In particular it may need to be increased when experimental errors are large. An alternative criterion is [math]\displaystyle{ \left|\frac{\Delta \beta_j}{\beta_j}\right| \lt 0.001, \qquad j=1,\dots,n. }[/math]

Again, the numerical value is somewhat arbitrary; 0.001 is equivalent to specifying that each parameter should be refined to 0.1% precision. This is reasonable when it is less than the largest relative standard deviation on the parameters.

Calculation of the Jacobian by numerical approximation

Main page: Numerical differentiation

There are models for which it is either very difficult or even impossible to derive analytical expressions for the elements of the Jacobian. Then, the numerical approximation [math]\displaystyle{ \frac{\partial f(x_i, \boldsymbol \beta)}{\partial \beta_j} \approx \frac{\delta f(x_i, \boldsymbol \beta)}{\delta \beta_j} }[/math] is obtained by calculation of [math]\displaystyle{ f(x_i, \boldsymbol \beta) }[/math] for [math]\displaystyle{ \beta_j }[/math] and [math]\displaystyle{ \beta_j+\delta \beta_j }[/math]. The increment,[math]\displaystyle{ \delta \beta_j }[/math], size should be chosen so the numerical derivative is not subject to approximation error by being too large, or round-off error by being too small.

Parameter errors, confidence limits, residuals etc.

Some information is given in the corresponding section on the linear least squares page.

Multiple minima

Multiple minima can occur in a variety of circumstances some of which are:

  • A parameter is raised to a power of two or more. For example, when fitting data to a Lorentzian curve [math]\displaystyle{ f(x_i, \boldsymbol \beta) = \frac{\alpha}{1 + \left(\frac{\gamma-x_i}{\beta} \right)^2} }[/math] where [math]\displaystyle{ \alpha }[/math] is the height, [math]\displaystyle{ \gamma }[/math] is the position and [math]\displaystyle{ \beta }[/math] is the half-width at half height, there are two solutions for the half-width, [math]\displaystyle{ \hat \beta }[/math] and [math]\displaystyle{ -\hat \beta }[/math] which give the same optimal value for the objective function.
  • Two parameters can be interchanged without changing the value of the model. A simple example is when the model contains the product of two parameters, since [math]\displaystyle{ \alpha \beta }[/math] will give the same value as [math]\displaystyle{ \beta \alpha }[/math].
  • A parameter is in a trigonometric function, such as [math]\displaystyle{ \sin \beta }[/math], which has identical values at [math]\displaystyle{ \hat \beta + 2n \pi }[/math]. See Levenberg–Marquardt algorithm for an example.

Not all multiple minima have equal values of the objective function. False minima, also known as local minima, occur when the objective function value is greater than its value at the so-called global minimum. To be certain that the minimum found is the global minimum, the refinement should be started with widely differing initial values of the parameters. When the same minimum is found regardless of starting point, it is likely to be the global minimum.

When multiple minima exist there is an important consequence: the objective function will have a maximum value somewhere between two minima. The normal equations matrix is not positive definite at a maximum in the objective function, as the gradient is zero and no unique direction of descent exists. Refinement from a point (a set of parameter values) close to a maximum will be ill-conditioned and should be avoided as a starting point. For example, when fitting a Lorentzian the normal equations matrix is not positive definite when the half-width of the band is zero.[2]

Transformation to a linear model

A non-linear model can sometimes be transformed into a linear one. Such an approximation is, for instance, often applicable in the vicinity of the best estimator, and it is one of the basic assumption in most iterative minimization algorithms. When a linear approximation is valid, the model can directly be used for inference with a generalized least squares, where the equations of the Linear Template Fit[3] apply.

Another example of a linear approximation would be, when the model is a simple exponential function, [math]\displaystyle{ f(x_i,\boldsymbol \beta)= \alpha e^{\beta x_i} }[/math] it can be transformed into a linear model by taking logarithms. [math]\displaystyle{ \log f(x_i,\boldsymbol \beta) = \log \alpha + \beta x_i }[/math] Graphically this corresponds to working on a semi-log plot. The sum of squares becomes [math]\displaystyle{ S = \sum_i (\log y_i-\log \alpha - \beta x_i)^2. }[/math] This procedure should be avoided unless the errors are multiplicative and log-normally distributed because it can give misleading results. This comes from the fact that whatever the experimental errors on y might be, the errors on log y are different. Therefore, when the transformed sum of squares is minimized different results will be obtained both for the parameter values and their calculated standard deviations. However, with multiplicative errors that are log-normally distributed, this procedure gives unbiased and consistent parameter estimates.

Another example is furnished by Michaelis–Menten kinetics, used to determine two parameters [math]\displaystyle{ V_{\max} }[/math] and [math]\displaystyle{ K_m }[/math]: [math]\displaystyle{ v = \frac{V_{\max}[S]}{K_{m} + [S]}. }[/math] The Lineweaver–Burk plot [math]\displaystyle{ \frac{1}{v} = \frac{1}{V_\max} + \frac{K_m}{V_{\max}[S]} }[/math] of [math]\displaystyle{ \frac{1}{v} }[/math] against [math]\displaystyle{ \frac{1}{[S]} }[/math] is linear in the parameters [math]\displaystyle{ \frac{1}{V_\max} }[/math] and [math]\displaystyle{ \frac{K_m}{V_\max} }[/math], but very sensitive to data error and strongly biased toward fitting the data in a particular range of the independent variable [math]\displaystyle{ [S] }[/math].

Algorithms

Gauss–Newton method

Main page: Gauss–Newton algorithm

The normal equations [math]\displaystyle{ \left( \mathbf{J}^\mathsf{T}\mathbf{WJ} \right)\Delta \boldsymbol\beta = \left( \mathbf{J}^\mathsf{T}\mathbf{W} \right) \Delta \mathbf{y} }[/math] may be solved for [math]\displaystyle{ \Delta \boldsymbol\beta }[/math] by Cholesky decomposition, as described in linear least squares. The parameters are updated iteratively [math]\displaystyle{ \boldsymbol\beta^{k+1} = \boldsymbol\beta^k + \Delta \boldsymbol\beta }[/math] where k is an iteration number. While this method may be adequate for simple models, it will fail if divergence occurs. Therefore, protection against divergence is essential.

Shift-cutting

If divergence occurs, a simple expedient is to reduce the length of the shift vector, [math]\displaystyle{ \Delta \boldsymbol\beta }[/math], by a fraction, f [math]\displaystyle{ \boldsymbol\beta^{k+1} = \boldsymbol\beta^k+f\ \Delta \boldsymbol\beta. }[/math] For example, the length of the shift vector may be successively halved until the new value of the objective function is less than its value at the last iteration. The fraction, f could be optimized by a line search.[4] As each trial value of f requires the objective function to be re-calculated it is not worth optimizing its value too stringently.

When using shift-cutting, the direction of the shift vector remains unchanged. This limits the applicability of the method to situations where the direction of the shift vector is not very different from what it would be if the objective function were approximately quadratic in the parameters, [math]\displaystyle{ \boldsymbol\beta^k. }[/math]

Marquardt parameter

Main page: Levenberg–Marquardt algorithm

If divergence occurs and the direction of the shift vector is so far from its "ideal" direction that shift-cutting is not very effective, that is, the fraction, f required to avoid divergence is very small, the direction must be changed. This can be achieved by using the Marquardt parameter.[5] In this method the normal equations are modified [math]\displaystyle{ \left( \mathbf{J}^\mathsf{T} \mathbf{WJ} + \lambda \mathbf{I} \right) \Delta \boldsymbol \beta = \left( \mathbf{J}^\mathsf{T} \mathbf{W} \right) \Delta \mathbf{y} }[/math] where [math]\displaystyle{ \lambda }[/math] is the Marquardt parameter and I is an identity matrix. Increasing the value of [math]\displaystyle{ \lambda }[/math] has the effect of changing both the direction and the length of the shift vector. The shift vector is rotated towards the direction of steepest descent when [math]\displaystyle{ \lambda \mathbf{I} \gg \mathbf{J}^\mathsf{T}\mathbf{WJ}, \ {\Delta \boldsymbol \beta} \approx \frac 1 \lambda \mathbf{J}^\mathsf{T}\mathbf{W}\ \Delta \mathbf{y}. }[/math] [math]\displaystyle{ \mathbf{J}^\mathsf{T}\mathbf{W}\, \Delta \mathbf{y} }[/math] is the steepest descent vector. So, when [math]\displaystyle{ \lambda }[/math] becomes very large, the shift vector becomes a small fraction of the steepest descent vector.

Various strategies have been proposed for the determination of the Marquardt parameter. As with shift-cutting, it is wasteful to optimize this parameter too stringently. Rather, once a value has been found that brings about a reduction in the value of the objective function, that value of the parameter is carried to the next iteration, reduced if possible, or increased if need be. When reducing the value of the Marquardt parameter, there is a cut-off value below which it is safe to set it to zero, that is, to continue with the unmodified Gauss–Newton method. The cut-off value may be set equal to the smallest singular value of the Jacobian.[6] A bound for this value is given by [math]\displaystyle{ 1 / \operatorname{tr} \left(\mathbf{J}^\mathsf{T}\mathbf{W J}\right)^{-1} }[/math] where tr is the trace function.[7]

QR decomposition

The minimum in the sum of squares can be found by a method that does not involve forming the normal equations. The residuals with the linearized model can be written as [math]\displaystyle{ \mathbf{r} = \Delta \mathbf{y} - \mathbf{J}\, \Delta\boldsymbol\beta. }[/math] The Jacobian is subjected to an orthogonal decomposition; the QR decomposition will serve to illustrate the process. [math]\displaystyle{ \mathbf{J} = \mathbf{QR} }[/math] where Q is an orthogonal [math]\displaystyle{ m \times m }[/math] matrix and R is an [math]\displaystyle{ m \times n }[/math] matrix which is partitioned into an [math]\displaystyle{ n \times n }[/math] block, [math]\displaystyle{ \mathbf{R}_n }[/math], and a [math]\displaystyle{ (m-n) \times n }[/math] zero block. [math]\displaystyle{ \mathbf{R}_n }[/math] is upper triangular.

[math]\displaystyle{ \mathbf{R}= \begin{bmatrix} \mathbf{R}_n \\ \mathbf{0} \end{bmatrix} }[/math]

The residual vector is left-multiplied by [math]\displaystyle{ \mathbf Q^\mathsf{T} }[/math].

[math]\displaystyle{ \mathbf{Q}^\mathsf{T} \mathbf{r} = \mathbf{Q}^\mathsf{T}\ \Delta \mathbf{y} - \mathbf{R}\ \Delta\boldsymbol\beta = \begin{bmatrix} \left(\mathbf{Q}^\mathsf{T}\ \Delta \mathbf{y} - \mathbf{R}\ \Delta\boldsymbol\beta \right)_n \\ \left(\mathbf{Q}^\mathsf{T}\ \Delta \mathbf{y} \right)_{m-n} \end{bmatrix} }[/math]

This has no effect on the sum of squares since [math]\displaystyle{ S = \mathbf{r}^\mathsf{T} \mathbf{Q} \mathbf{Q}^\mathsf{T} \mathbf{r} = \mathbf{r}^\mathsf{T} \mathbf{r} }[/math] because Q is orthogonal The minimum value of S is attained when the upper block is zero. Therefore, the shift vector is found by solving [math]\displaystyle{ \mathbf{R}_n\ \Delta\boldsymbol\beta = \left(\mathbf{Q}^\mathsf{T}\ \Delta \mathbf{y} \right)_n. }[/math]

These equations are easily solved as R is upper triangular.

Singular value decomposition

A variant of the method of orthogonal decomposition involves singular value decomposition, in which R is diagonalized by further orthogonal transformations.

[math]\displaystyle{ \mathbf{J} = \mathbf{U} \boldsymbol\Sigma \mathbf{V}^\mathsf{T} }[/math] where [math]\displaystyle{ \mathbf U }[/math] is orthogonal, [math]\displaystyle{ \boldsymbol\Sigma }[/math] is a diagonal matrix of singular values and [math]\displaystyle{ \mathbf V }[/math] is the orthogonal matrix of the eigenvectors of [math]\displaystyle{ \mathbf {J}^\mathsf{T}\mathbf{J} }[/math] or equivalently the right singular vectors of [math]\displaystyle{ \mathbf{J} }[/math]. In this case the shift vector is given by [math]\displaystyle{ \Delta\boldsymbol\beta = \mathbf{V} \boldsymbol\Sigma^{-1} \left( \mathbf{U}^\mathsf{T}\ \Delta \mathbf{y} \right)_n. }[/math]

The relative simplicity of this expression is very useful in theoretical analysis of non-linear least squares. The application of singular value decomposition is discussed in detail in Lawson and Hanson.[6]

Gradient methods

There are many examples in the scientific literature where different methods have been used for non-linear data-fitting problems.

  • Inclusion of second derivatives in The Taylor series expansion of the model function. This is Newton's method in optimization. [math]\displaystyle{ f(x_i, \boldsymbol \beta) = f^k(x_i, \boldsymbol \beta) +\sum_j J_{ij} \, \Delta \beta_j + \frac{1}{2}\sum_j\sum_k \Delta\beta_j \, \Delta\beta_k \,H_{jk_{(i)}},\ H_{jk_{(i)}} = \frac{\partial^2 f(x_i, \boldsymbol \beta)}{\partial \beta_j \, \partial \beta_k }. }[/math] The matrix H is known as the Hessian matrix. Although this model has better convergence properties near to the minimum, it is much worse when the parameters are far from their optimal values. Calculation of the Hessian adds to the complexity of the algorithm. This method is not in general use.
  • Davidon–Fletcher–Powell method. This method, a form of pseudo-Newton method, is similar to the one above but calculates the Hessian by successive approximation, to avoid having to use analytical expressions for the second derivatives.
  • Steepest descent. Although a reduction in the sum of squares is guaranteed when the shift vector points in the direction of steepest descent, this method often performs poorly. When the parameter values are far from optimal the direction of the steepest descent vector, which is normal (perpendicular) to the contours of the objective function, is very different from the direction of the Gauss–Newton vector. This makes divergence much more likely, especially as the minimum along the direction of steepest descent may correspond to a small fraction of the length of the steepest descent vector. When the contours of the objective function are very eccentric, due to there being high correlation between parameters, the steepest descent iterations, with shift-cutting, follow a slow, zig-zag trajectory towards the minimum.
  • Conjugate gradient search. This is an improved steepest descent based method with good theoretical convergence properties, although it can fail on finite-precision digital computers even when used on quadratic problems.[8]

Direct search methods

Direct search methods depend on evaluations of the objective function at a variety of parameter values and do not use derivatives at all. They offer alternatives to the use of numerical derivatives in the Gauss–Newton method and gradient methods.

  • Alternating variable search.[4] Each parameter is varied in turn by adding a fixed or variable increment to it and retaining the value that brings about a reduction in the sum of squares. The method is simple and effective when the parameters are not highly correlated. It has very poor convergence properties, but may be useful for finding initial parameter estimates.
  • Nelder–Mead (simplex) search. A simplex in this context is a polytope of n + 1 vertices in n dimensions; a triangle on a plane, a tetrahedron in three-dimensional space and so forth. Each vertex corresponds to a value of the objective function for a particular set of parameters. The shape and size of the simplex is adjusted by varying the parameters in such a way that the value of the objective function at the highest vertex always decreases. Although the sum of squares may initially decrease rapidly, it can converge to a nonstationary point on quasiconvex problems, by an example of M. J. D. Powell.

More detailed descriptions of these, and other, methods are available, in Numerical Recipes, together with computer code in various languages.

See also

References

  1. This implies that the observations are uncorrelated. If the observations are correlated, the expression [math]\displaystyle{ S = \sum_k \sum_j r_k W_{kj} r_j }[/math] applies. In this case the weight matrix should ideally be equal to the inverse of the error variance-covariance matrix of the observations.
  2. In the absence of round-off error and of experimental error in the independent variable the normal equations matrix would be singular
  3. Britzger, Daniel (2022). "The Linear Template Fit". Eur. Phys. J. C 82: 731. doi:10.1140/epjc/s10052-022-10581-w. 
  4. 4.0 4.1 M.J. Box, D. Davies and W.H. Swann, Non-Linear optimisation Techniques, Oliver & Boyd, 1969
  5. This technique was proposed independently by Levenberg (1944), Girard (1958), Wynne (1959), Morrison (1960) and Marquardt (1963). Marquardt's name alone is used for it in much of the scientific literature. See the main article for citation references.
  6. 6.0 6.1 C.L. Lawson and R.J. Hanson, Solving Least Squares Problems, Prentice–Hall, 1974
  7. R. Fletcher, UKAEA Report AERE-R 6799, H.M. Stationery Office, 1971
  8. M. J. D. Powell, Computer Journal, (1964), 7, 155.

Further reading