Ridge regression

Short description: Regularization technique for ill-posed problems

Ridge regression is a method of estimating the coefficients of multiple-regression models in scenarios where the independent variables are highly correlated. It has been used in many fields including econometrics, chemistry, and engineering. Also known as Tikhonov regularization, named for Andrey Tikhonov, it is a method of regularization of ill-posed problems. It is particularly useful to mitigate the problem of multicollinearity in linear regression, which commonly occurs in models with large numbers of parameters. In general, the method provides improved efficiency in parameter estimation problems in exchange for a tolerable amount of bias (see bias–variance tradeoff).

The theory was first introduced by Hoerl and Kennard in 1970 in their Technometrics papers "Ridge regressions: biased estimation of nonorthogonal problems" and "Ridge regressions: applications in nonorthogonal problems". This was the result of ten years of research into the field of ridge analysis.

Ridge regression was developed as a possible solution to the imprecision of least square estimators when linear regression models have some multicollinear (highly correlated) independent variables—by creating a ridge regression estimator (RR). This provides a more precise ridge parameters estimate, as its variance and mean square estimator are often smaller than the least square estimators previously derived.

Overview

In the simplest case, the problem of a near-singular moment matrix [math]\displaystyle{ \mathbf{X}^\mathsf{T}\mathbf{X} }[/math] is alleviated by adding positive elements to the diagonals, thereby decreasing its condition number. Analogous to the ordinary least squares estimator, the simple ridge estimator is then given by [math]\displaystyle{ \hat{\beta}_{R} = \left(\mathbf{X}^{\mathsf{T}} \mathbf{X} + \lambda \mathbf{I}\right)^{-1} \mathbf{X}^{\mathsf{T}} \mathbf{y} }[/math] where [math]\displaystyle{ \mathbf{y} }[/math] is the regressand, [math]\displaystyle{ \mathbf{X} }[/math] is the design matrix, [math]\displaystyle{ \mathbf{I} }[/math] is the identity matrix, and the ridge parameter [math]\displaystyle{ \lambda \geq 0 }[/math] serves as the constant shifting the diagonals of the moment matrix. It can be shown that this estimator is the solution to the least squares problem subject to the constraint [math]\displaystyle{ \beta^\mathsf{T}\beta = c }[/math], which can be expressed as a Lagrangian: [math]\displaystyle{ \min_{\beta} \, \left(\mathbf{y} - \mathbf{X} \beta\right)^\mathsf{T} \left(\mathbf{y} - \mathbf{X} \beta\right) + \lambda \left(\beta^\mathsf{T}\beta - c\right) }[/math] which shows that [math]\displaystyle{ \lambda }[/math] is nothing but the Lagrange multiplier of the constraint. Typically, [math]\displaystyle{ \lambda }[/math] is chosen according to a heuristic criterion, so that the constraint will not be satisfied exactly. Specifically in the case of [math]\displaystyle{ \lambda = 0 }[/math], in which the constraint is non-binding, the ridge estimator reduces to ordinary least squares. A more general approach to Tikhonov regularization is discussed below.

History

Tikhonov regularization was invented independently in many different contexts. It became widely known through its application to integral equations in the works of Andrey Tikhonov and David L. Phillips. Some authors use the term Tikhonov–Phillips regularization. The finite-dimensional case was expounded by Arthur E. Hoerl, who took a statistical approach, and by Manus Foster, who interpreted this method as a Wiener–Kolmogorov (Kriging) filter. Following Hoerl, it is known in the statistical literature as ridge regression, named after ridge analysis ("ridge" refers to the path from the constrained maximum).

Tikhonov regularization

Suppose that for a known matrix [math]\displaystyle{ A }[/math] and vector [math]\displaystyle{ \mathbf{b} }[/math], we wish to find a vector [math]\displaystyle{ \mathbf{x} }[/math] such that [math]\displaystyle{ A\mathbf{x} = \mathbf{b}, }[/math] where [math]\displaystyle{ \mathbf{x} }[/math] and [math]\displaystyle{ \mathbf{b} }[/math] may be of different sizes and [math]\displaystyle{ A }[/math] may be non-square.

