Tutorials › Multivariate Statistics (Graduate) › Nonlinear Eigenvalue Shrinkage Under Frobenius Loss

High-Dimensional Covariance Estimation · Tutorial 785 of 975

Nonlinear Eigenvalue Shrinkage Under Frobenius Loss

Learn why rotation equivariance preserves sample eigenvectors, how Frobenius loss determines oracle eigenvalues, and why the limiting correction is one under identity covariance.

Advanced 12 min read

What You'll Learn

  • Express Frobenius loss in the sample covariance eigenbasis
  • Derive the oracle eigenvalue for each sample eigenvector
  • Explain how rotation equivariance restricts covariance estimators
  • State the Marchenko–Pastur limit for identity population covariance
  • Verify the limiting nonlinear correction throughout the spectral bulk
  • Distinguish rank-induced zero eigenvalues from zeros due to a singular population covariance

Why Change Eigenvalues but Keep Eigenvectors?

In high dimensions, sample covariance eigenvalues can be substantially distorted even when the sample eigenvectors provide a useful coordinate system. A rotation-equivariant estimator respects that coordinate system: it transforms in the same way as the sample covariance when the variables are rotated. Under squared Frobenius loss, the key question is then how to choose an estimated variance along each sample eigenvector.

The answer starts with an oracle calculation. If the population covariance were known, we could find the best estimator among matrices diagonal in the sample eigenbasis. This oracle is not generally available as an estimator, because it uses the unknown population covariance. It does, however, identify the target eigenvalues that data-driven nonlinear shrinkage methods seek to estimate.

Definition: Let \(S\) be a \(p\times p\) sample covariance matrix with spectral decomposition \(S=U\operatorname{diag}(\lambda_1,\ldots,\lambda_p)U^T\), where \(U=(u_1,\ldots,u_p)\) is orthogonal. A spectral covariance estimator based on \(S\) has the form \(\hat{\Sigma}=U\operatorname{diag}(d_1,\ldots,d_p)U^T\). Its estimated eigenvalues \(d_i\) may depend on all the sample eigenvalues. Under Frobenius loss, the loss is \(\|\hat{\Sigma}-\Sigma\|_F^2=\operatorname{tr}\{(\hat{\Sigma}-\Sigma)^T(\hat{\Sigma}-\Sigma)\}\).

For a matrix function to be rotation-equivariant, it must satisfy \(T(QSQ^T)=QT(S)Q^T\) for every orthogonal \(Q\). When \(S\) has distinct eigenvalues, this condition forces \(T(S)\) to be diagonal in the eigenbasis of \(S\). To see why, consider changing the sign of one eigenvector: this leaves \(S\) unchanged, so equivariance requires the output to be unchanged too. Any off-diagonal entry involving that eigenvector would change sign and must therefore be zero. If an eigenvalue is repeated, rotations within its eigenspace leave \(S\) unchanged; equivariance requires the output to be a scalar multiple of the identity within that eigenspace. Thus spectral estimators retain the sample eigenspaces, while choosing new eigenvalues within them.

The Frobenius-Loss Oracle

Write the population covariance in sample coordinates as \(A=U^T\Sigma U\). Orthogonal invariance of the Frobenius norm gives \(\|\hat{\Sigma}-\Sigma\|_F^2=\|\operatorname{diag}(d_1,\ldots,d_p)-A\|_F^2\). The diagonal and off-diagonal entries contribute separately. In particular, the choice of \(d_i\) affects only the squared difference between \(d_i\) and \(A_{ii}\).

Theorem: For a fixed orthogonal matrix \(U\) and a fixed symmetric population covariance \(\Sigma\), the unique minimizer over real \(d_1,\ldots,d_p\) of \(\|U\operatorname{diag}(d_1,\ldots,d_p)U^T-\Sigma\|_F^2\) is \(d_i^{\mathrm{or}}=u_i^T\Sigma u_i\). If \(\Sigma\) is positive semidefinite, each oracle eigenvalue is nonnegative.

