Optimal selection of reduced rank estimators of high-dimensional matrices

Florentina Bunea, Yiyuan She, Marten H. Wegkamp

Introduction

where EE is a random m×nm\times n matrix, with independent entries with mean zero and variance σ2\sigma^{2}.

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 k≤n∧pk\leq n\wedge p 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 AA constrained to have rank equal to a given value kk are of asymptotic nature and are obtained for fixed pp, independent of the number of observations mm. Most of them are obtained in a likelihood framework, for Gaussian errors EijE_{ij}. Anderson (1999) relaxed this assumption and derived the asymptotic distribution of the estimate, when pp is fixed, the errors have two finite moments, and the rank of AA is known. Anderson (2002) continued this work by constructing asymptotic tests for rank selection, valid only for small and fixed values of pp.

The aim of our work is to develop a non-asymptotic class of methods that yield reduced rank estimators of AA that are easy to compute, have rank determined adaptively from the data, and are valid for any values of m,nm,n and pp, especially when the number of predictors pp 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 XX and YY. We refer to Chapter 6 in Izenman (2008) for a historical account of the latter.

We propose to estimate AA by minimizing the sum of squares ∥Y−XB∥F2=∑i∑j{Yij−(XB)ij}2\|Y-XB\|_{F}^{2}=\sum_{i}\sum_{j}\{Y_{ij}-(XB)_{ij}\}^{2} plus a penalty μr(B)\mu r(B), proportional to the rank r(B)r(B), over all matrices BB. It is immediate to see, using Pythagoras’ theorem, that this is equivalent with computing min⁡B{∥PY−XB∥F2+μr(B)}\min_{B}\left\{\|PY-XB\|_{F}^{2}+\mu r(B)\right\} or min⁡k{min⁡B: r(B)=k∥PY−XB∥F2+μk}\min_{k}\left\{\min_{B:\ r(B)=k}\|PY-XB\|_{F}^{2}+\mu k\right\}, with PP being the projection matrix onto the column space of XX. In Section 2.1 we show that the minimizer k^\widehat{k} of the above expression is the number of singular values dk(PY)d_{k}(PY) of PYPY that exceed μ1/2{\mu}^{1/2}. This observation reveals the prominent role of the tuning parameter μ\mu in constructing k^\widehat{k}. The final estimator A^\widehat{A} of the target matrix AA is the minimizer of ∥PY−XB∥F2\|PY-XB\|_{F}^{2} over matrices BB of rank k^\widehat{k}, and can be computed efficiently even for large pp, using the procedure that we describe in detail in Section 2.1 below.

The theoretical analysis of our proposed estimator A^\widehat{A} is presented in Sections 2.2 – 2.4. The rank of AA may not be the most appropriate measure of sparsity in multivariate regression models. For instance, suppose that the rank of AA 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 XAXA that are above a certain noise level. The relevant notion of noise level turns out to be the largest singular value of PEPE. This is central to our results, and influences the choice of the tuning sequence μ\mu. In Appendix C we prove that the expected value of the largest singular value of PEPE is bounded by (q+n)1/2(q+n)^{1/2}, where q≤m∧pq\leq m\wedge p is the rank of XX. The effective noise level is at most (m+n)1/2(m+n)^{1/2}, for instance in the model Y=A+EY=A+E, but it can be substantially lower, of order (q+n)1/2(q+n)^{1/2}, in model (1).

In Section 2.2 we give tight conditions under which k^\widehat{k}, the rank of our proposed estimator A^\widehat{A}, coincides with the effective rank. As an immediate corollary we show when k^\widehat{k} equals the rank of AA. We give finite sample performance bounds for ∥XA^−XA∥F2\|X\widehat{A}-XA\|_{F}^{2} in Section 2.3. These results show that A^\widehat{A} mimics the behavior of reduced rank estimates based on the ideal effective rank, had this been known prior to estimation. If XX has a restricted isometrity property, our estimate is minimax adaptive. In the asymptotic setting, for n+(m∧p)≥n+q→∞n+(m\wedge p)\geq n+q\rightarrow\infty, 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 N(0,σ2)N(0,\sigma^{2}) errors EijE_{ij} in order to obtain sharp, explicit numerical constants for the penalty term. To avoid technicalities, we assume that σ2\sigma^{2} is known in most cases, and we treat the case of unknown σ2\sigma^{2} in Section 2.4.

We contrast our estimator with the penalized least squares estimator A~\widetilde{A} corresponding to a penalty term τ∥B∥1\tau\|B\|_{1} proportional to the nuclear norm ∥B∥1=∑jdj(B)\|B\|_{1}=\sum_{j}d_{j}(B), the sum of the singular values of BB. 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 X\mathcal{X} 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 ∥XA~−XA∥F2\|X\widetilde{A}-XA\|_{F}^{2} 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 AA by the penalized least squares estimator