The standard approach is ordinary least squares linear regression. However, if no [math]\displaystyle{ \mathbf{x} }[/math] satisfies the equation or more than one [math]\displaystyle{ \mathbf{x} }[/math] does—that is, the solution is not unique—the problem is said to be ill posed. In such cases, ordinary least squares estimation leads to an overdetermined, or more often an underdetermined system of equations. Most real-world phenomena have the effect of low-pass filters in the forward direction where [math]\displaystyle{ A }[/math] maps [math]\displaystyle{ \mathbf{x} }[/math] to [math]\displaystyle{ \mathbf{b} }[/math]. Therefore, in solving the inverse-problem, the inverse mapping operates as a high-pass filter that has the undesirable tendency of amplifying noise (eigenvalues / singular values are largest in the reverse mapping where they were smallest in the forward mapping). In addition, ordinary least squares implicitly nullifies every element of the reconstructed version of [math]\displaystyle{ \mathbf{x} }[/math] that is in the null-space of [math]\displaystyle{ A }[/math], rather than allowing for a model to be used as a prior for [math]\displaystyle{ \mathbf{x} }[/math]. Ordinary least squares seeks to minimize the sum of squared residuals, which can be compactly written as [math]\displaystyle{ \left\|A\mathbf{x} - \mathbf{b}\right\|_2^2, }[/math] where [math]\displaystyle{ \|\cdot\|_2 }[/math] is the Euclidean norm.

In order to give preference to a particular solution with desirable properties, a regularization term can be included in this minimization: [math]\displaystyle{ \left\|A\mathbf{x} - \mathbf{b}\right\|_2^2 + \left\|\Gamma \mathbf{x}\right\|_2^2 }[/math] for some suitably chosen Tikhonov matrix [math]\displaystyle{ \Gamma }[/math]. In many cases, this matrix is chosen as a scalar multiple of the identity matrix ([math]\displaystyle{ \Gamma = \alpha I }[/math]), giving preference to solutions with smaller norms; this is known as L₂ regularization. In other cases, high-pass operators (e.g., a difference operator or a weighted Fourier operator) may be used to enforce smoothness if the underlying vector is believed to be mostly continuous. This regularization improves the conditioning of the problem, thus enabling a direct numerical solution. An explicit solution, denoted by [math]\displaystyle{ \hat{x} }[/math], is given by [math]\displaystyle{ \hat{x} = \left(A^\mathsf{T} A + \Gamma^\mathsf{T} \Gamma\right)^{-1} A^\mathsf{T} \mathbf{b}. }[/math] The effect of regularization may be varied by the scale of matrix [math]\displaystyle{ \Gamma }[/math]. For [math]\displaystyle{ \Gamma = 0 }[/math] this reduces to the unregularized least-squares solution, provided that (A^TA)⁻¹ exists.

L₂ regularization is used in many contexts aside from linear regression, such as classification with logistic regression or support vector machines, and matrix factorization.

Generalized Tikhonov regularization

For general multivariate normal distributions for [math]\displaystyle{ \mathbf x }[/math] and the data error, one can apply a transformation of the variables to reduce to the case above. Equivalently, one can seek an [math]\displaystyle{ \mathbf x }[/math] to minimize

[math]\displaystyle{ \left\|A \mathbf x - \mathbf b\right\|_P^2 + \left\|\mathbf x - \mathbf x_0\right\|_Q^2, }[/math]

where we have used [math]\displaystyle{ \left\|\mathbf{x}\right\|_Q^2 }[/math] to stand for the weighted norm squared [math]\displaystyle{ \mathbf{x}^\mathsf{T} Q \mathbf{x} }[/math] (compare with the Mahalanobis distance). In the Bayesian interpretation [math]\displaystyle{ P }[/math] is the inverse covariance matrix of [math]\displaystyle{ \mathbf b }[/math], [math]\displaystyle{ \mathbf x_0 }[/math] is the expected value of [math]\displaystyle{ \mathbf x }[/math], and [math]\displaystyle{ Q }[/math] is the inverse covariance matrix of [math]\displaystyle{ \mathbf x }[/math]. The Tikhonov matrix is then given as a factorization of the matrix [math]\displaystyle{ Q = \Gamma^\mathsf{T} \Gamma }[/math] (e.g. the Cholesky factorization) and is considered a whitening filter.

This generalized problem has an optimal solution [math]\displaystyle{ \mathbf x^* }[/math] which can be written explicitly using the formula

[math]\displaystyle{ \mathbf x^* = \left(A^\mathsf{T} PA + Q\right)^{-1} \left(A^\mathsf{T} P \mathbf{b} + Q \mathbf{x}_0\right), }[/math]

or equivalently, when Q is not a null matrix:

[math]\displaystyle{ \mathbf x^* = \mathbf x_0 + \left(A^\mathsf{T} P A + Q \right)^{-1} \left(A^\mathsf{T} P \left(\mathbf b - A \mathbf x_0\right)\right). }[/math]

Lavrentyev regularization

