Optimal selection of reduced rank estimators of high-dimensional matrices
Florentina Bunea, Yiyuan She, Marten H. Wegkamp
Introduction
where is a random matrix, with independent entries with mean zero and variance .
Standard least squares estimation in (1), under no constraints, is equivalent to regressing each response on the predictors separately. It completely ignores the multivariate nature of the possibly correlated responses, see, for instance, Izenman (2008) for a discussion of this phenomenon. Estimators restricted to have rank equal to a fixed number were introduced to remedy this drawback. The history of such estimators dates back to the 1950’s, and was initiated by Anderson (1951). Izenman (1975) introduced the term reduced-rank regression for this class of models and provided further study of the estimates. A number of important works followed, including Robinson (1973, 1974) and Rao (1978). The monograph on reduced rank regression by Reinsel and Velu (1998) has an excellent, comprehensive account of more recent developments and extensions of the model. All theoretical results to date for estimators of constrained to have rank equal to a given value are of asymptotic nature and are obtained for fixed , independent of the number of observations . Most of them are obtained in a likelihood framework, for Gaussian errors . Anderson (1999) relaxed this assumption and derived the asymptotic distribution of the estimate, when is fixed, the errors have two finite moments, and the rank of is known. Anderson (2002) continued this work by constructing asymptotic tests for rank selection, valid only for small and fixed values of .
The aim of our work is to develop a non-asymptotic class of methods that yield reduced rank estimators of that are easy to compute, have rank determined adaptively from the data, and are valid for any values of and , especially when the number of predictors is large. The resulting estimators can then be used to construct a possibly much smaller number of new transformed predictors or can be used to construct the most important canonical variables based on the original and . We refer to Chapter 6 in Izenman (2008) for a historical account of the latter.
We propose to estimate by minimizing the sum of squares plus a penalty , proportional to the rank , over all matrices . It is immediate to see, using Pythagoras’ theorem, that this is equivalent with computing or , with being the projection matrix onto the column space of . In Section 2.1 we show that the minimizer of the above expression is the number of singular values of that exceed . This observation reveals the prominent role of the tuning parameter in constructing . The final estimator of the target matrix is the minimizer of over matrices of rank , and can be computed efficiently even for large , using the procedure that we describe in detail in Section 2.1 below.
The theoretical analysis of our proposed estimator is presented in Sections 2.2 – 2.4. The rank of may not be the most appropriate measure of sparsity in multivariate regression models. For instance, suppose that the rank of is 100, but only three of its singular values are large and the remaining 97 are nearly zero. This is an extreme example, and in general one needs an objective method for declaring singular values as “large” or “small”. We introduce in Section 2.1 a slightly different notion of sparsity, that of effective rank. The effective rank counts the number of singular values of the signal that are above a certain noise level. The relevant notion of noise level turns out to be the largest singular value of . This is central to our results, and influences the choice of the tuning sequence . In Appendix C we prove that the expected value of the largest singular value of is bounded by , where is the rank of . The effective noise level is at most , for instance in the model , but it can be substantially lower, of order , in model (1).
In Section 2.2 we give tight conditions under which , the rank of our proposed estimator , coincides with the effective rank. As an immediate corollary we show when equals the rank of . We give finite sample performance bounds for in Section 2.3. These results show that mimics the behavior of reduced rank estimates based on the ideal effective rank, had this been known prior to estimation. If has a restricted isometrity property, our estimate is minimax adaptive. In the asymptotic setting, for , all our results hold with probability close to one, for tuning parameter chosen proportionally to the square of the noise level.
We often particularize our main findings to the setting of Gaussian errors in order to obtain sharp, explicit numerical constants for the penalty term. To avoid technicalities, we assume that is known in most cases, and we treat the case of unknown in Section 2.4.
We contrast our estimator with the penalized least squares estimator corresponding to a penalty term proportional to the nuclear norm , the sum of the singular values of . This estimator has been studied by, among others, Yuan et al. (2007) and Lu et al (2010), for model (1). Nuclear norm penalized estimators in general models y=\mbox{\mathcal{X}}(A)+\varepsilon involving linear maps have been studied by Candès and Plan (2010) and Negahban and Wainwright (2009). A special case of this model is the challenging matrix completion problem, first investigated theoretically, in the noiseless case, by Candès and Tao (2010). Rohde and Tsybakov (2010) studied a larger class of penalized estimators, that includes the nuclear norm estimator, in the general model y=\mbox{\mathcal{X}}(A)+\varepsilon.
In Section 3 we give bounds on that are similar in spirit to those from Section 2. While the error bounds of the two estimators are comparable, albeit with cleaner results and milder conditions for our proposed estimator, there is one aspect in which the estimates differ in important ways. The nuclear norm penalized estimator is far less parsimonious than the estimate obtained via our rank selection criterion. In Section 3, we offer a correction of the former estimate that yields a correct rank estimate.
Section 4 complements our theoretical results by an extensive simulation study that supports our theoretical findings and suggests strongly that the proposed estimator behaves very well in practice, in most situations is preferable to the nuclear norm penalized estimator and it is always much faster to compute.
Technical results and some intermediate proofs are presented in Appendices A – D.
The Rank Selection Criterion
We propose to estimate by the penalized least squares estimator
We denote its rank by . The minimization is taken over all matrices . Here and in what follows is the rank of and denotes the Frobenius norm for any generic matrix . The choice of the tuning parameter is discussed in Section 2.2. Since
one needs to compute the restricted rank estimators that minimize over all matrices of rank . The following computationally efficient procedure for calculating each has been suggested by Reinsel and Velu (1998). Let be the Gram matrix, be its Moore-Penrose inverse and let be the projection matrix onto the column space of .
Compute the eigenvectors , corresponding to the ordered eigenvalues arranged from largest to smallest, of the symmetric matrix .
Compute the least squares estimator . Construct and . Form and .
Compute the final estimator .
In step 2 above, denotes the matrix obtained from by retaining all its rows and only its first columns, and is obtained from by retaining its first rows and all its columns.
Our first result, Proposition 1 below, characterizes the minimizer of (3) as the number of eigenvalues of the square matrix that exceed or, equivalently, as the number of singular values of the matrix that exceed . The final estimator of is then .
Lemma 14 in Appendix B shows that the fitted matrix is equal to based on the singular value decomposition of the projection .
Let be the ordered eigenvalues of . We have with
For given above, and by the Pythagorean theorem, we have
and we observe that . By Lemma 14 in Appendix B, we have
where denotes the -th largest singular value of a matrix . Then, the penalized least squares criterion reduces to
and we find that equals
It is easy to see that is minimized by taking as the largest index for which , since then the sum only consists of negative terms. This concludes our proof. ∎
Remark. The two matrices and , that yield the final solution , have the following properties: (i) is the identity matrix; and (ii) is a diagonal matrix. Moreover, the decomposition of as a product of two matrices with properties (i) and (ii) is unique, see, for instance, Theorem 2.2 in Reinsel and Vélu (1998). As an immediate consequence, one can construct new orthogonal predictors as the columns of . If is much smaller than , this can result in a significant dimension reduction of the predictors’ space.
2 Consistent effective rank estimation
In this section we study the properties of . We will state simple conditions that guarantee that equals with high probability. First, we describe in Theorem 2 what estimates and what quantities need to be controlled for consistent estimation. It turns out that estimates the number of the singular values of the signal above the threshold , for any value of the tuning parameter . The quality of estimation is controlled by the probability that this threshold level exceeds the largest singular value of the projected noise matrix . We denote the th singular value of a generic matrix by and we use the convention that the singular values are indexed in decreasing order.
Suppose that there exists an index such that
Using the characterization of given in Proposition 1 we have
Invoke the conditions on and to complete the proof. ∎
Theorem 2 indicates that we can consistently estimate the index provided we use a large enough value for our tuning parameter to guarantee that the probability of the event approaches one. We call the effective rank of relative to , and denote it by .
Let and assume that are independent random variables. Then
In view of this result, we take as our measure of the noise level. The following corollary summarizes the discussion above and lists the main results of this section: the proposed estimator based on the rank selection criterion (RSC) recovers consistently the effective rank and, in particular, the rank of .
Assume that has independent entries. For any , set
with as in Theorem 2. Then we have, for any ,
In particular, if and , then
Remark. Corollary 4 holds when . If stays bounded, but , the consistency results continue to hold when is replaced by in the expression of the tuning parameter given above. Lemma 3 justifies this choice. The same remark applies to all theoretical results in this paper.
for all . The conclusion of Corollary 4 then holds for with large enough. Moreover, all oracle inequalities presented in the next sections remain valid for this choice of the tuning parameter, if has independent subGaussian entries.
3 Errors bounds for the RSC estimator
In this section we study the performance of by obtaining bounds for . First we derive a bound for the fit , based on the restricted rank estimator , for each value of .
Set . For any , we have
for all matrices of rank . Working out the squares we obtain
for generic matrices and . The inner product , operator norm and nuclear norm are related via the inequality . As a consequence we find
Using the inequality with twice, we obtain that is bounded above by
Hence we obtain, for any , the inequality
Lemma 14 in the Appendix B states that the minimum of over all matrices of rank is achieved for the GSVD of and the minimum equals . The claim follows after choosing and . ∎
Assume that has independent entries. Set . Then, for any , the inequality
holds with probability . In addition,
The symbol means that the inequality holds up to multiplicative numerical constants.
Set for some . From Lemma 3, it follows that
Theorem 5 bounds the error by an approximation error, , and a stochastic term, , with probability one. The approximation error is decreasing in and vanishes for .
We observe that is essentially the number of free parameters of the restricted rank problem. Indeed, our parameter space consists of all matrices of rank and each matrix has free parameters. Hence we can interpret the bound in Corollary 6 above as the squared bias plus the dimension of the parameter space.
Remark(ii), following Corollary 8 below, shows that is also the minimax lower bound for , if the smallest eigenvalue of is larger than a strictly positive constant. This means that is a minimax estimator under this assumption.
We now turn to the penalized estimator and show that it achieves the best (squared) bias-variance trade-off among all rank restricted estimators for the appropriate choice of the tuning parameter in the penalty .
We have, for any , on the event ,
for any matrix . In particular, we have, for
for all matrices . Working out the squares we obtain
Consequently, using the inequality twice, we obtain, for any and ,
Hence, if , we obtain
for any and . Lemma 14 in Appendix B evaluates the minimum of over all matrices of rank and shows that it equals . We conclude our proof by choosing and . ∎
Remark. The first two parts of the theorem show that achieves the best (squared) bias-variance trade-off among all reduced rank estimators if . Moreover, the index which minimizes essentially coincides with the effective rank defined in the previous section. Therefore, the fit of the selected estimator is comparable with that of the estimator with rank . Since the ideal depends on the unknown matrix , this ideal estimator cannot be computed. Although our estimator is constructed independently of , it mimics the behavior of the ideal estimator and we say that the bound on adapts to .
The last part of our result is a particular case of the second part, but it is perhaps easier to interpret. Taking the index equal to the rank , the bias term disappears and the bound reduces to up to constants. This shows clearly the important role played by in the estimation accuracy: the smaller the rank of , the smaller the estimation error.
For Gaussian errors, we have the following precise bounds.
Assume that has independent entries. Set
with arbitrary. Let . Then, we have
Apply Lemma 16 in Appendix D to deduce that
Remarks. (i) We note that for large,
(iii) The same type of upper bound as the one of Corollary 8 can be proved if the entries of are subGaussian: take for some large enough, and invoke Proposition 15 in Appendix C.
(iv) Although the error bounds of are guaranteed for all and , the analysis of the estimation performance of depends on . If , for some constant , then, provided with arbitrary,
(v) Our results are slightly more general than stated. In fact, our analysis does not require that the postulated multivariate linear model holds exactly. We denote the expected value of by and write . We denote the projection of onto the column space of by , that is, . Because minimizing is equivalent with minimizing by Pythagoras’ theorem, our least squares procedure estimates , the mean of . The statements of Theorems 2 and 7 remain unchanged, except that is the mean of the projection of , not the mean of itself.
4 A data adaptive penalty term
In this section we construct a data adaptive penalty term that employs the unbiased estimator
of . Set, for any , and ,
Notice that the estimator requires that be large, which holds whenever or and is large. The challenging case is left for future research.
Assume that is an matrix with independent entries. Using the penalty given above we have, for ,
It remains to bound the expected value of
We split the expectation into two parts: and its complement. We observe first that
using Lemma 16 for the last inequality. Next, we observe that
using Lemmas 16 and 17 in Appendix D for the last inequality. This proves the result. ∎
Remark. We see that for large values of and ,
as the additional terms in the theorem above decrease exponentially fast in and . This bound is similar to the one in Corollary 8, obtained for the RSC estimator corresponding to the penalty term that employs the theoretical value of .
Comparison with nuclear norm penalized estimators
In this section we compare our RSC estimator with the alternative estimator that minimizes
On the event , we have, for any ,
for all matrices . Working out the squares we obtain
on the event , we obtain the claim using the triangle inequality. ∎
We see that balances the bias term with the penalty term , provided . Since , we have . We immediately obtain the following corollary using the results for of Lemma 3.
Assume that has independent entries. For
The same result, up to constants, can be obtained if the errors are subGaussian, if we replace in the choice of above by a suitably large constant . The proof of this generalization uses Proposition 15 in Appendix C in lieu of Lemma 3. The same remark applies for all the results in this section.
The next result obtains an oracle inequality for that resembles the oracle inequality for the RSC estimator in Theorem 7. We stress the fact that Theorem 12 below requires that ; this was not required for the derivation of the oracle bound on in Theorem 7, which holds for all . We denote the condition number of by .
Assume that has independent entries. For
Both inequalities hold with probability at least . The symbol means that the inequality holds up to multiplicative numerical constants (depending on ).
To keep the paper self contained, we give a simple proof of this result in Appendix A. Similar results for the NNP estimator of in the general model y=\mbox{\mathcal{X}}(A)+\varepsilon, where is a random linear map, have been obtained by Negahban and Wainwright (2009) and Candès and Plan (2010), each under different sets of assumptions on . We refer to Rohde and Tsybakov (2010) for more general results on Schatten norm penalized estimators of in the model y=\mbox{\mathcal{X}}(A)+\varepsilon, and a very thorough discussion on the assumptions on under which these results hold.
Theorem 10 shows that the error bounds of the nuclear norm penalized (NNP) estimator and the RSC estimator are comparable, although it is worth pointing out that our bounds for are much cleaner and obtained under fewer restrictions on the design matrix. However, there is one aspect in which the two estimators differ radically: correct rank recovery. We showed in Section 2.2 that the RSC estimator corresponding to the effective value of the tuning sequence has the correct rank and achieves the optimal bias-variance trade-off. This is also visible in the left panel of Figure 1 which shows the plots of the MSE and rank of the RSC estimate as we varied the tuning parameter of the procedure over a large grid. The numbers on the vertical axis correspond to the range of values of the rank of the estimator considered in this experiment, 1 to 25. The rank of is 10. We notice that for the same range of values of the tuning parameter, RSC has both the smallest MSE value and the correct rank. We repeated this experiment for the NNP estimator. The right panel shows that the smallest MSE and the correct rank are not obtained for the same value of the tuning parameter. Therefore, a different strategy for correct rank estimation via NNP is in order.
Rather than taking the rank of as the estimator of the rank of , we consider instead, for ,
Let and assume that . Then
If has independent entries and , the above probability is bounded by .
Empirical Studies
We performed an extensive simulation study to evaluate the performance of the proposed method, RSC, and compare it with the NNP method. The RSC estimator was computed via the procedure outlined in Section 2.1. This method is computationally efficient in large dimensions. Its computational complexity is the same as that of PCA. Our choice for the tuning parameter was based on our theoretical findings in Section 2. In particular, Corollary 4 and Corollary 8 guarantee good rank selection and prediction performance of RSC provided that is just a little bit larger than . Under the assumption that , we can estimate by ; see Section 2.4 for details. In our simulations we used the adaptive tuning parameter . We experimented with other constants and found that the constant equal to 2 was optimal; constants slightly larger than 2 gave very similar results.
We compared the RSC estimator with the NNP estimator and with the proposed trimmed or calibrated NNP estimator, denoted in what follows by NNP(c). The NNP estimator is the minimizer of the convex criterion By the equivalent SDP characterization of the NNP-norm given in Fazel (2002), the original minimization problem is equivalent to the convex optimization problem
In our simulation study we compared the rank selection and the estimation performances of the RSC estimator RSC, corresponding to , with the optimally tuned RSC estimator, and the optimally tuned NNP and NNP(c) estimators. The last three estimators are called RSC, NNP and NNP. They correspond to those tuning parameters , and , respectively, that gave the best prediction accuracy, when prediction was evaluated on a very large independent validation set. This comparison helps us understand the true potential of each method in an ideal situation, and allows us to draw a stable performance comparison between the proposed adaptive RSC estimator and the best possible versions of RSC and NNP.
We considered the following large sample-size set up and large dimensionality set up.
We constructed the matrix of dependent variables by generating its rows as i.i.d. realizations from a multivariate normal distribution , with , , . The coefficient matrix , with , is a matrix and is a matrix. All entries in and are i.i.d. . Each row in is then generated as , , with denoting the -th row of the noise matrix which has independent entries .
Experiment 2 (p>m(>q)𝑝annotated𝑚absent𝑞p>m(>q))
Each simulated model is characterized by the following control parameters: (sample size), (number of independent variables), (number of response variables), (rank of ), (design correlation), (rank of the design), and (signal strength). In Experiment 1, we set , and varied the correlation coefficient and signal strength . All combinations of correlation and signal strength are covered in the simulations. The results are summarized in Table 1. In Experiment 2, we set , , , , , and varied the correlation and signal strength . The corresponding results are reported in Table 2. In both tables, MSE() and MSE() denote the trimmed-means of and , respectively. We also report the median rank estimates (RE) and the successful rank recovery percentages (RRP).
(i) We found that the RSC estimator corresponding to the adaptive choice of the tuning parameter has excellent performance. It behaves as well as the RSC estimator that uses the parameter tuned on the large validation set or the RSC estimator corresponding to the theoretical .
(ii) When the signal-to-noise ratio SNR := is moderate or high, with values approximately 1, 1.5 and 2, corresponding to , and for low to moderate correlation between the predictors (), RSC has excellent behavior in terms of rank selection and means squared errors. Interestingly, NNP does not have optimal behavior in this set-up: its mean squared errors are slightly higher than those of the RSC estimator. When the noise is very large relative to the signal strength, corresponding to in Table 1, or when the correlation between some covariates is very high, in Table 1, NNP may be slightly more accurate than the RSC.
(iii) The NNP does not recover the correct rank, when its regularization parameter is tuned by validation. Both Tables 1 and 2 show that the correct rank ( in Experiment 1 and in Experiment 2) is overestimated by NNP. Our trimmed estimator, NNP(c), provides a successful improvement over NNP in this respect. This supports Theorem 13.
In additional simulations, we found that especially for low or moderate SNRs, the NNP parameter tuning problem is much more challenging than the RSC parameter tuning. NNP cannot accurately estimate and consistently select the rank at the same time, for the same value of the tuning parameter. This echoes the findings presented in Figure 1, and is to be expected: in NNP regularization, the threshold value also controls the amount of shrinkage, which should be mild for large samples with relatively low contamination. This is the case for moderate SNR and moderate correlation between predictors: the tuned tends to be too small, so it cannot introduce enough sparsity. The same continues to be true for slightly larger values of that compensate for high noise level and very high correlation between predictors. In summary, one may not be able to build an accurate and parsimonious model via the NNP method, without further adjustments.
Overall, RSC is recommended over the NNP estimators, especially when we suspect that the SNR is not very low. With large validation tuning, NNP(c) has the same properties as RSC – they coincide when both methods select the same rank. But in general, the rank estimation via NNP(c) is much more difficult to tune and much more computationally involved than RSC.
For data with low SNR, an immediate extension of the RSC estimator that involves a second penalty term, of ridge-type, may induce the right amount of shrinkage needed to offset the noise in the data. This conjecture will be investigated carefully in future research.
2 Tightness of the rank consistency results
It can be shown, using arguments similar to those used in the proof of Theorem 2, that
On the other hand, the proof of Theorem 2 reveals that
Appendix A Proof of Theorem 12
that holds on the event . The inequality can be deduced from the proof of Theorem 10. Then, by Lemmas 3.4 and 2.3 in Recht et al (2007) there exist two matrices and such that
.
Using and , we obtain
The proof is complete by choosing the truncated GSVD under metric , see Lemma 14 below. ∎
Appendix B Generalized singular value decomposition
with and is a fixed matrix of rank . By the Eckhart-Young theorem, we have the lower bound
for all matrices of rank . We now show that this infimum is achieved by the generalized singular value decomposition (GSVD) under metric , limited to its largest generalized singular values. Following Takane and Hunter (2001, pages 399-400), the GSVD of under metric is where is an matrix, , is an matrix, and is a diagonal matrix, and It can be computed via the (regular) SVD of . From , the generalized singular values are the regular singular values of . Let by retaining as usual the first columns of and .
Let be the GSVD of under metric , restricted to the largest generalized singular values. We have
Since and , we obtain
using the notation for the matrix consisting of the last column vectors of , is the diagonal matrix based on the last singular values, and for the matrix consisting of the last column vectors of . Finally,
Recall that in the construction of the GSVD, the generalized singular values are the singular values of . Since
Remark. The rank restricted estimator given in Section 2.1 is the GSVD of the least squares estimator under the metric , see Takane and Hwang (2007).
Appendix C Largest singular values of transformations of subGaussian matrices
We call a random variable subGaussian with subGaussian moment , if
for all . Markov’s inequality implies that has Gaussian type tails:
holds for any . Normal random variables are subGaussian with . General results on the largest singular values of matrices with subGaussian entries can be found in the survey paper by Rudelson and Vershynin (2010). The analysis of our estimators require bounds for the largest singular values of and , for which the standard results on do not apply directly.
Let be a matrix with independent subGaussian entries with subGaussian moment . Let be an matrix of rank and let be the projection matrix on . Then, for each ,
with . Let be a -net of and be a -net for with . Since the dimension of is and for each , we need at most elements in to cover and elements to cover , see Kolmogorov and Tikhomirov (1961). A standard discretization trick, see, for instance, Rudelson and Vershynin (2010, proof of Proposition 2.4), gives
Next, we write and note that each is subGaussian with moment , as
It follows that each term in is subGaussian, and is subGaussian with subGaussian moment . This implies the tail bound
for each fixed and and all . Combining the previous two steps, we obtain
for all . Taking we obtain the first claim. The second claim follows from this tail bound. ∎
Appendix D Auxiliary results
The following string of inequalities are self-evident:
This proves our first claim. The second claim is easily deduced as follows:
Let be a random variable with degrees of freedom. Then
See Cavalier et al (2002, page 857) for the first claim. The second claim follows by taking . ∎
Acknowledgement. We would like to thank Emmanuel Candès, Angelika Rohde and Sasha Tsybakov for stimulating conversations in Oberwolfach, Tallahassee and Paris, respectively. We also thank the associate editor and the referees for their constructive remarks.