We denote its rank by k^\widehat{k}. The minimization is taken over all p×np\times n matrices BB. Here and in what follows r(B)r(B) is the rank of BB and ∥C∥F=(∑i∑jCij2)1/2\|C\|_{F}=\left(\sum_{i}\sum_{j}C_{ij}^{2}\right)^{1/2} denotes the Frobenius norm for any generic matrix CC. The choice of the tuning parameter μ>0\mu>0 is discussed in Section 2.2. Since

one needs to compute the restricted rank estimators B^k\widehat{B}_{k} that minimize ∥Y−XB∥F2\|Y-XB\|_{F}^{2} over all matrices BB of rank kk. The following computationally efficient procedure for calculating each B^k\widehat{B}_{k} has been suggested by Reinsel and Velu (1998). Let M=X′XM=X^{\prime}X be the Gram matrix, M−M^{-} be its Moore-Penrose inverse and let P=XM−X′P=XM^{-}X^{\prime} be the projection matrix onto the column space of XX.

Compute the eigenvectors V=[v1,v2,⋯ ,vn]V=[v_{1},v_{2},\cdots,v_{n}], corresponding to the ordered eigenvalues arranged from largest to smallest, of the symmetric matrix Y′PYY^{\prime}PY.

Compute the least squares estimator B^=M−X′Y\widehat{B}=M^{-}X^{\prime}Y. Construct W=B^VW=\widehat{B}V and G=V′G=V^{\prime}. Form Wk=W[ ,1:k]W_{k}=W[\,,1:k] and Gk=G[1:k, ]G_{k}=G[1:k,\,].

Compute the final estimator B^k=WkGk\widehat{B}_{k}=W_{k}G_{k}.

In step 2 above, WkW_{k} denotes the matrix obtained from WW by retaining all its rows and only its first kk columns, and GkG_{k} is obtained from GG by retaining its first kk rows and all its columns.

Our first result, Proposition 1 below, characterizes the minimizer k^=r(A^)\widehat{k}=r(\widehat{A}) of (3) as the number of eigenvalues of the square matrix Y′PYY^{\prime}PY that exceed μ\mu or, equivalently, as the number of singular values of the matrix PYPY that exceed μ1/2{\mu}^{1/2}. The final estimator of AA is then A^=B^k^\widehat{A}=\widehat{B}_{\widehat{k}}.

Lemma 14 in Appendix B shows that the fitted matrix XA^X\widehat{A} is equal to ∑j≤k^djujvj′\sum_{j\leq\widehat{k}}d_{j}u_{j}v_{j}^{\prime} based on the singular value decomposition UDV=∑jdjujvj′UDV=\sum_{j}d_{j}u_{j}v_{j}^{\prime} of the projection PYPY.

Let λ1(Y′PY)≥λ2(Y′PY)≥⋯\lambda_{1}(Y^{\prime}PY)\geq\lambda_{2}(Y^{\prime}PY)\geq\cdots be the ordered eigenvalues of Y′PYY^{\prime}PY. We have A^=B^k^\widehat{A}=\widehat{B}_{\widehat{k}} with

For B^k\widehat{B}_{k} given above, and by the Pythagorean theorem, we have

and we observe that XB^=PYX\widehat{B}=PY. By Lemma 14 in Appendix B, we have

where dj(C)d_{j}(C) denotes the jj-th largest singular value of a matrix CC. Then, the penalized least squares criterion reduces to

and we find that min⁡B{∥Y−XB∥F2+μr(B)}\min_{B}\left\{\|Y-XB\|_{F}^{2}+\mu r(B)\right\} equals

It is easy to see that ∑j>k{λj(Y′PY)−μ}\sum_{j>k}\left\{\lambda_{j}(Y^{\prime}PY)-\mu\right\} is minimized by taking kk as the largest index jj for which λj(Y′PY)−μ≥0\lambda_{j}(Y^{\prime}PY)-\mu\geq 0, since then the sum only consists of negative terms. This concludes our proof. ∎