In some situations, one can avoid using the transpose [math]\displaystyle{ A^\mathsf{T} }[/math], as proposed by Mikhail Lavrentyev. For example, if [math]\displaystyle{ A }[/math] is symmetric positive definite, i.e. [math]\displaystyle{ A = A^\mathsf{T} \gt 0 }[/math], so is its inverse [math]\displaystyle{ A^{-1} }[/math], which can thus be used to set up the weighted norm squared [math]\displaystyle{ \left\|\mathbf x\right\|_P^2 = \mathbf x^\mathsf{T} A^{-1} \mathbf x }[/math] in the generalized Tikhonov regularization, leading to minimizing [math]\displaystyle{ \left\|A \mathbf x - \mathbf b\right\|_{A^{-1}}^2 + \left\|\mathbf x - \mathbf x_0 \right\|_Q^2 }[/math] or, equivalently up to a constant term, [math]\displaystyle{ \mathbf x^\mathsf{T} \left(A+Q\right) \mathbf x - 2 \mathbf x^\mathsf{T} \left(\mathbf b + Q \mathbf x_0\right). }[/math]

This minimization problem has an optimal solution [math]\displaystyle{ \mathbf x^* }[/math] which can be written explicitly using the formula [math]\displaystyle{ \mathbf x^* = \left(A + Q\right)^{-1} \left(\mathbf b + Q \mathbf x_0\right), }[/math] which is nothing but the solution of the generalized Tikhonov problem where [math]\displaystyle{ A = A^\mathsf{T} = P^{-1}. }[/math]

The Lavrentyev regularization, if applicable, is advantageous to the original Tikhonov regularization, since the Lavrentyev matrix [math]\displaystyle{ A + Q }[/math] can be better conditioned, i.e., have a smaller condition number, compared to the Tikhonov matrix [math]\displaystyle{ A^\mathsf{T} A + \Gamma^\mathsf{T} \Gamma. }[/math]

Regularization in Hilbert space

Typically discrete linear ill-conditioned problems result from discretization of integral equations, and one can formulate a Tikhonov regularization in the original infinite-dimensional context. In the above we can interpret [math]\displaystyle{ A }[/math] as a compact operator on Hilbert spaces, and [math]\displaystyle{ x }[/math] and [math]\displaystyle{ b }[/math] as elements in the domain and range of [math]\displaystyle{ A }[/math]. The operator [math]\displaystyle{ A^* A + \Gamma^\mathsf{T} \Gamma }[/math] is then a self-adjoint bounded invertible operator.

Relation to singular-value decomposition and Wiener filter

With [math]\displaystyle{ \Gamma = \alpha I }[/math], this least-squares solution can be analyzed in a special way using the singular-value decomposition. Given the singular value decomposition

[math]\displaystyle{ A = U \Sigma V^\mathsf{T} }[/math]

with singular values [math]\displaystyle{ \sigma _i }[/math], the Tikhonov regularized solution can be expressed as

[math]\displaystyle{ \hat{x} = V D U^\mathsf{T} b, }[/math]

where [math]\displaystyle{ D }[/math] has diagonal values

[math]\displaystyle{ D_{ii} = \frac{\sigma_i}{\sigma_i^2 + \alpha^2} }[/math]

and is zero elsewhere. This demonstrates the effect of the Tikhonov parameter on the condition number of the regularized problem. For the generalized case, a similar representation can be derived using a generalized singular-value decomposition.

Finally, it is related to the Wiener filter:

[math]\displaystyle{ \hat{x} = \sum _{i=1}^q f_i \frac{u_i^\mathsf{T} b}{\sigma_i} v_i, }[/math]

where the Wiener weights are [math]\displaystyle{ f_i = \frac{\sigma _i^2}{\sigma_i^2 + \alpha^2} }[/math] and [math]\displaystyle{ q }[/math] is the rank of [math]\displaystyle{ A }[/math].

Determination of the Tikhonov factor

The optimal regularization parameter [math]\displaystyle{ \alpha }[/math] is usually unknown and often in practical problems is determined by an ad hoc method. A possible approach relies on the Bayesian interpretation described below. Other approaches include the discrepancy principle, cross-validation, L-curve method, restricted maximum likelihood and unbiased predictive risk estimator. Grace Wahba proved that the optimal parameter, in the sense of leave-one-out cross-validation minimizes

[math]\displaystyle{ G = \frac{\operatorname{RSS}}{\tau^2} = \frac{\left\|X \hat{\beta} - y\right\|^2}{ \left[\operatorname{Tr}\left(I - X\left(X^T X + \alpha^2 I\right)^{-1} X^T\right)\right]^2}, }[/math]