Proof. Put \(A=U^T\Sigma U\). The Frobenius norm is unchanged by multiplication on either side by an orthogonal matrix, so the loss equals \(\|\operatorname{diag}(d_1,\ldots,d_p)-A\|_F^2\). Expanding it entry by entry gives \[ \sum_{i=1}^p(d_i-A_{ii})^2+\sum_{i\ne j}A_{ij}^2. \] The second sum does not depend on any \(d_i\). Each term in the first sum is uniquely minimized at \(d_i=A_{ii}=u_i^T\Sigma u_i\). If \(\Sigma\) is positive semidefinite, then \(u_i^T\Sigma u_i\geq 0\), as required. \(\square\)

The oracle eigenvalue is the population variance in the direction of a sample eigenvector; it is not, in general, the population eigenvalue matched by rank. The sample eigenvalue \(\lambda_i\) does not appear in the pointwise minimization. It matters in practice because an estimator must infer the oracle target from the observed spectrum and its relation to the sample eigenvectors.

Worked Example: Minimizing the Loss in a Sample Eigenbasis

Suppose the sample eigenvectors are \(u_1=(1,1)^T/\sqrt{2}\) and \(u_2=(-1,1)^T/\sqrt{2}\), and the population covariance is \(\Sigma=\begin{pmatrix}3&1\\1&2\end{pmatrix}\). Find the Frobenius-loss oracle eigenvalues and the minimum loss among estimators diagonal in this basis.

The diagonal entries of \(U^T\Sigma U\) are the two directional variances: \(u_1^T\Sigma u_1=(3+1+1+2)/2=7/2\) and \(u_2^T\Sigma u_2=(3-1-1+2)/2=3/2\). The off-diagonal entry is \(u_1^T\Sigma u_2=(-3+1-1+2)/2=-1/2\).

Consequently, the oracle estimate is \(U\operatorname{diag}(7/2,3/2)U^T\). The loss decomposition shows that its minimum loss is just the squared off-diagonal contribution: \(2(-1/2)^2=1/2\). Choosing any other pair \(d_1,d_2\) adds \((d_1-7/2)^2+(d_2-3/2)^2\) to this loss.

Identity Population Covariance: The Limiting Correction

Now suppose the observations are independent \(N(0,I_p)\) vectors and \(S=XX^T/n\), where \(X\) is a \(p\times n\) matrix of independent standard normal entries. Let \(p/n\to c\), with \(0<c<1\). The population covariance \(I_p\) is positive definite. Since the Gaussian data matrix has full row rank with probability one when \(p<n\), there are no rank-induced zero sample eigenvalues in this setting. This qualification matters: for a singular population covariance, zero sample eigenvalues can occur even when \(p/n<1\). In a general limiting model, keeping the population spectrum bounded away from zero is an additional condition if one wants the limiting sample spectrum bounded away from zero.

For this identity model, every unit vector \(u_i\) satisfies \(u_i^TI_pu_i=1\). Thus the oracle eigenvalue is exactly one for every sample eigenvector, at every finite dimension. The sample eigenvalues themselves do not concentrate at one: their limiting empirical distribution is the Marchenko–Pastur distribution. The following calculation shows explicitly how the nonlinear spectral correction recovers the oracle value throughout the limiting bulk.

Theorem: In the identity population model above, the limiting sample eigenvalue distribution has support \([a,b]\), where \(a=(1-\sqrt c)^2\) and \(b=(1+\sqrt c)^2\), and density \(f_c(x)=\sqrt{(b-x)(x-a)}/(2\pi c x)\) for \(a<x<b\). Define the Stieltjes transform by \(m(z)=\int (t-z)^{-1}\,dF(t)\). For an interior bulk value \(x\), define the spectral correction \(\delta(x)=x/|1-c-cx\,m(x+i0)|^2\). Then \(\delta(x)=1\) for every \(a<x<b\).