Remark. The two matrices Wk^W_{\widehat{k}} and Gk^G_{\widehat{k}}, that yield the final solution A^=Wk^Gk^\widehat{A}=W_{\widehat{k}}G_{\widehat{k}}, have the following properties: (i) Gk^Gk^′G_{\widehat{k}}G_{\widehat{k}}^{\prime} is the identity matrix; and (ii) Wk^′MWk^W_{\widehat{k}}^{\prime}MW_{\widehat{k}} is a diagonal matrix. Moreover, the decomposition of A^\widehat{A} 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 Z=XWk^Z=XW_{\widehat{k}}. If k^\widehat{k} is much smaller than pp, 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 k^=r(A^)\widehat{k}=r(\widehat{A}). We will state simple conditions that guarantee that k^\widehat{k} equals r=r(A)r=r(A) with high probability. First, we describe in Theorem 2 what k^\widehat{k} estimates and what quantities need to be controlled for consistent estimation. It turns out that k^\widehat{k} estimates the number of the singular values of the signal XAXA above the threshold μ1/2\mu^{1/2}, for any value of the tuning parameter μ\mu. The quality of estimation is controlled by the probability that this threshold level exceeds the largest singular value d1(PE)d_{1}(PE) of the projected noise matrix PEPE. We denote the jjth singular value of a generic matrix CC by dj(C)d_{j}(C) and we use the convention that the singular values are indexed in decreasing order.

Suppose that there exists an index s≤rs\leq r such that

Using the characterization of k^\widehat{k} given in Proposition 1 we have

Invoke the conditions on ds+1(XA)d_{s+1}({XA}) and ds(XA)d_{s}({XA}) to complete the proof. ∎

Theorem 2 indicates that we can consistently estimate the index ss provided we use a large enough value for our tuning parameter μ\mu to guarantee that the probability of the event {d1(PE)≤δμ1/2}\left\{d_{1}(PE)\leq\delta\mu^{1/2}\right\} approaches one. We call ss the effective rank of AA relative to μ\mu, and denote it by re=re(μ)r_{e}=r_{e}(\mu).

Let q=r(X)q=r(X) and assume that EijE_{ij} are independent N(0,σ2)N(0,\sigma^{2}) random variables. Then

In view of this result, we take μ1/2>σ(n1/2+q1/2)\mu^{1/2}>\sigma(n^{1/2}+q^{1/2}) 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 rer_{e} and, in particular, the rank of AA.

Assume that EE has independent N(0,σ2)N(0,\sigma^{2}) entries. For any θ>0\theta>0, set

with δ\delta as in Theorem 2. Then we have, for any θ>0\theta>0,

In particular, if dr(XA)>2μ1/2d_{r}(XA)>2\mu^{1/2} and μ1/2=(1+θ)σ(n+q)\mu^{1/2}=(1+\theta)\sigma(\sqrt{n}+\sqrt{q}), then

Remark. Corollary 4 holds when q+n→∞q+n\rightarrow\infty. If q+nq+n stays bounded, but m→∞m\rightarrow\infty, the consistency results continue to hold when qq is replaced by qln⁡(m)q\ln(m) in the expression of the tuning parameter μ\mu given above. Lemma 3 justifies this choice. The same remark applies to all theoretical results in this paper.

for all x>0x>0. The conclusion of Corollary 4 then holds for μ=C0ΓE(n+q)\mu=C_{0}\Gamma_{E}(n+q) with C0C_{0} large enough. Moreover, all oracle inequalities presented in the next sections remain valid for this choice of the tuning parameter, if EE has independent subGaussian entries.

3 Errors bounds for the RSC estimator

In this section we study the performance of A^\widehat{A} by obtaining bounds for ∥XA^−XA∥F2\|X\widehat{A}-XA\|_{F}^{2}. First we derive a bound for the fit ∥XB^k−XA∥F2\|X\widehat{B}_{k}-XA\|_{F}^{2}, based on the restricted rank estimator B^k\widehat{B}_{k}, for each value of kk.

Set c(θ)=1+2/θc(\theta)=1+2/\theta. For any θ>0\theta>0, we have

for all p×np\times n matrices BB of rank kk. Working out the squares we obtain

for generic m×nm\times n matrices CC and DD. The inner product <C,D>F<C,D>_{F}, operator norm ∥C∥2=d1(C)\|C\|_{2}=d_{1}(C) and nuclear norm ∥D∥1=∑jdj(D)\|D\|_{1}=\sum_{j}d_{j}(D) are related via the inequality <C,D>F≤∥C∥2∥D∥1<C,D>_{F}\leq\|C\|_{2}\|D\|_{1}. As a consequence we find

Using the inequality 2xy≤x2/a+ay22xy\leq x^{2}/a+ay^{2} with a>0a>0 twice, we obtain that ∥XB^k−XA∥F2\|X\widehat{B}_{k}-XA\|_{F}^{2} is bounded above by

Hence we obtain, for any a,b>0a,b>0, the inequality