where [math]\displaystyle{ \operatorname{RSS} }[/math] is the residual sum of squares, and [math]\displaystyle{ \tau }[/math] is the effective number of degrees of freedom.

Using the previous SVD decomposition, we can simplify the above expression: [math]\displaystyle{ \operatorname{RSS} = \left\| y - \sum_{i=1}^q (u_i' b) u_i \right\|^2 + \left\| \sum _{i=1}^q \frac{\alpha^2}{\sigma_i^2 + \alpha^2} (u_i' b) u_i \right\|^2, }[/math]

[math]\displaystyle{ \operatorname{RSS} = \operatorname{RSS}_0 + \left\| \sum_{i=1}^q \frac{\alpha^2}{\sigma_i^2 + \alpha^2} (u_i' b) u_i \right\|^2, }[/math]

and

[math]\displaystyle{ \tau = m - \sum_{i=1}^q \frac{\sigma_i^2}{\sigma_i^2 + \alpha^2} = m - q + \sum_{i=1}^q \frac{\alpha^2}{\sigma _i^2 + \alpha^2}. }[/math]

Relation to probabilistic formulation

The probabilistic formulation of an inverse problem introduces (when all uncertainties are Gaussian) a covariance matrix [math]\displaystyle{ C_M }[/math] representing the a priori uncertainties on the model parameters, and a covariance matrix [math]\displaystyle{ C_D }[/math] representing the uncertainties on the observed parameters. In the special case when these two matrices are diagonal and isotropic, [math]\displaystyle{ C_M = \sigma_M^2 I }[/math] and [math]\displaystyle{ C_D = \sigma_D^2 I }[/math], and, in this case, the equations of inverse theory reduce to the equations above, with [math]\displaystyle{ \alpha = {\sigma_D}/{\sigma_M} }[/math].

Bayesian interpretation

Although at first the choice of the solution to this regularized problem may look artificial, and indeed the matrix [math]\displaystyle{ \Gamma }[/math] seems rather arbitrary, the process can be justified from a Bayesian point of view. Note that for an ill-posed problem one must necessarily introduce some additional assumptions in order to get a unique solution. Statistically, the prior probability distribution of [math]\displaystyle{ x }[/math] is sometimes taken to be a multivariate normal distribution. For simplicity here, the following assumptions are made: the means are zero; their components are independent; the components have the same standard deviation [math]\displaystyle{ \sigma _x }[/math]. The data are also subject to errors, and the errors in [math]\displaystyle{ b }[/math] are also assumed to be independent with zero mean and standard deviation [math]\displaystyle{ \sigma _b }[/math]. Under these assumptions the Tikhonov-regularized solution is the most probable solution given the data and the a priori distribution of [math]\displaystyle{ x }[/math], according to Bayes' theorem.

If the assumption of normality is replaced by assumptions of homoscedasticity and uncorrelatedness of errors, and if one still assumes zero mean, then the Gauss–Markov theorem entails that the solution is the minimal unbiased linear estimator.

See also

• LASSO estimator is another regularization method in statistics.
• Elastic net regularization
• Matrix regularization

Further reading

• Gruber, Marvin (1998). Improving Efficiency by Shrinkage: The James–Stein and Ridge Regression Estimators. Boca Raton: CRC Press. ISBN 0-8247-0156-9. https://books.google.com/books?id=wmA_R3ZFrXYC. 
• Kress, Rainer (1998). "Tikhonov Regularization". Numerical Analysis. New York: Springer. pp. 86–90. ISBN 0-387-98408-9. https://books.google.com/books?id=Jv_ZBwAAQBAJ&pg=PA86. 
• Press, W. H.; Teukolsky, S. A.; Vetterling, W. T.; Flannery, B. P. (2007). "Section 19.5. Linear Regularization Methods". Numerical Recipes: The Art of Scientific Computing (3rd ed.). New York: Cambridge University Press. ISBN 978-0-521-88068-8. http://apps.nrbook.com/empanel/index.html#pg=1006. 
• Saleh, A. K. Md. Ehsanes; Arashi, Mohammad; Kibria, B. M. Golam (2019). Theory of Ridge Regression Estimation with Applications. New York: John Wiley & Sons. ISBN 978-1-118-64461-4. https://books.google.com/books?id=v0KCDwAAQBAJ. 
• Taddy, Matt (2019). "Regularization". Business Data Science: Combining Machine Learning and Economics to Optimize, Automate, and Accelerate Business Decisions. New York: McGraw-Hill. pp. 69–104. ISBN 978-1-260-45277-8. https://books.google.com/books?id=yPOUDwAAQBAJ&pg=PA69.