Nuclear norm penalization and optimal rates for noisy low rank matrix completion
Vladimir Koltchinskii, Alexandre B. Tsybakov, Karim Lounici
Introduction
It will be convenient to write the model (1.1) in the form
Here , where denotes the distribution of . The corresponding semi-norm is given by
Example 1. Matrix Completion. Assume that the design matrices are i.i.d. uniformly distributed on the set
One can also consider more general matrix measurement models in which, for a given orthonormal basis in the space of matrices, a random sample of Fourier coefficients of the target matrix is observed subject to a random noise. For more discussion on matrix completion with other types of sampling, see and references therein.
Example 4. Fixed design. Assume that all the are Dirac measures, so that the design matrices are non-random. Then and we get the problem of trace regression with fixed design, cf. . In particular, if , and and are diagonal matrices the trace regression model (1.2) becomes the usual linear regression model. Accordingly, the rank of becomes the number of its non-zero diagonal elements. This observation will allow us to deduce, as a consequence of our general argument, an oracle inequality for the usual Lasso in sparse linear regression with fixed design improving in the sense that the inequality is sharp (cf. Theorem 2 and Section 5.4).
The general oracle inequalities that we will prove in Section 2 can be successfully applied to the above examples. The emphasis in this paper will be on the matrix completion problem (Example 1), for which the previously obtained results were suboptimal.
Statistical estimation of low-rank matrices has recently become a very active field with a rapidly growing literature. The most popular methods are based on penalized empirical risk minimization with nuclear-norm penalty . Estimators with other types of penalization, such as the Schatten- norm , the von Neumann entropy , penalization by the rank or some combined penalties are also discussed.
The emphasis in this paper is on the noisy matrix completion setting. Then the estimator has a particularly simple form; it is obtained from the matrix by soft thresholding of its singular values. One of the main results of this paper is to show that our estimators are rate optimal (up to logarithmic factors) under the Frobenius error for a simple class of matrices defined by two restrictions: the rank of is not larger than given and all the entries of are bounded in absolute value by a constant . This rather intuitive class has been first considered in . However, the construction of the estimator in requires the exact knowledge of and the upper bound on the Frobenius error obtained in is suboptimal (see the details in Section 3). The recent paper obtains suboptimal bounds of ”slow rate” type for matrix completion while focuses on complex-valued Hermitian matrices with nuclear norm equal to 1, which is motivated by density matrix estimation problem in quantum state tomography. These papers do not address the optimality issue. Optimal rates in noisy matrix completion are derived in , but on different classes of matrices and with the empirical prediction error rather than with the Frobenius error. Finally, discusses the optimality issue for the Frobenius error on the classes defined in terms of a ”spikiness index” of , which are not related to , and suggests estimators that require prior knowledge about this index.
The main contributions of this paper are the following. In Section 2 we derive a general oracle inequality for the prediction error . This oracle inequality is sharp, i.e., with leading constant 1, both in the case of ”slow rate” (for matrices with small nuclear norm) and in the case of ”fast rate” (for matrices with small rank). As a particular instance of this general result, in Section 3 we obtain an oracle inequality for the matrix completion problem. In Section 4, we establish minimax lower bounds showing that the rates for matrix completion obtained in Section 3 are optimal up to a logarithmic factor. In Section 5, we briefly discuss some other implications and extensions of our method. Finally, Section 6 is devoted to the control of the stochastic term appearing in the proof of the upper bound.
General oracle inequalities
The Schatten- (quasi-)norm of matrix is defined by
Recall the well-known trace duality property:
We will also use the fact that the subdifferential of the convex function is the following set of matrices:
We will need the following assumption on the distribution of the matrices .
where we set for brevity . Under the assumption this yields
where The condition that belongs to the normal cone at the point implies that and (2.6) follows.
for an arbitrary By monotonicity of subdifferentials of convex functions, On the other hand, by (2.1), the following representation holds
where is an arbitrary matrix with It follows from the trace duality that there exists with such that
where in the first equality we used that has the support For this particular choice of (2.7) implies that
To provide an upper bound on we use the following decomposition
where . This implies, due to the trace duality,
and , , we have
and to Assumption 1, it follows from (2) and (2) that
Using the above bounds on and we obtain from (2) that
The following immediate corollary of Theorem 1 provides a bound for the Frobenius error.
and, for define the following cone of matrices:
Upper bounds for matrix completion
We can also write explicitly:
where , are the singular values and are the left and right singular vectors of . Thus, has a particularly simple form; it is obtained by soft thresholding of singular values in the SVD of . To see why (3.2) gives the solution of (3.1), note that, in view of (2.1), the subdifferential of is the set of matrices
where correspond to the SVD of . Since is strictly convex, the minimizer is unique, and the condition is necessary and sufficient characterization of the minimum, where is the zero matrix. Considering
it is easy to check that (3.2) satisfies this condition.
We will see that the soft thresholding representation (3.2) helps to understand in an easy way some theoretical properties of . However, it may not be always preferable for computational issues. Indeed, the standard techniques of computation of the SVD can become numerically instable when the dimension is high. On the other hand, we can always compute from (3.1) using the methods of convex programming free from this drawback.
In view of Theorem 1, to get the oracle inequalities in a closed form it remains only to specify the value of regularization parameter such that with high probability. This requires some assumptions on the distribution of , and the value of will be different under different assumptions. We will consider only the following two cases of particular interest.
and for some constant .
Statistical learning setting. There exists a constant such that almost surely.
In both cases, we obtain the upper bounds for (that we call the stochastic error) using the non-commutative Bernstein inequalities, cf. Section 6. The resulting values of and the corresponding oracle inequalities are given in the next two theorems.
Set . In what follows, we will denote by absolute positive constants, possibly different on different occasions.
Let be i.i.d. uniformly distributed on , and the pairs be i.i.d. Assume that for some constant , and that condition (3.3) holds. For consider the regularization parameter satisfying
Let be i.i.d. uniformly distributed on . Assume that almost surely for some constant . For consider the regularization parameter satisfying
Theorems 3 and 4 follow immediately from Theorem 1 and Lemmas 1, 2 and 3 with .
Note that the natural choice of in Theorems 3 and 4 is of the order , since a larger leads to slower rate of convergence and a smaller does not improve the rate but makes the concentration probability smaller. Note also that, under this choice of , the second terms under the maxima in (3.4) and (3.6) are negligible for the values of such that the term containing in (3.5) is meaningful. Indeed, if is of the order , the condition that necessarily implies . On the other hand, the negligibility of the second terms under the maxima in (3.4) and (3.6) is approximately equivalent to and respectively. Based on these remarks, we can choose in the form
where equals either or and the constant is large enough, and we can state the following corollary that will be further useful for minimax considerations. Define by
where , and .
Let one of the sets of conditions (i) or (ii) below be satisfied:
(ii) The assumptions of Theorem 4 with , as in (3.7), , and .
Then, with probability at least ,
where , and . Furthermore, with the same probability,
proof. Inequalities (3.8) and (3.9) are straightforward in view of Theorems 3 and 4. To prove (3.10) it suffices to note that, for any , ,
Inequality (3.9) guarantees that the normalized Frobenius error of the estimator is small whenever with a large enough . This quantifies the sample size necessary for successful matrix completion from noisy data.
Note that we can choose not necessarily equal but also greater or equal to the right hand side of (3.7), or equivalently, for any . Then the resulting oracle inequalities will remain of the same form with multiplied by the constant .
Keshavan et al. , Theorem 1.1, under a sampling scheme different from ours (sampling without replacement) and sub-gaussian errors, proposed an estimator satisfying, with probability at least ,
where is a constant, and is the aspect ratio. A drawback is that the construction of in requires the exact knowledge of (although it does not seem to require the knowledge of ). Furthermore, the bound (3.11) is suboptimal for ”very rectangular” matrices, i.e., when . Candes and Plan provide a coarser bound than (3.11), not guaranteeing a simple consistency when whatever are and (see for more detailed comments on ).
Lower Bounds
In this section, we prove the minimax lower bounds showing that the rates attained by our estimator are optimal up to logarithmic factors. The argument here is close to where the lower bounds are obtained on the Schatten balls. However, we consider different classes that consist of matrices with uniformly (in ) bounded entries. We cannot apply directly the lower bounds of Theorem 6 in for USR matrix completion on the Schatten balls because they are achieved on matrices with entries, which are not uniformly bounded for .
We will need the following assumption, which is similar in spirit but, in general, substantially weaker than the usual Restricted Isometry condition.
(Restricted Isometry in Expectation.) For some and some that there exists a constant such that
For the particular case of fixed (cf. Example 4 in the Introduction), Assumption 2 coincides with the matrix version of scaled restricted isometry with scaling factor .
Remark 1. Inspection of the proof of Theorem 5 shows that it remains valid if we replace and by arbitrary positive constants and such that . We use the formulation involving only to ease parallels to the usual restricted isometry condition.
Fix and an integer . Let Assumption 2 be satisfied with some . Assume that , and that conditionally on , the variables are Gaussian , , for . Then there exist absolute constants and , such that
proof. Without loss of generality, assume that . For some constant we define
and consider the associated set of block matrices
where denotes the zero matrix, and is the integer part of .
is satisfied for any if is chosen as a sufficiently small numerical constant depending on . In view of (4.3) and (4.5), the result now follows by application of Theorem 2.5 in .
Fix and an integer such that , . Let the matrices be i.i.d. uniformly distributed on and let, conditionally on , the variables be Gaussian , , for . Then there exist absolute constants and , such that
Comparing Theorem 6 with Corollary 2(i) we see that, in the case of Gaussian errors , the rate of convergence of our estimator given in (3.9) is optimal (up to a logarithmic factor) in a minimax sense on the class of matrices .
Similar conclusion can be obtained for the statistical learning setting. Indeed, assume that the pairs are i.i.d. realizations of a random pair with distribution belonging to the class
where is the uniform distribution on , is an integer, and .
Let be as in Theorem 5. Let be i.i.d. realizations of a random pair with distribution . Then there exist absolute constants and , such that
proof. We act as in the proof of Theorem 5 with some modifications. Assuming that and we define the class of matrices
Using the inequality , , and the fact that , we find that the expression under the expectation in (4.8) is bounded by . This implies
The remaining arguments are analogous to those in the proof of Theorem 5.
Further results and examples
A notable property of the estimator in matrix completion setting is that it has the same rank as the underlying matrix with probability close to 1. As a consequence we can establish a lower bound for the Frobenius error of with the rates matching up to constants the upper bounds of Corollary 2.
Let be i.i.d. uniformly distributed on and let satisfy the inequality (as in Theorem 1). Consider the estimator with for some . Set . Then
If, in addition, , then
proof. Note that . Using standard matrix perturbation argument (cf. , page 203), we get, for all ,
Since, by (3.2), , we find that . This implies (5.1). Now, if we get
Let the assumptions of Corollary 2 be satisfied. Consider the estimator with
for some . Set . Then with probability at least . If, in addition,
then and
We note that the lower bound for in (5.4) is not excessively high, since is a ”typical” order of the largest singular value for non-lacunary matrices . For example, if all the entries of are equal to some constant , the left hand side of (5.4) is equal to .
2 Risk bounds in statistical learning
We illustrate this by an example dealing with USR matrix completion. Specifically, Theorem 4 is reformulated in the following way.
Let be i.i.d. uniformly distributed on . Assume that almost surely for some constant . For consider the regularization parameter satisfying (3.6). Then with probability at least we have
This theorem can be also viewed as a result about the approximate sparsity. We do not know whether the true underlying model is described by some matrix but we can guarantee that our estimator is not far from the best approximation provided by matrices with small rank or small nuclear norm.
Note that the results of Theorem 9 are uniform over the class of distributions
where is the uniform distribution on , and is a constant. The corresponding lower bound is given in the next theorem.
Let be as in Theorem 5. Let be i.i.d. realizations of a random pair with distribution . Then
where and are absolute constants.
Inequalities (5.7) and (5.8) imply minimax rate optimality of up to a logarithmic factor in the statistical learning setting.
3 Risks bounds in spectral norm
Let be i.i.d. uniformly distributed on . Consider the estimator defined in (3.1). If , then
As a consequence of the above theorem, we can derive the optimal rate (up a to logarithmic factor) of USR matrix completion for the spectral norm when the noise is sub-exponential or in the statistical learning setting.
Let one of the sets of conditions (i) or (ii) in Corollary 2 be satisfied. Then, with probability at least , we have
proof. The proof of this result is immediate by combining Theorem 11 and Lemmas 1, 2 and 3.
(i) Let the conditions of Theorem 6 be satisfied. Then
where and are absolute constants.
(ii) Let the conditions of Theorem 7 be satisfied. Then
where and are absolute constants.
proof. Note first that, in the USR matrix completion problem, Assumption 2 is satisfied with and .
We prove part (i) of the theorem. Consider the set of matrices introduced in the proof of Theorem 5. For any two distinct matrices of , we have
Next, (4.5) is satisfied for any if is chosen as a sufficiently small numerical constant depending on .
Combining (5.11) with (4.5) and Theorem 2.5 in gives the result.
The proof of (ii) follows the same arguments.
4 Sharp oracle inequalities for the Lasso
As we already mentioned in Example 4 and in the remark after Theorem 2, one can exploit (2.19) to derive sparsity oracle inequalities for the usual Lasso. This is detailed in the present subsection. It is noteworthy that the obtained inequalities are sharp (i.e., with leading constant ), which was not achieved in the previous work on the Lasso.
Note that, if and and are diagonal matrices, then the trace regression model (1.2) becomes
The estimator defined in (1.7) becomes the usual Lasso estimator
For simplicity, the result is stated only in the case of Gaussian noise.
where Then, with probability at least , we have
proof. Combine Theorem 2 and a standard bound on the tail of the Gaussian distribution, which assures that with probability at least ,
We recall the Restricted Eigenvalue condition of :
Condition . For some integer such that , and a positive number the following condition holds:
Let the assumptions of Theorem 14 hold, and let condition be satisfied for some . Then, with probability at least
Since Condition is satisfied, Theorem 14 yields the result.
Remark 2. Oracle inequalities (5.12) and (5.13) extend straightforwardly to the model
Control of the stochastic error
In this section, we obtain the probability inequalities for the stochastic error . For brevity, we will write throughout . The following proposition is an immediate consequence of the matrix version of Bernstein’s inequality (Corollary 9.1 in ).
Then, for all with probability at least we have
Furthermore, it is possible to replace the -bound on in the above inequality by bounds on the weaker -norms of defined by
This is an easy consequence of Proposition 2 in , which provides an analogous result for Hermitian matrices . Its extension to rectangular matrices stated in Proposition 2 is straightforward via the self-adjoint dilation, cf., for example, the proof of Corollary 9.1 in .
The next lemma gives a control of the stochastic error for USR matrix completion in the statistical learning setting.
Let be i.i.d. uniformly distributed on . Assume that almost surely for some constant . Then for any with probability at least we have
Therefore, , , and the result follows from Proposition 1.
We now consider the USR matrix completion with sub-exponential errors. Recall that in this case we assume that the pairs are i.i.d. We have
We treat the terms and separately in the two lemmas below.
Finally, in view of Condition (3.3) and Bernstein’s inequality for sub-exponential noise, we have for any , with probability at least ,
Let be i.i.d. random variables uniformly distributed in . Then, for all with probability at least we have
If for some , then with the same probability