Lemma 14 in the Appendix B states that the minimum of ∥XA−XB∥F2\|XA-XB\|_{F}^{2} over all matrices BB of rank kk is achieved for the GSVD of AA and the minimum equals ∑j>kdj2(XA)\sum_{j>k}d_{j}^{2}(XA). The claim follows after choosing a=(2+θ)/2a=(2+\theta)/2 and b=θ/2b=\theta/2. ∎

Assume that EE has independent N(0,σ2)N(0,\sigma^{2}) entries. Set c(θ)=1+2/θc(\theta)=1+2/\theta. Then, for any θ,ξ>0\theta,\xi>0, the inequality

holds with probability 1−exp⁡(−ξ2(n+q)/2)1-\exp(-\xi^{2}(n+q)/2). In addition,

The symbol ≲\lesssim means that the inequality holds up to multiplicative numerical constants.

Set t=(1+ξ)2σ2(n+q)2t=(1+\xi)^{2}\sigma^{2}(\sqrt{n}+\sqrt{q})^{2} for some ξ>0\xi>0. From Lemma 3, it follows that

Theorem 5 bounds the error ∥XB^k−XA∥F2\|X\widehat{B}_{k}-XA\|_{F}^{2} by an approximation error, ∑j>kdj2(XA)\sum_{j>k}d_{j}^{2}(XA), and a stochastic term, kd12(PE)kd_{1}^{2}(PE), with probability one. The approximation error is decreasing in kk and vanishes for k>r(XA)k>r(XA).

We observe that k(n+q)k(n+q) is essentially the number of free parameters of the restricted rank problem. Indeed, our parameter space consists of all p×np\times n matrices BB of rank kk and each XBXB matrix has k(n+q−k)k(n+q-k) 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 k(n+q)k(n+q) is also the minimax lower bound for ∥XB^k−XA∥F2\|X\widehat{B}_{k}-XA\|_{F}^{2}, if the smallest eigenvalue of X′XX^{\prime}X is larger than a strictly positive constant. This means that XB^kX\widehat{B}_{k} is a minimax estimator under this assumption.

We now turn to the penalized estimator A^\widehat{A} and show that it achieves the best (squared) bias-variance trade-off among all rank restricted estimators B^k\widehat{B}_{k} for the appropriate choice of the tuning parameter μ\mu in the penalty pen(B)=μr(B)\text{pen}(B)=\mu r(B).

We have, for any θ>0\theta>0, on the event (1+θ)d12(PE)≤μ(1+\theta)d_{1}^{2}(PE)\leq\mu,

for any p×np\times n matrix BB. In particular, we have, for μ≥(1+θ)d12(PE)\mu\geq(1+\theta)d_{1}^{2}(PE)

for all p×np\times n matrices BB. Working out the squares we obtain

Consequently, using the inequality 2xy≤x2/a+ay22xy\leq x^{2}/a+ay^{2} twice, we obtain, for any a>0a>0 and b>0b>0,

Hence, if (a+b)d12(PE)−μ≤0(a+b)d_{1}^{2}(PE)-\mu\leq 0, we obtain

for any a>1a>1 and b>0b>0. Lemma 14 in Appendix B evaluates the minimum of ∥XA−XB∥F2\|XA-XB\|_{F}^{2} over all matrices BB of rank kk and shows that it equals ∑j>kdj2(XA)\sum_{j>k}d_{j}^{2}(XA). We conclude our proof by choosing a=1+θ/2a=1+\theta/2 and b=θ/2b=\theta/2. ∎

Remark. The first two parts of the theorem show that A^\widehat{A} achieves the best (squared) bias-variance trade-off among all reduced rank estimators B^k\widehat{B}_{k} if μ>d12(PE)\mu>d_{1}^{2}(PE). Moreover, the index kk which minimizes ∑j>k{dj2(XA)+μk}\sum_{j>k}\{d_{j}^{2}(XA)+\mu k\} essentially coincides with the effective rank re=re(μ)r_{e}=r_{e}(\mu) defined in the previous section. Therefore, the fit of the selected estimator XA^X\widehat{A} is comparable with that of the estimator XB^kX\widehat{B}_{k} with rank k=rek=r_{e}. Since the ideal rer_{e} depends on the unknown matrix AA, this ideal estimator cannot be computed. Although our estimator A^\widehat{A} is constructed independently of rer_{e}, it mimics the behavior of the ideal estimator B^re\widehat{B}_{r_{e}} and we say that the bound on ∥XA^−XA∥F2\|X\widehat{A}-XA\|_{F}^{2} adapts to re≤rr_{e}\leq r.