Proof. The boundary value of the Stieltjes transform for the Marchenko–Pastur density, with the stated sign convention, is \[ m(x+i0)=\frac{1-c-x+i\sqrt{(b-x)(x-a)}}{2cx},\qquad a<x<b. \] Substitution gives \[ 1-c-cx\,m(x+i0) =\frac{1-c+x-i\sqrt{(b-x)(x-a)}}{2}. \] Therefore \[ |1-c-cx\,m(x+i0)|^2 =\frac{(1-c+x)^2+(b-x)(x-a)}{4}. \] To simplify the numerator, use \(a+b=2(1+c)\) and \(ab=(1-c)^2\). Expanding the product yields \((b-x)(x-a)=-x^2+(a+b)x-ab\). Adding \((1-c+x)^2=x^2+2(1-c)x+(1-c)^2\) cancels the \(x^2\) and constant terms. The coefficient of \(x\) is \(2(1-c)+(a+b)=2(1-c)+2(1+c)=4\). Hence the sum is \(4x\), so the squared modulus is \(x\) and \(\delta(x)=x/x=1\). This agrees with the exact finite-sample oracle \(u_i^TI_pu_i=1\). \(\square\)

Worked Example: Evaluating the Correction at an Interior Eigenvalue

Take \(c=1/4\). Then \(a=1/4\) and \(b=9/4\), so \(x=1\) is inside the limiting spectral support. Compute the correction at this value.

At \(x=1\), \(\sqrt{(b-x)(x-a)}=\sqrt{(5/4)(3/4)}=\sqrt{15}/4\). The boundary transform is \(m(1+i0)=(-1/4+i\sqrt{15}/4)/(1/2)=-1/2+i\sqrt{15}/2\).

It follows that \(1-c-cx\,m=3/4-(1/4)(-1/2+i\sqrt{15}/2)=7/8-i\sqrt{15}/8\). Its squared modulus is \(49/64+15/64=1\). Thus \(\delta(1)=1/1=1\), as predicted.

Worked Example: Sample Eigenvalues Need Not Be Oracle Eigenvalues

Continue with \(c=1/4\). The asymptotic sample spectrum spans values from \(1/4\) to \(9/4\), whereas the population covariance has every eigenvalue equal to one. Consider two possible interior sample eigenvalues, \(x=1/2\) and \(x=2\). Both lie strictly between the support endpoints.

For either value, the theorem gives \(|1-c-cx\,m(x+i0)|^2=x\). Therefore the nonlinear correction is \(x/x=1\) at \(x=1/2\), and also \(x/x=1\) at \(x=2\). The two sample eigenvalues are different, but both map to the same oracle target. This is not a claim that the observed eigenvalues equal one; it is a correction for their sampling distortion under the specified identity model.

Interpretation, Limits, and a Common Pitfall

The identity calculation is a transparent benchmark for nonlinear shrinkage. The oracle uses the sample eigenvectors but assigns the same population variance to every direction. In more general covariance models, the oracle values \(u_i^T\Sigma u_i\) need not be equal, and their limiting relationship to sample eigenvalues depends on the population spectrum and the eigenvector–eigenvalue relationship. The identity result should not be read as a universal rule to replace all sample eigenvalues by one: it is specific to identity population covariance and Frobenius loss.

A common mistake is to assume that the eigenvalue with rank \(i\) in the sample must estimate the population eigenvalue with the same rank. The Frobenius oracle calculation shows why this need not be true: it targets the population variance in the sample eigenvector direction, not a rank-matched population eigenvalue. Another mistake is to claim that \(c<1\) alone rules out zero sample eigenvalues for every population covariance. It rules out rank-induced zeros when the population covariance is positive definite and the data matrix has full row rank; a singular population covariance can still produce zeros. The limiting correction derived here uses a nonsingular identity population and interior points of its limiting spectral support.

Key takeaway: Under Frobenius loss, the oracle eigenvalue attached to a sample eigenvector is its population directional variance \(u_i^T\Sigma u_i\). In the identity population model, this oracle is exactly one, and the Marchenko–Pastur spectral correction reduces to one throughout the bulk.

Check Your Understanding

Use the definitions and derivations above to answer these questions.

  1. Why does the Frobenius-loss minimization separate into one term for each estimated eigenvalue and a term that cannot be changed by those eigenvalues?
  2. What population quantity does the oracle eigenvalue \(d_i^{\mathrm{or}}\) represent?
  3. For the stated Stieltjes-transform convention, what is the imaginary part of \(m(x+i0)\) inside the Marchenko–Pastur support?
  4. Why can a sample eigenvalue differ substantially from one in the identity population model even though its oracle correction is one?
  5. Give a condition under which \(c<1\) prevents rank-induced zero sample eigenvalues, and explain why that condition is needed.
Clinical Biostats