Noisy Matrix Completion: Understanding Statistical Guarantees for Convex Relaxation via Nonconvex Optimization: Convex Relaxation Attains Near-Optimal Frobenius, Entrywise and Spectral Norm ErrorsResearch Paper
Motivation
Matrix completion asks for a low-rank matrix from a small random subset of its entries. It models collaborative filtering (recommendation systems), sensor-network localisation, and structure from motion. In practice the revealed entries are corrupted by noise, and the most widely used estimator is nuclear-norm regularized least squares: fit the observed entries in squared error while penalizing the sum of singular values. The estimator is a convex program, so it is solved reliably by proximal and singular-value-thresholding methods, and it needs no knowledge of the rank.
Its statistical accuracy was not understood. In the noiseless case, convex relaxation recovers the matrix exactly from nearly the information-theoretic minimum number of samples (Candès–Recht 2009, Candès–Tao 2010, Gross 2011). With noise, the available bounds for the plain convex program (Candès–Plan 2010) give under random noise. That can be a factor of order above the oracle lower bound, and the numerical performance of the program is far better. Sharper rates were known only for modified estimators: Negahban–Wainwright 2012 add a spikiness constraint, Koltchinskii–Lounici–Tsybakov 2011 use a different estimator. In 2019, Chen, Chi, Fan, Ma and Yan (arXiv:1902.07698; SIAM J. Optim. 30(4), 2020, doi:10.1137/19M1290000) showed that the unmodified convex program attains near-optimal error in the Frobenius, entrywise and spectral norms simultaneously. Their argument goes through nonconvex gradient descent. This mission formalizes that theorem.
Setting
The unknown matrix is of rank , with singular value decomposition . The columns of are orthonormal, and with , and condition number . Write for the largest Euclidean norm of a row of . is -incoherent if and (Definition 1). Incoherence says that the energy of is not concentrated on a few rows or columns.
Under Assumption 1, each index enters the observed set independently with probability . One observes for , where the are i.i.d., zero-mean, and have sub-Gaussian norm . Let keep the entries indexed by and zero the rest. The convex estimator is any minimizer of
where is the nuclear norm, is the spectral norm and is the largest absolute entry.
Formalization targets
Goal: Theorem 2
Suppose and . Then with probability , every minimizer of (3) satisfies
Every best rank- approximation is within of in Frobenius norm, and satisfies the same three bounds.
Milestones
- Deterministic layer: Lemma 20 (balancing an SVD factorization), Claim 3, Claim 2 and Lemma 2. Lemma 2 says that a point with for the factored objective
satisfies under Conditions 1–2. 2. Probabilistic layer: Lemma 3 (Condition 1(a)), Lemma 19, Lemmas 7 and 8, and Lemma 4. Lemma 4 gives Conditions 1(b) and 2 uniformly over an neighbourhood of the truth. 3. Algorithmic layer: Lemma 5, the error bounds and the small-gradient iterate of gradient descent started at the truth, and (31), its consequence for .
Significance
The bounds of Theorem 2 match the minimax lower bounds up to logarithmic factors when and are bounded. This holds in all three norms and for noise of any size below a threshold. The theorem also shows that the convex solution is nearly of rank . Neither statement requires modifying the estimator, so practitioners running standard solvers get the guarantee as is. The entrywise bound (11b) shows an implicit regularization: the convex program spreads its error over all entries without being told to.
The theorem has a complete published proof. What this mission adds is a machine-checked version, which has value of its own here. The proof chains about twenty lemmas, including a leave-one-out analysis of gradient descent with polynomially many iterations. Its constants are hidden behind "" at nearly every step. The formal development fixes the quantifier order of every such constant. It also records the hypotheses the proof uses without stating them, such as the independence of the sampling pattern from the noise. No part of this paper has been formalized before. The deterministic layer (Lemmas 20, 2, Claims 2–3) applies to any Burer–Monteiro factorization of a nuclear-norm problem, not only to this one.
Difficulty
The obvious approach is to analyse the convex program directly through a dual certificate, as in the noiseless theory. With noise there is no exact recovery, and the certificate approach loses a factor polynomial in . The difficulty is to transfer statistical bounds from a point one can control, the nonconvex iterate, to a point one cannot construct, the convex minimizer. The two are compared through an approximate first-order condition. That comparison is useful only if injectivity and concentration hold uniformly over all factor pairs near the truth, because the iterates depend on and on the noise. A fixed-tangent-space injectivity bound, the usual tool, does not suffice. The entrywise control of the iterates requires leave-one-out sequences, and those sequences make up most of the appendix.
Formalization scope
Matrices are Matrix (Fin n) (Fin n) ℝ and factors Matrix (Fin n) (Fin r) ℝ, using the published matrix_completion_basic, _svd and _tangent definitions (nuclear norm, SVD data, , , spectral, Frobenius and entrywise norms). comes with an explicit SVD with positive singular values, so its rank is exactly . Definition 1 is MatrixCompletion.A0. The sub-Gaussian bound is the moment form of [Ver12, Definition 5.7] above, not Mathlib's MGF form. The randomness lives on an arbitrary probability space. The sampling indicators and the noise entries are each mutually independent, and the two families are independent of each other. This last independence is an added hypothesis, which the paper's proofs use. "With probability " is stated as an outer-measure bound on the failure event. Every unnamed constant ("", "sufficiently large ", "sufficiently small ") is an existential placed before and all data, together with an . Explicit constants of the paper are kept: the exponents and , , , , , , , the factors in (31), the in (12), and in Lemma 2.
The goal keeps the randomness in the statement. A version with fixed and that assumes Conditions 1–2 or the conclusions of Lemma 5 would be Lemma 2 plus (31), not Theorem 2, and is not acceptable as a proof of the goal. The goal quantifies over every minimizer of (3) and every best rank- approximation. Excluded: Theorem 1 (the corollary), the noise condition of Remark 4, approximate minimizers (Remark 6), Lemmas 1 and 6, and the internal lemmas of the leave-one-out analysis (App. D), which solvers may add as supporting theorems. The development needs nuclear-norm subgradients, Weyl's inequality, matrix Bernstein and spectral-norm bounds for sparse random matrices. These are reusable well beyond this mission, and contributions of them are welcome. Theorem and equation numbers are those of the authors' preprint (Sep. 2019, arXiv:1902.07698) and may differ in the SIAM typesetting.
Selected references
- Y. Chen, Y. Chi, J. Fan, C. Ma, Y. Yan, Noisy Matrix Completion: Understanding Statistical Guarantees for Convex Relaxation via Nonconvex Optimization, SIAM J. Optim. 30(4), 2020. https://doi.org/10.1137/19M1290000 (preprint: https://arxiv.org/abs/1902.07698)
- E. J. Candès, B. Recht, Exact Matrix Completion via Convex Optimization, Found. Comput. Math. 9, 2009. https://doi.org/10.1007/s10208-009-9045-5
- E. J. Candès, Y. Plan, Matrix Completion With Noise, Proc. IEEE 98(6), 2010. https://doi.org/10.1109/JPROC.2009.2035722
- S. Negahban, M. J. Wainwright, Restricted strong convexity and weighted matrix completion: optimal bounds with noise, J. Mach. Learn. Res. 13, 2012. https://jmlr.org/papers/v13/negahban12a.html
- V. Koltchinskii, K. Lounici, A. B. Tsybakov, Nuclear-norm penalization and optimal rates for noisy low-rank matrix completion, Ann. Statist. 39(5), 2011. https://doi.org/10.1214/11-AOS894
- C. Ma, K. Wang, Y. Chi, Y. Chen, Implicit Regularization in Nonconvex Statistical Estimation, Found. Comput. Math. 20, 2020. https://doi.org/10.1007/s10208-019-09429-9
- R. Vershynin, Introduction to the non-asymptotic analysis of random matrices, in Compressed Sensing: Theory and Applications, CUP, 2012. https://arxiv.org/abs/1011.3027