The last part of our result is a particular case of the second part, but it is perhaps easier to interpret. Taking the index kk equal to the rank rr, the bias term disappears and the bound reduces to rd12(PE)rd_{1}^{2}(PE) up to constants. This shows clearly the important role played by rr in the estimation accuracy: the smaller the rank of AA, the smaller the estimation error.

For Gaussian errors, we have the following precise bounds.

Assume that EE has independent N(0,σ2)N(0,\sigma^{2}) entries. Set

with θ,ξ>0\theta,\xi>0 arbitrary. Let c(θ)=1+2/θc(\theta)=1+2/\theta. Then, we have

Apply Lemma 16 in Appendix D to deduce that

Remarks. (i) We note that for n+qn+q large,

(iii) The same type of upper bound as the one of Corollary 8 can be proved if the entries of EE are subGaussian: take pen(B)=C(n+q)r(B)\text{pen}(B)=C(n+q)r(B) for some CC large enough, and invoke Proposition 15 in Appendix C.

(iv) Although the error bounds of ∥XA^−XA∥F\|X\widehat{A}-XA\|_{F} are guaranteed for all XX and AA, the analysis of the estimation performance of A^\widehat{A} depends on XX. If λp(M)≥ρ>0\lambda_{p}(M)\geq\rho>0, for some constant ρ\rho, then, provided μ>(1+θ)d12(PE)\mu>(1+\theta)d_{1}^{2}(PE) with θ>0\theta>0 arbitrary,

(v) Our results are slightly more general than stated. In fact, our analysis does not require that the postulated multivariate linear model Y=XA+EY=XA+E holds exactly. We denote the expected value of YY by Θ\Theta and write Y=Θ+EY=\Theta+E. We denote the projection of Θ\Theta onto the column space of XX by XAXA, that is, PΘ=XAP\Theta=XA. Because minimizing ∥Y−XB∥F2+μr(B)\|Y-XB\|_{F}^{2}+\mu r(B) is equivalent with minimizing ∥PY−XB∥F2+μr(B)\|PY-XB\|_{F}^{2}+\mu r(B) by Pythagoras’ theorem, our least squares procedure estimates XAXA, the mean of PYPY. The statements of Theorems 2 and 7 remain unchanged, except that XAXA is the mean of the projection PYPY of YY, not the mean of YY itself.

4 A data adaptive penalty term

In this section we construct a data adaptive penalty term that employs the unbiased estimator

of σ2\sigma^{2}. Set, for any θ>0\theta>0, ξ>0\xi>0 and 0<δ<10<\delta<1,

Notice that the estimator S2S^{2} requires that n(m−q)n(m-q) be large, which holds whenever m>>qm>>q or m−q≥1m-q\geq 1 and nn is large. The challenging case m=q<<pm=q<<p is left for future research.

Assume that EE is an m×nm\times n matrix with independent N(0,σ2)N(0,\sigma^{2}) entries. Using the penalty given above we have, for c(θ)=1+2/θc(\theta)=1+2/\theta,

It remains to bound the expected value of

We split the expectation into two parts: S2≥(1−δ)σ2S^{2}\geq(1-\delta)\sigma^{2} 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 n+qn+q and n(m−q)n(m-q),

as the additional terms in the theorem above decrease exponentially fast in n+qn+q and n(m−q)n(m-q). 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 σ2\sigma^{2}.

Comparison with nuclear norm penalized estimators

In this section we compare our RSC estimator A^\widehat{A} with the alternative estimator A~\widetilde{A} that minimizes

On the event d1(X′E)≤τd_{1}(X^{\prime}E)\leq\tau, we have, for any BB,

for all m×nm\times n matrices BB. Working out the squares we obtain

on the event d1(X′E)≤τd_{1}(X^{\prime}E)\leq{\tau}, we obtain the claim using the triangle inequality. ∎

We see that A~\widetilde{A} balances the bias term ∥XA−XB∥F2\|XA-XB\|_{F}^{2} with the penalty term τ∥B∥1\tau\|B\|_{1}, provided τ>d1(X′E)\tau>d_{1}(X^{\prime}E). Since X′E=X′PE+X′(I−P)E=X′PEX^{\prime}E=X^{\prime}PE+X^{\prime}(I-P)E=X^{\prime}PE, we have d1(X′E)≤d1(X)d1(PE)d_{1}(X^{\prime}E)\leq d_{1}(X)d_{1}(PE). We immediately obtain the following corollary using the results for d1(PE)d_{1}(PE) of Lemma 3.

Assume that EE has independent N(0,σ2)N(0,\sigma^{2}) entries. For

The same result, up to constants, can be obtained if the errors EijE_{ij} are subGaussian, if we replace σ\sigma in the choice of τ\tau above by a suitably large constant CC. 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 A~\widetilde{A} that resembles the oracle inequality for the RSC estimator A^\widehat{A} in Theorem 7. We stress the fact that Theorem 12 below requires that λp(X′X)>0\lambda_{p}(X^{\prime}X)>0; this was not required for the derivation of the oracle bound on ∥XA^−XA∥F2\|X\widehat{A}-XA\|_{F}^{2} in Theorem 7, which holds for all XX. We denote the condition number of M=X′XM=X^{\prime}X by c0(M)=λ1(M)/λp(M)c_{0}(M)=\lambda_{1}(M)/\lambda_{p}(M).

Assume that EE has independent N(0,σ2)N(0,\sigma^{2}) entries. For

Both inequalities hold with probability at least 1−exp⁡(−θ2(n+q)/2)1-\exp\left(-\theta^{2}(n+q)/2\right). The symbol ≲\lesssim means that the inequality holds up to multiplicative numerical constants (depending on θ\theta).

To keep the paper self contained, we give a simple proof of this result in Appendix A. Similar results for the NNP estimator of AA in the general model y=\mbox{\mathcal{X}}(A)+\varepsilon, where X\mathcal{X} 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 X\mathcal{X}. We refer to Rohde and Tsybakov (2010) for more general results on Schatten norm penalized estimators of AA in the model y=\mbox{\mathcal{X}}(A)+\varepsilon, and a very thorough discussion on the assumptions on X\mathcal{X} under which these results hold.

Theorem 10 shows that the error bounds of the nuclear norm penalized (NNP) estimator A~\widetilde{A} and the RSC estimator A^\widehat{A} are comparable, although it is worth pointing out that our bounds for A^\widehat{A} 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 μe\mu_{e} 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 AA 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 A~\widetilde{A} as the estimator of the rank of AA, we consider instead, for M=X′XM=X^{\prime}X,

Let r=r(A)r=r(A) and assume that dr(MA)>4τd_{r}(MA)>4\tau. Then

If EE has independent N(0,σ2)N(0,\sigma^{2}) entries and τ=(1+θ)σd1(X)(n+q)\tau=(1+\theta)\sigma d_{1}(X)(\sqrt{n}+\sqrt{q}), the above probability is bounded by exp⁡(−θ2(n+q)/2)\exp\left(-\theta^{2}(n+q)/2\right).

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 A^\widehat{A} 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 μ\mu 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 μ\mu is just a little bit larger than σ2(n+q)2\sigma^{2}(\sqrt{n}+\sqrt{q})^{2}. Under the assumption that q<mq<m, we can estimate σ2\sigma^{2} by S2S^{2}; see Section 2.4 for details. In our simulations we used the adaptive tuning parameter μadap=2S2(n+q)\mu_{adap}=2S^{2}(n+q). 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 A~\widetilde{A} 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 ∥Y−XB∥F2+2τ∥B∥1.\|Y-XB\|_{F}^{2}+2\tau\|B\|_{1}. 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∣adap|_{adap}, corresponding to μadap\mu_{adap}, with the optimally tuned RSC estimator, and the optimally tuned NNP and NNP(c) estimators. The last three estimators are called RSC∣val|_{val}, NNP∣val|_{val} and NNP(c)∣val{}^{(c)}|_{val}. They correspond to those tuning parameters μval\mu_{val}, τval\tau_{val} and τval\tau_{val}, 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 X=[x1,x2,⋯ ,xm]′X=[x_{1},x_{2},\cdots,x_{m}]^{\prime} by generating its rows xix_{i} as i.i.d. realizations from a multivariate normal distribution \mboxMVN(0,Σ)\mbox{MVN}(\boldsymbol{0},\Sigma), with Σjk=ρ∣j−k∣\Sigma_{jk}=\rho^{|j-k|}, ρ>0\rho>0, 1≤j,k≤p1\leq j,k\leq p. The coefficient matrix A=bB0B1A=bB_{0}B_{1}, with b>0b>0, B0B_{0} is a p×rp\times r matrix and B1B_{1} is a r×nr\times n matrix. All entries in B0B_{0} and B1B_{1} are i.i.d. N(0,1)N(0,1). Each row in Y=[y1,⋯ ,ym]′Y=[y_{1},\cdots,y_{m}]^{\prime} is then generated as yi=xi′A+Eiy_{i}=x_{i}^{\prime}A+E_{i}, 1≤i≤m1\leq i\leq m, with EiE_{i} denoting the ii-th row of the noise matrix EE which has m×nm\times n independent N(0,1)N(0,1) entries EijE_{ij}.

Experiment 2 (p>m(>q)𝑝annotated𝑚absent𝑞p>m(>q))

Each simulated model is characterized by the following control parameters: mm (sample size), pp (number of independent variables), nn (number of response variables), rr (rank of AA), ρ\rho (design correlation), qq (rank of the design), and bb (signal strength). In Experiment 1, we set m=100, p=25, n=25, r=10m=100,\,p=25,\,n=25,\,r=10, and varied the correlation coefficient ρ=0.1,0.5,0.9\rho=0.1,0.5,0.9 and signal strength b=0.1,0.2,0.3,0.4b=0.1,0.2,0.3,0.4. All combinations of correlation and signal strength are covered in the simulations. The results are summarized in Table 1. In Experiment 2, we set m=20m=20, p=100p=100, n=25n=25, q=10q=10, r=5r=5, and varied the correlation ρ=0.1, 0.5, 0.9\rho=0.1,\,0.5,\,0.9 and signal strength b=0.1, 0.2, 0.3b=0.1,\,0.2,\,0.3. The corresponding results are reported in Table 2. In both tables, MSE(AA) and MSE(XAXA) denote the 40%40\% trimmed-means of 100⋅∥A−B^∥F2/(pn)100\cdot\|A-\hat{B}\|_{F}^{2}/(pn) and 100⋅∥XA−XB^∥F2/(mn)100\cdot\|XA-X\hat{B}\|_{F}^{2}/(mn), 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 μadap=2S2(n+q)\mu_{adap}=2S^{2}(n+q) has excellent performance. It behaves as well as the RSC estimator that uses the parameter μ\mu tuned on the large validation set or the RSC estimator corresponding to the theoretical μ=2σ2(n+q)\mu=2\sigma^{2}(n+q).

(ii) When the signal-to-noise ratio SNR := dr(XA)/(q+n){d_{r}(XA)}/{(\sqrt{q}+\sqrt{n})} is moderate or high, with values approximately 1, 1.5 and 2, corresponding to b=0.2,0.3,0.4b=0.2,0.3,0.4, and for low to moderate correlation between the predictors (ρ=0.1,0.5\rho=0.1,0.5), 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 b=0.1b=0.1 in Table 1, or when the correlation between some covariates is very high, ρ=0.9\rho=0.9 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 rr (r=10r=10 in Experiment 1 and r=5r=5 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 AA 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 τ\tau 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 τ\tau tends to be too small, so it cannot introduce enough sparsity. The same continues to be true for slightly larger values of τ\tau 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 d1(X′E)≤τd_{1}(X^{\prime}E)\leq\tau. 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 A~1\widetilde{A}_{1} and A~2\widetilde{A}_{2} such that

A~=A~1+A~2\widetilde{A}=\widetilde{A}_{1}+\widetilde{A}_{2}

∥A~−B∥1=∥A~1−B∥1+∥A~2∥1\|\widetilde{A}-B\|_{1}=\|\widetilde{A}_{1}-B\|_{1}+\|\widetilde{A}_{2}\|_{1}

∥A~−B∥F2=∥A~1−B∥F2+∥A~2∥F2≥∥A~1−B∥F2\|\widetilde{A}-B\|_{F}^{2}=\|\widetilde{A}_{1}-B\|_{F}^{2}+\|\widetilde{A}_{2}\|_{F}^{2}\geq\|\widetilde{A}_{1}-B\|_{F}^{2}

∥A~∥1=∥A~1∥1+∥A~2∥1\|\widetilde{A}\|_{1}=\|\widetilde{A}_{1}\|_{1}+\|\widetilde{A}_{2}\|_{1}.

Using λp(M)∥A~−B∥F2≤∥XA~−XB∥F2\lambda_{p}(M)\|\widetilde{A}-B\|_{F}^{2}\leq\|X\widetilde{A}-XB\|_{F}^{2} and 2xy≤x2/2+2y22xy\leq x^{2}/2+2y^{2}, we obtain

The proof is complete by choosing the truncated GSVD B′B^{\prime} under metric MM, see Lemma 14 below. ∎

Appendix B Generalized singular value decomposition

with M=X′X=NNM=X^{\prime}X=NN and B0B_{0} is a fixed p×np\times n matrix of rank rr. By the Eckhart-Young theorem, we have the lower bound

for all p×np\times n matrices BB of rank kk. We now show that this infimum is achieved by the generalized singular value decomposition (GSVD) under metric MM, limited to its kk largest generalized singular values. Following Takane and Hunter (2001, pages 399-400), the GSVD of B0B_{0} under metric MM is UDV′UDV^{\prime} where UU is an p×rp\times r matrix, U′MU=IrU^{\prime}MU=I_{r}, VV is an n×rn\times r matrix, V′V=IrV^{\prime}V=I_{r} and DD is a diagonal r×rr\times r matrix, and NB0=NUDV′.NB_{0}=NUDV^{\prime}. It can be computed via the (regular) SVD UˉDˉVˉ′\bar{U}\bar{D}\bar{V}^{\prime} of NB0NB_{0}. From B0′X′XB0=VD2V′B_{0}^{\prime}X^{\prime}XB_{0}=VD^{2}V^{\prime}, the generalized singular values djd_{j} are the regular singular values of NB0NB_{0}. Let Bk=UkDkVk′B_{k}=U_{k}D_{k}V_{k}^{\prime} by retaining as usual the first kk columns of UU and VV.

Let BkB_{k} be the GSVD of B0B_{0} under metric MM, restricted to the kk largest generalized singular values. We have

Since NB0=NUDV′NB_{0}=NUDV^{\prime} and NBk=NUkDkVk′NB_{k}=NU_{k}D_{k}V_{k}^{\prime}, we obtain

using the notation U(k)U_{(k)} for the p×(r−k)p\times(r-k) matrix consisting of the last r−kr-k column vectors of UU, D(k)D_{(k)} is the diagonal (r−k)×(r−k)(r-k)\times(r-k) matrix based on the last r−kr-k singular values, and V(k)V_{(k)} for the n×(r−k)n\times(r-k) matrix consisting of the last r−kr-k column vectors of VV. Finally,

Recall that in the construction of the GSVD, the generalized singular values djd_{j} are the singular values of NB0NB_{0}. Since

Remark. The rank restricted estimator B^k\widehat{B}_{k} given in Section 2.1 is the GSVD of the least squares estimator B^\widehat{B} under the metric M=X′XM=X^{\prime}X, see Takane and Hwang (2007).

Appendix C Largest singular values of transformations of subGaussian matrices

We call a random variable WW subGaussian with subGaussian moment ΓW\Gamma_{W}, if

for all t>0t>0. Markov’s inequality implies that WW has Gaussian type tails:

holds for any t>0t>0. Normal N(0,σ2)N(0,\sigma^{2}) random variables are subGaussian with ΓW=σ2\Gamma_{W}=\sigma^{2}. General results on the largest singular values of matrices EE 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 PEPE and X′EX^{\prime}E, for which the standard results on EE do not apply directly.

Let EE be a m×nm\times n matrix with independent subGaussian entries EijE_{ij} with subGaussian moment ΓE\Gamma_{E}. Let XX be an m×pm\times p matrix of rank qq and let P=X(X′X)−X′P=X(X^{\prime}X)^{-}X^{\prime} be the projection matrix on R[X]R[X]. Then, for each x>0x>0,

with U=PSp−1={u=Ps: s∈Sp−1}U=PS^{p-1}=\{u=Ps:\ s\in S^{p-1}\}. Let M\mathcal{M} be a δ\delta-net of UU and N\mathcal{N} be a δ\delta-net for Sn−1S^{n-1} with δ=1/2\delta=1/2. Since the dimension of UU is qq and∥u∥≤1\|u\|\leq 1 for each u∈Uu\in U, we need at most 5q5^{q} elements in M\mathcal{M} to cover UU and 5n5^{n} elements to cover Sn−1S^{n-1}, 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 <u,Ev>=∑i=1mui<Ei,v><u,Ev>=\sum_{i=1}^{m}u_{i}<E_{i},v> and note that each <Ei,v><E_{i},v> is subGaussian with moment ΓE\Gamma_{E}, as

It follows that each term in ∑i=1mui<Ei,v>\sum_{i=1}^{m}u_{i}<E_{i},v> is subGaussian, and <u,Ev><u,Ev> is subGaussian with subGaussian moment ΓE∑i=1mui2=ΓE\Gamma_{E}\sum_{i=1}^{m}u_{i}^{2}=\Gamma_{E}. This implies the tail bound

for each fixed uu and vv and all t>0t>0. Combining the previous two steps, we obtain

for all t>0t>0. Taking t2=2{ln⁡(5)(n+q)+x}ΓEt^{2}=2\{\ln(5)(n+q)+x\}\Gamma_{E} 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 ZdZ_{d} be a χd2\chi^{2}_{d} random variable with dd degrees of freedom. Then

See Cavalier et al (2002, page 857) for the first claim. The second claim follows by taking x=t(d/2)1/2x=t(d/2)^{1/2}. ∎

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.

References