Nearly-optimal Robust Matrix Completion

Yeshwanth Cherapanamjeri, Kartik Gupta, Prateek Jain

Introduction

In this paper, we study the Robust Matrix Completion (RMC) problem where the goal is to recover an underlying low-rank matrix by observing a small number of sparsely corrupted entries from the matrix. Formally,

RMC is an important problem with several applications such as recommendation systems with outliers. Similarly, the problem is also heavily used to model PCA under gross outliers as well as erasures [JRVS11]. Finally, as we show later, an efficient solution to RMC enables faster solution for the robust PCA (RPCA) problem as well. The goal in RPCA is to find a low-rank matrix L∗L^{*} and sparse matrix S~∗\widetilde{S}^{*} by observing their sum, i.e., M=L∗+S~∗M=L^{*}+\widetilde{S}^{*}. State-of-the-art results for RPCA shows exact recovery of a rank-rr, μ\mu-incoherent L∗L^{*} (see Assumption 1, Section 3) if at most ρ=1μ2r\rho=\frac{1}{\mu^{2}r} fraction of the entries in each row/column of S~∗\widetilde{S}^{*} are corrupted [HKZ11, NUNS+14].

However, the existing state-of-the-art results for RMC with optimal ρ=1μ2r\rho=\frac{1}{\mu^{2}r} fraction of corrupted entries, either require at least a constant fraction of the entries of L∗L^{*} to be observed [CJSC11, CLMW11] or require restrictive assumptions like support of corruptions S~∗\widetilde{S}^{*} being uniformly random [Li13]. [KLT14] also considers RMC problem but studies the noisy setting and do not provide exact recovery bounds. Moreover, most of the existing methods for RMC use convex relaxation for both low-rank and sparse components, and in general exhibit large time complexity (O(m2n)O(m^{2}n)).

In this work, we attempt to answer the following open question (assuming m≤nm\leq n): Can RMC be solved exactly by using ∣Ω∣=O(rnlog⁡n)|\Omega|=O(rn\log n) observations out of which O(1μ2r)O(\frac{1}{\mu^{2}r}) fraction of the observed entries in each row/column are corrupted. Note that both ∣Ω∣|\Omega| (for uniformly random Ω\Omega) and ρ\rho values mentioned in the question above denote the information theoretic limits. Hence, the goal is to solve RMC for nearly-optimal number of samples and nearly-optimal fraction of corruptions.

Under standard assumptions on L∗, S∗, ΩL^{*},\ S^{*},\ \Omega and for n=O(m)n=O(m), we answer the above question in affirmative albeit with ∣Ω∣|\Omega| which is O(r)O(r) (ignoring log factors) larger than the optimal sample complexity (see Theorem 1). In particular, we propose a simple projected gradient (PGD) style method for RMC that alternately cleans up corrupted entries by hard-thresholding; our method’s computational complexity is also nearly optimal (O(∣Ω∣r+(m+n)r2+r3)O(|\Omega|r+(m+n)r^{2}+r^{3})). Our algorithm is based on projected gradient descent for estimating L∗L^{*} and alternating projection on set of sparse matrices for estimating S∗S^{*}. Note that projection is onto non-convex sets of low-rank matrices (for L∗L^{*}) and sparse matrices (for S∗S^{*}), hence standard convex analysis techniques cannot be used for our algorithm.

In a concurrent and independent work, [YPCC16] also studied the RMC problem and obtained similar results. They study an alternating minimization style algorithm while our algorithm is based on low-rank projected gradient descent. Moreover, our sample complexity, corruption complexity as well as time complexity differs along certain critical parameters: a) Sample complexity: Our sample complexity bound is dependent only logarithmically on κ\kappa, the condition number of the matrix L∗L^{*} (see Table 1). On the other hand, result of [YPCC16] depends quadratically on κ\kappa, which can be significantly large. However, our sample complexity bound depends logarithmically on the final error ϵ\epsilon (defined as ϵ=∥L−L∗∥2\epsilon=\left\lVert L-L^{*}\right\rVert_{2}); this implies that for typical finite precision computation, our sample complexity bound can be worse by a constant factor. b) Our result allows the fraction of corrupted entries to be information theoretic optimal (up to a constant) O(1μ2r)O(\frac{1}{\mu^{2}r}), while the result of [YPCC16] allows only O(min⁡(1μ2rrκ,1μ2κ2r))O\left(\min\left(\frac{1}{\mu^{2}r\sqrt{r\kappa}},\frac{1}{\mu^{2}\kappa^{2}r}\right)\right) fraction of corrupted entries. c) As a consequence of the sample complexity bounds, running time of the method by [YPCC16] depends quintically on κ\kappa. On the other hand, our algorithm has optimal sparsity (up to a constant factor) independent of κ\kappa and polylogarithmic dependence on κ\kappa for sample and running time complexities.

Several recent results [JN15, NUNS+14, JTK14, HW14, Blu11] show that under certain assumptions, projection onto non-convex sets indeed lead to provable algorithms with fast convergence to the global optima. However, as explained in Section 3, RMC presents unique set of challenges as we have to perform error analysis with the errors arising due to missing entries as well as sparse corruptions, both of which interact among themselves as well. In fact, our careful error analysis also enables us to improve results for the matrix completion as well as the RPCA problem.

Matrix Completion (MC): The goal of MC is to find rank-rr L∗L^{*} using PΩ(L∗)\mathcal{P}_{\Omega}(L^{*}). State-of-the-art result for MC uses nuclear norm minimization and requires ∣Ω∣≥μ2nr2log⁡2n|\Omega|\geq\mu^{2}nr^{2}\log^{2}n under standard μ\mu-incoherence assumption (see Section 3), but the method requires O(m2n)O(m^{2}n) time in general. The best sample complexity result for a non-convex iterative method (with at most logarithmic dependence on the condition number of L∗L^{*}) achieve exact recovery when ∣Ω∣≥μ6nr5log⁡2n|\Omega|\geq\mu^{6}nr^{5}\log^{2}n and needs O(∣Ω∣r)O(|\Omega|r) computational steps. In contrast, assuming n=O(m)n=O(m), our method achieves nearly the same sample complexity of trace-norm but with nearly linear time algorithm (O(∣Ω∣r)O(|\Omega|r)). See Table 1 for a detailed comparison of our result with the existing methods.

RPCA: Several recent results show that RPCA can be solved if ρ=O(1μ2r)\rho=O(\frac{1}{\mu^{2}r})-fraction of entries in each row and column of L∗L^{*} are corrupted [NUNS+14, HKZ11] where L∗L^{*} is assumed to be μ\mu-incoherent. Moreover, St-NcRPCA algorithm [NUNS+14] can solve the problem in time O(mnr2)O(mnr^{2}). Corollary 2 shows that by sampling Ω\Omega uniformly at random, we can solve the problem in time O(nr3)O(nr^{3}) only. That is, we can recover L∗L^{*} without even observing the entire input matrix. Moreover, if the goal is to recover the sparse corruption as well, then we can obtain a two-pass (over the input matrix) algorithm that solves the RPCA problem exactly. St-NcRPCA algorithm requires r2log⁡(1/ϵ)r^{2}\log(1/\epsilon) passes over the data. Our method has significantly smaller space complexity as well.

Our empirical results on synthetic data demonstrates effectiveness of our method. We also apply our method to the foreground background separation problem; our method is an order of magnitude faster than the state-of-the-art method (St-NcRPCA) while achieving similar accuracy.

In summary, this paper’s main contributions are: (a) RMC: We propose a nearly linear time method that solves RMC with ∣Ω∣=O(nr2log⁡2nlog⁡2∥M∥2/ϵ)|\Omega|=O(nr^{2}\log^{2}n\log^{2}\|M\|_{2}/\epsilon) random entries and with optimal fraction of corruptions (ρ=1μ2r\rho=\frac{1}{\mu^{2}r}). (b) Matrix Completion: Our result improves upon the existing linear time algorithm’s sample complexity by an O(r3)O(r^{3}) factor, and time complexity by O(r4)O(r^{4}) factor, although with an extra O(log⁡∥L∗∥/ϵ)O(\log\|L^{*}\|/\epsilon) factor in both time and sample complexity. (c) RPCA: We present a nearly linear time (O(nr3)O(nr^{3})) algorithm for RPCA under optimal fraction of corruptions, improving upon O(mnr2)O(mnr^{2}) time complexity of the existing methods.

Paper Organization: We present our main algorithm in Section 2 and our main results in Section 3. We also present an overview of the proof in Section 3. Section 4 presents our empirical result. Due to lack of space, we present most of the proofs and useful lemmas in Appendix.

Algorithm

For the above problem, we propose a simple iterative algorithm that combines projected gradient descent (for LL) with alternating projections (for SS). In particular, we maintain iterates L(t)L^{(t)} (with rank k≤rk\leq r) and sparse S(t)S^{(t)}. L(t+1)L^{(t+1)} is computed using gradient descent step for objective (3) and then projecting back onto the set of rank kk matrices. That is,

where Pk(A)P_{k}(A) denotes projection of AA onto the set of rank-kk matrices and can be computed efficiently using SVD of AA, p=∣Ω∣mnp=\frac{|\Omega|}{mn}. S(t+1)S^{(t+1)} is computed by projecting the residual PΩ(M−L(t+1)){\mathcal{P}}_{\Omega}(M-L^{(t+1)}) onto set of sparse matrices using a hard-thresholding operator, i.e.,

Unfortunately, just the above two simple iterations cannot handle problems where L∗L^{*} has poor condition number, as the intermediate errors can be significantly larger than the smallest singular values of L∗L^{*}, making recovery of the corresponding singular vectors challenging. To alleviate this issue, we propose an algorithm that proceeds in stages. In the qq-th stage, we project L(t)L^{(t)} onto set of rank-kqk_{q} matrices. Rank kqk_{q} is monotonic w.r.t. qq. Under standard assumptions, we show that we can increase kqk_{q} in a manner such that after each stage ∥L(t)−L∗∥∞\left\lVert L^{(t)}-L^{*}\right\rVert_{\infty} decreased by at least a constant factor. Hence, the number of stages is only logarithmic in the condition number of L∗L^{*}.

See Algorithm 1 (PG-RMC ) for a pseudo-code of the algorithm. We require an upper bound of the first singular value for our algorithm to work. Specifically, we require σ=O(σ1∗)\sigma=O\left({\sigma^{*}_{1}}\right). Alternatively, we can also obtain an estimate of σ1∗\sigma^{*}_{1} by using the thresholding technique from [YPCC16] although this requires an estimate of the number of corruptions in each row and column. We also use a simplified version of Algorithm 5 from [HW14] to form independent sets of samples for each iteration which is required for our theoretical analysis. Our algorithm has an “outer loop” (see Line 6) which sets rank kqk_{q} of iterates L(t)L^{(t)} appropriately (see Line 7). We then update L(t)L^{(t)} and S(t)S^{(t)} in the “inner loop” using (4), (5). We set threshold for the hard-thresholding operator using singular values of current gradient descent update (see Line 12). Note that, we divide Ω\Omega uniformly into Q⋅TQ\cdot T sets, where QQ is an upper bound on the number of outer iterations and TT is the number of inner iterations. This division ensures independence across iterates that is critical to application of standard concentration bounds; such division is a standard technique in the matrix completion related literature [JN15, HW14, Rec11]. Also, η\eta is a tunable parameter which should be less than one and is smaller for “easier” problems.

Note that updating S(t)S^{(t)} requires O(∣Ω∣⋅r+(m+n)⋅r)O(|\Omega|\cdot r+(m+n)\cdot r) computational steps. Computation of L(t+1)L^{(t+1)} requires computing SVD for projection PrP_{r}, which can be computed in time O(∣Ω∣⋅r+(m+n)⋅r2+r3)O(|\Omega|\cdot r+(m+n)\cdot r^{2}+r^{3}) time (ignoring log⁡\log factors); see [JMD10] for more details. Hence, the computational complexity of each step of the algorithm is linear in ∣Ω∣⋅r|\Omega|\cdot r (assuming ∣Ω∣≥r⋅(m+n)|\Omega|\geq r\cdot(m+n)). As we show in the next section, the algorithm exhibits geometric convergence rate under standard assumptions and hence the overall complexity is still nearly linear in ∣Ω∣|\Omega| (assuming rr is just a constant).

Rank based Stagewise algorithm: We also provide a rank-based stagewise algorithm (R-RMC) where the outer loop increments kqk_{q} by one at each stage, i.e., the rank is qq in the qq-th stage. Our analysis extends for this algorithm as well, however, its time and sample complexity trades off a factor of O(log⁡(σ1/ϵ))O(\log(\sigma_{1}/\epsilon)) from the complexity of PG-RMC with a factor of rr (rank of L∗L^{*}). We provide the detailed algorithm in Appendix 5.3 due to lack of space (see Algorithm 3).

Analysis

We now present our analysis for both of our algorithms PG-RMC (Algorithm 1) and R-RMC (Algorithm 3). In general the problem of Robust PCA with Missing Entries (3) is harder than the standard Matrix Completion problem and hence is NP-hard [HMRW14]. Hence, we need to impose certain (by now standard) assumptions on L∗L^{*}, S~∗\widetilde{S}^{*}, and Ω\Omega to ensure tractability of the problem:

Sparsity of S~∗\widetilde{S}^{*}, S∗S^{*}: We assume that at most ρ≤cμ2r\rho\leq\frac{c}{\mu^{2}r} fraction of the elements in each row and column of S~∗\widetilde{S}^{*} are non-zero for a small enough constant cc. Moreover, we assume that Ω\Omega is independent of S~∗\widetilde{S}^{*}. Hence, S∗=PΩ(S~∗)S^{*}=\mathcal{P}_{\Omega}(\widetilde{S}^{*}) also has at most p⋅ρp\cdot\rho fraction of the entries in expectation.

Assumptions 1, 2 are standard assumptions in the provable matrix completion literature [CR09, Rec11, JN15], while Assumptions 1, 3 are standard assumptions in the robust PCA (low-rank+sparse matrix recovery) literature [CSPW11, CLMW11, HKZ11]. Hence, our setting is a generalization of both the standard and popular problems and as we show later in the section, our result can be used to meaningfully improve the state-of-the-art for both these problems.

We first present our main result for Algorithm 1 under the assumptions given above.

Let Assumptions 1, 2 and 3 on L∗L^{*}, S~∗\widetilde{S}^{*} and Ω\Omega hold respectively. Let m≤nm\leq n, n=O(m)n=O(m), and let the number of samples ∣Ω∣|\Omega| satisfy:

where CC is a global constant. Then, with probability at least 1−n−log⁡α21-n^{-\log\frac{\alpha}{2}}, Algorithm 1 with η=4μ2rm\eta=\frac{4\mu^{2}r}{m}, at most O(log⁡(∥M∥2/ϵ)))O(\log(\|M\|_{2}/\epsilon))) outer iterations and O(log⁡(μ2r∥M∥2ϵ))O(\log(\frac{\mu^{2}r\|M\|_{2}}{\epsilon})) inner iterations, outputs a matrix L^\hat{L} such that:

Note that our number of samples increase with the desired accuracy ϵ\epsilon. However, using argument similar to that of [JN15], we should be able to replace ϵ\epsilon by σmin⁡∗\sigma_{\min}^{*} which should modify the ϵ\epsilon term to be log⁡2κ\log^{2}\kappa where κ=σ1(L∗)/σr(L∗)\kappa=\sigma_{1}(L^{*})/\sigma_{r}(L^{*}). We leave ironing out the details for future work.

Note that the number of samples matches information theoretic bound upto O(rlog⁡nlog⁡2σ1∗/ϵ)O(r\log n\log^{2}\sigma_{1}^{*}/\epsilon) factor. Also, the number of allowed corruptions in S~∗\widetilde{S}^{*} also matches the known lower bounds (up to a constant factor) and cannot be improved upon information theoretically.

We now present our result for the rank based stagewise algorithm (Algorithm 3).

Under Assumptions 1, 2 and 3 on L∗L^{*}, S~∗\widetilde{S}^{*} and Ω\Omega respectively and Ω\Omega satisfying:

for a large enough constant CC, then Algorithm 3 with η\eta set to 4μ2rm\frac{4\mu^{2}r}{m} outputs a matrix L^\hat{L} such that: ∥L^−L∗∥F≤ϵ,\left\lVert\hat{L}-L^{*}\right\rVert_{F}\leq\epsilon, w.p. ≥1−n−log⁡α2\geq 1-n^{-\log\frac{\alpha}{2}}.

Notice that the sample complexity of Algorithm 3 has an additional multiplicative factor of O(r)O(r) when compared to that of Algorithm 1, but shaves off a factor of O(log⁡(κ))O(\log(\kappa)). Similarly, computational complexity of Algorithm 3 also trades off a O(log⁡κ)O(\log\kappa) factor for O(r)O(r) factor from the computational complexity of Algorithm 1.

Result for Matrix Completion: Note that for S~∗=0\widetilde{S}^{*}=0, the RMC problem with Assumptions 1,2 is exactly the same as the standard matrix completion problem and hence, we get the following result as a corollary of Theorem 1:

Suppose we observe Ω\Omega and PΩ(L∗)P_{\Omega}(L^{*}) where Assumptions 1,2 hold for L∗L^{*} and Ω\Omega. Also, let E[∣Ω∣]≥Cα2μ4r2nlog⁡2nlog⁡2σ1/ϵE[|\Omega|]\geq C\alpha^{2}\mu^{4}r^{2}n\log^{2}n\log^{2}\sigma_{1}/\epsilon and m≤nm\leq n. Then, w.p. ≥1−n−log⁡α2\geq 1-n^{-\log\frac{\alpha}{2}}, Algorithm 1 outputs L^\hat{L} s.t. ∥L^−L∗∥2≤ϵ\|\hat{L}-L^{*}\|_{2}\leq\epsilon.

Table 1 compares our sample and time complexity bounds for low-rank MC. Note that our sample complexity is nearly the same as that of nuclear-norm methods while the running time of our algorithm is significantly better than the existing results that have at most logarithmic dependence on the condition number of L∗L^{*}.

Result for Robust PCA: Consider the standard Robust PCA problem (RPCA), where the goal is to recover L∗L^{*} from M=L∗+S~∗M=L^{*}+\widetilde{S}^{*}. For RPCA as well, we can randomly sample ∣Ω∣|\Omega| entries from MM, where Ω\Omega satisfies the assumption required by Theorem 1. This leads us to the following corollary:

Suppose we observe M=L∗+S~∗M=L^{*}+\widetilde{S}^{*}, where Assumptions 1, 3 hold for L∗L^{*} and S~∗\widetilde{S}^{*}. Generate Ω∈[m]×[n]\Omega\in[m]\times[n] by sampling each entry uniformly at random with probability pp, s.t., E[∣Ω∣]≥Cα2μ4r2nlog⁡2nlog⁡2σ1/ϵE[|\Omega|]\geq C\alpha^{2}\mu^{4}r^{2}n\log^{2}n\log^{2}\sigma_{1}/\epsilon. Let m≤nm\leq n. Then, w.p. ≥1−n−log⁡α2\geq 1-n^{-\log\frac{\alpha}{2}}, Algorithm 1 outputs L^\hat{L} s.t. ∥L^−L∗∥2≤ϵ\|\hat{L}-L^{*}\|_{2}\leq\epsilon.

Hence, using Theorem 1, we will still be able to recover L∗L^{*} but using only the sampled entries. Moreover, the running time of the algorithm is only O(μ2nr3log⁡2nlog⁡2(σ1/ϵ))O(\mu^{2}nr^{3}\log^{2}n\log^{2}(\sigma_{1}/\epsilon)), i.e., we are able to solve RPCA problem in time linear in nn. To the best of our knowledge, the existing state-of-the-art methods for RPCA require at least O(n2r)O(n^{2}r) time to perform the same task [NUNS+14, GWL16]. Similarly, we don’t need to load the entire data matrix in memory, but we can just sample the matrix and work with the obtained sparse matrix with at most linear number of entries. Hence, our method significantly reduces both time and space complexity, and as demonstrated empirically in Section 4 can help scale our algorithm to very large data sets without losing accuracy.

We now provide an outline of our proof for Theorem 1 and motivate some of our proof techniques; the proof of Theorem 2 follows similarly. Recall that we assume that M=L∗+S~∗M=L^{*}+\widetilde{S}^{*} and define S∗=PΩ(S~∗)S^{*}=\mathcal{P}_{\Omega}(\widetilde{S}^{*}). Similarly, we define S~(t)=HTζ(M−L(t))\widetilde{S}^{(t)}=\mathcal{HT}_{\zeta}(M-L^{(t)}). Critically, S(t)=PΩ(S~(t))S^{(t)}=\mathcal{P}_{\Omega}(\widetilde{S}^{(t)}) (see Line 9 of Algorithm 1), i.e., S~(t)\widetilde{S}^{(t)} is the set of iterates that we “could” obtain if entire MM was observed. Note that we cannot compute S~(t)\widetilde{S}^{(t)}, it is introduced only to simplify our analysis.

We first re-write the projected gradient descent step for L(t+1)L^{(t+1)} as described in (4):

That is, L(t+1)L^{(t+1)} is obtained by rank-kqk_{q} SVD of a perturbed version of L∗L^{*}: L∗+E1+E3L^{*}+E_{1}+E_{3}. As we perform entrywise thresholding to reduce ∥S~∗−S~(t)∥∞\|\widetilde{S}^{*}-\widetilde{S}^{(t)}\|_{\infty}, we need to bound ∥L(t+1)−L∗∥∞\|L^{(t+1)}-L^{*}\|_{\infty}. To this end, we use techniques from [JN15], [NUNS+14] that explicitly model singular vectors of L(t+1)L^{(t+1)} and argue about the infinity norm error using a Taylor series expansion. However, in our case, such an error analysis requires analyzing the following key quantities (H=E1+E3H=E_{1}+E_{3}):

Note that E1=0E_{1}=0 in the case of standard RPCA which was analyzed in [NUNS+14], while E3=0E_{3}=0 in the case of standard MC which was considered in [JN15]. In contrast, in our case both E1E_{1} and E3E_{3} are non-zero. Moreover, E3E_{3} is dependent on random variable Ω\Omega. Hence, for j≥2j\geq 2, we will get cross terms between E3E_{3} and E1E_{1} that will also have dependent random variables which precludes application of standard Bernstein-style tail bounds. To this end, we use a technique similar to that of [EKYY13, JN15] to provide a careful combinatorial-style argument to bound the above given quantity. That is, we can provide the following key lemma:

Let L∗L^{*}, Ω\Omega, and S~∗\widetilde{S}^{*} satisfy Assumptions 1, 2 and 3 respectively. Let L∗=U∗Σ∗(V∗)⊤L^{*}=U^{*}\Sigma^{*}(V^{*})^{\top} be the singular value decomposition of L∗L^{*}. Furthermore, suppose that in the ttht^{\textrm{th}} iteration of the qthq^{\textrm{th}} stage, S~(t)\widetilde{S}^{(t)} defined as HTζ(M−L(t))HT_{\zeta}(M-L^{(t)}) satisfies Supp(S~(t))⊆Supp(S~∗)Supp(\widetilde{S}^{(t)})\subseteq Supp(\widetilde{S}^{*}), then we have:

∀c>0\forall c>0 w.p ≥1−n−2log⁡c4+4\geq 1-n^{-2\log\frac{c}{4}+4}, where E1,E2 and E3E_{1},E_{2}\text{ and }E_{3} are defined in (6), Aa,Ba,Ca,DaA_{a},B_{a},C_{a},D_{a} are defined in (3.1).

Remark: We would like to note that even for the standard MC setting, i.e., when E1=0E_{1}=0, we obtain better bound than that of [JN15] as we can bound max⁡i∥eiT(E3)qU∥2\max_{i}\|e_{i}^{T}(E_{3})^{q}U\|_{2} directly rather than the weaker rmax⁡i∥eiT(E3)quj∥\sqrt{r}\max_{i}\|e_{i}^{T}(E_{3})^{q}u_{j}\| bound that [JN15] uses.

Now, using Lemmas 1 and 7 and by using a hard-thresholding argument we can bound ∥L(t+1)−L∗∥∞≤2μ2rm(σkq+1∗+(12)tσkq∗)\|L^{(t+1)}-L^{*}\|_{\infty}\leq\frac{2\mu^{2}r}{m}(\sigma^{*}_{k_{q}+1}+\left(\frac{1}{2}\right)^{t}\sigma^{*}_{k_{q}}) (see Lemma 9) in the qq-th stage. Hence, after O(log⁡(σ1∗/ϵ))O(\log(\sigma^{*}_{1}/\epsilon)) “inner” iterations, we can guarantee in the qq-th stage:

Moreover, by using sparsity of S~∗\widetilde{S}^{*} and the special structure of E3E_{3} (See Lemma 7), we have: ∥E1+E3∥2≤c⋅σkq+1∗{\|E_{1}+E_{3}\|_{2}}\leq c\cdot\sigma^{*}_{k_{q}+1}, where cc is a small constant.

Now, the outer iteration sets the next stage’s rank kq+1k_{q+1} as: kq+1=∣{i:σi(L∗+E1+E3)≥0.5⋅σkq+1(L∗+E1+E3)}∣k_{q+1}=|\{i:\sigma_{i}(L^{*}+E_{1}+E_{3})\geq 0.5\cdot\sigma_{k_{q}+1}(L^{*}+E_{1}+E_{3})\}|. Hence, using bound on ∥E1+E3∥2\|E_{1}+E_{3}\|_{2} and Weyl’s eigenvalue perturbation bound (Lemma 2), we have: σkq+1∗≥0.6 σkq+1∗\sigma^{*}_{k_{q+1}}\geq 0.6\,\sigma^{*}_{k_{q}+1} and σkq+1∗≤σkq∗\sigma^{*}_{k_{q}+1}\leq\sigma^{*}_{k_{q}}. Hence, after Q=O(log⁡(σ1∗/ϵ))Q=O(\log(\sigma_{1}^{*}/\epsilon)) “outer” iterations, Algorithm 1 converges to an ϵ\epsilon-approximate solution to L∗L^{*}.

Experiments

In this section we discuss the performance of Algorithm 1 on synthetic data and its use in foreground background separation. The goal of the section is two-fold: a) to demonstrate practicality and effectiveness of Algorithm 1 for the RMC problem, b) to show that Algorithm 1 indeed solves RPCA problem in significantly smaller time than that required by the existing state-of-the-art algorithm (St-NcRPCA [NUNS+14]). To this end, we use synthetic data as well as video datasets where the goal is to perform foreground-background separation [CLMW11].

We implemented our algorithm in MATLAB and the results for the synthetic data set were obtained by averaging over 20 runs. We obtained a matlab implementation of St-NcRPCA [NUNS+14] from the authors of [NUNS+14]. Note that if the sampling probability is p=1p=1, then our method is similar to St-NcRPCA; the key difference being how rank is selected in each stage. We also implemented the Alternating Minimzation based algorithm from [GWL16]. However, we found it to be an order of magnitude slower than Algorithm 1 on the foreground-background separation task. For example, on the escalator video, the algorithm did not converge in less than 150 seconds despite discounting for the expensive sorting operation in the truncation step. On the other hand, our algorithm finds the foreground in about 8 seconds.

Parameters. The algorithm has three main parameters: 1) threshold η\eta, 2) incoherence μ\mu and 3) sampling probability pp (E[∣Ω∣]=p⋅mnE[|\Omega|]=p\cdot mn). In the experiments on synthetic data we observed that keeping λ∼μ∥M−S(t)∥2/n\lambda\sim\mu\left\lVert M-S^{(t)}\right\rVert_{2}/{\sqrt{n}} speeds up the recovery while for background extraction keeping λ∼μ∥M−S(t)∥2/n\lambda\sim\mu\left\lVert M-S^{(t)}\right\rVert_{2}/{n} gives a better quality output. The value of μ\mu for real world data sets was figured out using cross validation while for the synthetic data the same value was used as used in data generation. The sampling probability for the synthetic data could be kept as low as 2rlog⁡2(n)/n2r\log^{2}(n)/{n} while for the real world data set we got good results for p=0.05p=0.05. Also, rather than splitting samples, we use entire set of observed entries to perform our updates (see Algorithm 1).

Figure 1 (a) plots recovery error (∥L−L∗∥F\|L-L^{*}\|_{F}) vs computational time for our PG-RMC method (with different sampling probabilities) as well as the St-NcRPCA algorithm. Note that even for very small values of sampling pp, we can achieve same recovery error using significantly small values. For example, our method with p=0.1p=0.1 achieve 0.010.01 error (∥L−L∗∥F\|L-L^{*}\|_{F}) in ≈2.5s\approx 2.5s while St-NcRPCA method requires ≈10s\approx 10s to achieve the same accuracy. Note that we do not compare against the convex relaxation based methods like IALM from [CLMW11], as [NUNS+14] shows that St-NcRPCA is significantly faster than IALM and several other convex relaxation solvers.

Figure 1 (b) plots time required to achieve different recovery errors (∥L−L∗∥F\|L-L^{*}\|_{F}) as the sampling probability pp increases. As expected, we observe a linear increase in the run-time with pp. Interestingly, for very small values of pp, we observe an increase in running time. In this regime, ∥PΩ(M)∥2p\frac{\|\mathcal{P}_{\Omega}(M)\|_{2}}{p} becomes very large (as pp doesn’t satisfy the sampling requirements). Hence, increase in the number of iterations (T≈log⁡∥PΩ(M)∥2pϵT\approx\log\frac{\|\mathcal{P}_{\Omega}(M)\|_{2}}{p\epsilon}) dominates the decrease in per iteration time complexity.

Figure 1 (c), (d) plots computation time required by our method (PG-RMC , Algorithm 1) versus rank and incoherence, respectively. As expected, as these two problem parameters increase, our method requires more time. Note that our run-time dependence on rank seems to be linear, while our existing results require O(r3)O(r^{3}) time. This hints at the possibility of further improving the computational complexity analysis of our algorithm.

We also study phase transition for different values of sampling probability pp. Figure 3 (a) in Appendix 5.5 show a phase transition phenomenon where beyond p>.06p>.06 the probability of recovery is almost 11 while below it, it is almost .

Foreground-background separation. We also applied our technique to the problem of foreground-background separation. We use the usual method of stacking up the vectorized video frames to construct a matrix. The background, being static, will form the low rank component while the foreground can be considered to be the noise.

We applied our PG-RMC method (with varying pp) to several videos. Figure 2 (a), (d) shows one frame each from two videos (a shopping center video, a restaurant video). Figure 2 (b), (d) shows the extracted background from the two videos by using our method (PG-RMC , Algorithm 1) with probability of sampling p=0.05p=0.05. Figure 2 (c), (f) compares objective function value for different pp values. Clearly, PG-RMC can recover the true background with pp as small as 0.050.05. We also observe an order of magnitude speedup (≈5\approx 5x) over St-NcRPCA [NUNS+14]. We present results on the video Escalator in Appendix 5.5.

Conclusion. In this work, we studied the Robust Matrix Completion problem. For this problem, we provide exact recovery of the low-rank matrix L∗L^{*} using nearly optimal number of observations as well as nearly optimal fraction of corruptions in the observed entries. Our RMC result is based on a simple and efficient PGD algorithm that has nearly linear time complexity as well. Our result improves state-of-the-art for the related Matrix Completion as well as Robust PCA problem. For Robust PCA, we provide first nearly linear time algorithm under standard assumptions.

Our sample complexity depends on ϵ\epsilon, the desired accuracy in L∗L^{*}. Moreover, improving dependence of sample complexity on rr (from r2r^{2} to rr) also represents an important direction. Finally, similar to foreground background separation, we would like to explore more applications of RMC/RPCA.

References

Appendix

We divide this section into five parts. In the first part we prove some common lemmas. In the second part we give the convergence guarantee for PG-RMC . In the third part we give another algorithm which has a sample complexity of O(μ4r3nlog⁡2nlog⁡μ2rσ1∗ϵ)O(\mu^{4}r^{3}n\log^{2}{n}\log{\frac{\mu^{2}r\sigma_{1}^{*}}{\epsilon}}) and prove its convergence guarantees. In the fourth part we prove a generalized form of lemma 1. In the fifth part we present some additional experiments.

For the sake of convenience in the following proofs, we will define some notations here.

We define p=∣Ωk,t∣mnp=\frac{\left\lvert\Omega_{k,t}\right\rvert}{mn} and we consider the following equivalent update step for L(t+1)L^{(t+1)} in the analysis:

The singular values of L∗L^{*} are denoted by σ1∗,…,σr∗\sigma^{*}_{1},\ldots,\sigma^{*}_{r} where ∣σ1∗∣≥…≥∣σr∗∣\left\lvert\sigma^{*}_{1}\right\rvert\geq\ldots\geq\left\lvert\sigma^{*}_{r}\right\rvert and we will let λ1,…,λn\lambda_{1},\ldots,\lambda_{n} denote the singular values of M(t)M^{(t)} where ∣λ1∣≥…≥∣λn∣\left\lvert\lambda_{1}\right\rvert\geq\ldots\geq\left\lvert\lambda_{n}\right\rvert.

We will begin by restating some lemmas from previous work that we will use in our proofs.

First, we restate Weyl’s perturbation lemma from [Bha97], a key tool in our analysis:

This lemma establishes a bound on the spectral norm of a sparse matrix.

For any pair of unit vectors uu and vv, we have:

Lemma now follows by using ∥S∥2=max⁡u,v,∥u∥2=1,∥v∥2=1uTSv\|S\|_{2}=\max_{u,v,\|u\|_{2}=1,\|v\|_{2}=1}u^{T}Sv. ∎

Now, we define a -mean random matrix with small higher moments values.

HH is a random matrix of size m×nm\times n with each of its entries drawn independently satisfying the following moment conditions:

for i,j∈[n]i,j\in[n] and 2≤k≤2log⁡n2\leq k\leq 2\log n.

We now restate two useful lemmas from [JN15]:

Suppose HH satisfies Definition 1. Then, w.p. ≥1−1/n10+log⁡α\geq 1-1/n^{10+\log\alpha}, we have: ∥H∥2≤3α.\left\lVert H\right\rVert_{2}\leq 3\sqrt{\alpha}.

Let AA be a m×nm\times n matrix with n≥mn\geq m. Suppose Ω⊆[m]×[n]\Omega\subseteq[m]\times[n] is obtained by sampling each element with probability p∈[14n,0.5]p\in\left[\frac{1}{4n},0.5\right]. Then, the following matrix HH satisfies Defintion 1:

∥A−AUΛ−1U⊤A∥2≤∣σk∣+5∥C∥2\left\lVert A-AU\Lambda^{-1}U^{\top}A\right\rVert_{2}\leq\left\lvert\sigma_{k}\right\rvert+5\left\lVert C\right\rVert_{2},

∥AUΛ−aU⊤A∥2≤4(∣σk∣2)−a+2∀a≥2\left\lVert AU\Lambda^{-a}U^{\top}A\right\rVert_{2}\leq 4\left(\frac{\left\lvert\sigma_{k}\right\rvert}{2}\right)^{-a+2}\quad\forall a\geq 2.

We now provide a lemma that bounds ∥⋅∥∞\|\cdot\|_{\infty} norm of an incoherent matrix with its operator norm.

Let A=UΣV⊤A=U\Sigma V^{\top}. Then, ACA=UU⊤ACAVV⊤ACA=UU^{\top}ACAVV^{\top}. The lemma now follows by using definition of incoherence with the fact that ∥U⊤ACAV∥2≤∥ACA∥2\|U^{\top}ACAV\|_{2}\leq\|ACA\|_{2}. ∎

We now present a lemma that shows improvement in the error ∥L−L∗∥∞\|L-L^{*}\|_{\infty} by using gradient descent on L(t)L^{(t)}.

Let L∗L^{*}, Ω\Omega, S~∗\widetilde{S}^{*} satisfy Assumptions 1,2,3 respectively. Also, let the following hold for the tt-th inner-iteration of any stage qq:

∥L∗−L(t)∥∞≤2μ2rm(σk+1∗+(12)zσk∗)\left\lVert L^{*}-L^{(t)}\right\rVert_{\infty}\leq\frac{2\mu^{2}r}{m}\left(\sigma^{*}_{k+1}+\left(\frac{1}{2}\right)^{z}\sigma^{*}_{k}\right)

∥S~∗−S~(t)∥∞≤8μ2rm(σk+1∗+(12)zσk∗)\left\lVert\widetilde{S}^{*}-\widetilde{S}^{(t)}\right\rVert_{\infty}\leq\frac{8\mu^{2}r}{m}\left(\sigma^{*}_{k+1}+\left(\frac{1}{2}\right)^{z}\sigma^{*}_{k}\right)

Supp(S~(t))⊆Supp(S~∗)Supp(\widetilde{S}^{(t)})\subseteq Supp(\widetilde{S}^{*})

where z≥−3z\geq-3 and σk∗\sigma_{k}^{*} and σk+1∗\sigma_{k+1}^{*} are the kk and (k+1)th(k+1)^{\textit{th}} singular values of L∗L^{*}. Also, let E1=S~(t)−S~∗E_{1}=\widetilde{S}^{(t)}-\widetilde{S}^{*} and E3=(I−PΩq,tp)(L(t)−L∗+S~(t)−S~∗)E_{3}=\left(\mathcal{I}-\frac{\mathcal{P}_{\Omega_{q,t}}}{p}\right)\left(L^{(t)}-L^{*}+\widetilde{S}^{(t)}-\widetilde{S}^{*}\right) be the error terms defined also in (6). Then, the following holds w.p ≥1−n−(10+log⁡α)\geq 1-n^{-(10+\log\alpha)}:

satisfies definition 1 with β=2np⋅∥L(t)−L∗+S~(t)−S~∗∥∞\beta=\frac{2\sqrt{n}}{\sqrt{p}}\cdot\|L^{(t)}-L^{*}+\widetilde{S}^{(t)}-\widetilde{S}^{*}\|_{\infty}.

We now bound the spectral norm of E1+E3E_{1}+E_{3} as follows:

where (ζ1)(\zeta_{1}) follows from Lemma 3 and 4. (ζ2)(\zeta_{2}) follows by our assumptions on ρ\rho, ∥L(t)−L∗∥∞\left\lVert L^{(t)}-L^{*}\right\rVert_{\infty}, and ∥S~(t)−S~∗∥∞\left\lVert\widetilde{S}^{(t)}-\widetilde{S}^{*}\right\rVert_{\infty}. (ζ3)(\zeta_{3}) follows from our assumption on pp. ∎

In the following lemma, we prove that the value of the threshold computed using σk(M(t))=σk(L∗+E1+E3)\sigma_{k}(M^{(t)})=\sigma_{k}(L^{*}+E_{1}+E_{3}), where E1,E3E_{1},E_{3} are defined in (6), closely tracks the threshold that we would have gotten had we had access to the true eigenvalues of L∗L^{*}, σk∗\sigma^{*}_{k}.

Let L∗L^{*}, Ω\Omega, S~∗\widetilde{S}^{*} satisfy Assumptions 1,2,3 respectively. Also, let the following hold for the tt-th inner-iteration of any stage qq:

∥L∗−L(t)∥∞≤2μ2rm(σk+1∗+(12)zσk∗)\left\lVert L^{*}-L^{(t)}\right\rVert_{\infty}\leq\frac{2\mu^{2}r}{m}\left(\sigma^{*}_{k+1}+\left(\frac{1}{2}\right)^{z}\sigma^{*}_{k}\right)

∥S~∗−S~(t)∥∞≤8μ2rm(σk+1∗+(12)zσk∗)\left\lVert\widetilde{S}^{*}-\widetilde{S}^{(t)}\right\rVert_{\infty}\leq\frac{8\mu^{2}r}{m}\left(\sigma^{*}_{k+1}+\left(\frac{1}{2}\right)^{z}\sigma^{*}_{k}\right)

Supp(S~(t))⊆Supp(S~∗)Supp(\widetilde{S}^{(t)})\subseteq Supp(\widetilde{S}^{*})

where z≥−3z\geq-3 and σk∗\sigma_{k}^{*} and σk+1∗\sigma_{k+1}^{*} are the kk and (k+1)th(k+1)^{\textit{th}} singular values of L∗L^{*}. Also, let E1=S~(t)−S~∗E_{1}=\widetilde{S}^{(t)}-\widetilde{S}^{*} and E3=(I−PΩq,tp)(L(t)−L∗+S~(t)−S~∗)E_{3}=\left(\mathcal{I}-\frac{\mathcal{P}_{\Omega_{q,t}}}{p}\right)\left(L^{(t)}-L^{*}+\widetilde{S}^{(t)}-\widetilde{S}^{*}\right) be the error terms defined also in (6). Then, the following holds ∀z>−3\forall z>-3 w.p ≥1−n−(10+log⁡α)\geq 1-n^{-(10+\log\alpha)}:

where λk:=σk(M(t))=σk(L∗+E1+E3)\lambda_{k}:=\sigma_{k}(M^{(t)})=\sigma_{k}(L^{*}+E_{1}+E_{3}) and E1,E3E_{1},E_{3} are defined in (6).

Using Weyl’s inequality (Lemma 2), we have: : ∣λk−σk∗∣≤∥E1+E3∥2\left\lvert\lambda_{k}-\sigma^{*}_{k}\right\rvert\leq\|E_{1}+E_{3}\|_{2} and ∣λk+1−σk+1∗∣≤∥E1+E3∥2\left\lvert\lambda_{k+1}-\sigma^{*}_{k+1}\right\rvert\leq\|E_{1}+E_{3}\|_{2} We now proceed to prove the lemma as follows:

where (ζ)(\zeta) follows from Lemma 7 and the last inequality follows from the assumption that z≥−3z\geq-3. ∎

Next, we show that the projected gradient descent update (6) leads to a better estimate of L∗L^{*}, i.e., we bound ∥L(t+1)−L∗∥∞\|L^{(t+1)}-L^{*}\|_{\infty}. Under the assumptions of the below given Lemma, the proof follows arguments similar to [NUNS+14] with additional challenge arises due to more involved error terms E1E_{1}, E3E_{3}.

Our proof proceeds by first symmetrizing our matrices by rectangular dilation. We first begin by noting some properties of symmetrized matrices used in the proof of the following lemma.

Let AA be a m×nm\times n dimensional matrix with singular value decomposition UΣV⊤U\Sigma V^{\top}. We denote its symmetrized version be As≔[0A⊤A0]A_{s}\coloneqq\begin{bmatrix}0&A^{\top}\\ A&0\end{bmatrix}. Then:

The eigenvalue decomposition of AsA_{s} is given by As=UsΣsUs⊤A_{s}=U_{s}\Sigma_{s}U_{s}^{\top} where

P2k(As)=[0Pk(A⊤)Pk(A)0]\mathcal{P}_{2k}\left(A_{s}\right)=\begin{bmatrix}0&\mathcal{P}_{k}(A^{\top})\\ \mathcal{P}_{k}(A)&0\end{bmatrix}

We have As2j=[(A⊤A)j00(AA⊤)j]A_{s}^{2j}=\begin{bmatrix}(A^{\top}A)^{j}&0\\ 0&(AA^{\top})^{j}\end{bmatrix} As2j+1=[0(A⊤A)jA⊤(AA⊤)jA0]A_{s}^{2j+1}=\begin{bmatrix}0&(A^{\top}A)^{j}A^{\top}\\ (AA^{\top})^{j}A&0\end{bmatrix}

Let L(t)=Pk(L∗+H)L^{(t)}=P_{k}(L^{*}+H), where HH is any perturbation matrix that satisfies the following:

∥H∥2≤σk∗4\left\lVert H\right\rVert_{2}\leq\frac{\displaystyle\sigma^{*}_{k}}{\displaystyle 4}

∀i∈[n], a∈⌈log⁡n2⌉\forall i\in[n],\ a\in\lceil\frac{\log n}{2}\rceil with υ≤σk∗4\upsilon\leq\frac{\displaystyle\sigma^{*}_{k}}{\displaystyle 4}

where σk∗\sigma_{k}^{*} is the kthk^{\textit{th}} singular value of L∗L^{*}. Also, let L∗L^{*} satisfy Assumption 1. Then, the following holds:

where μ\mu and rr are the rank and incoherence of the matrix L∗L^{*} respectively.

Let l=m+nl=m+n. Let λ1,⋯ ,λl\lambda_{1},\cdots,\lambda_{l} be the eigenvalues of Ms(t)=Ls∗+HsM^{(t)}_{s}=L^{*}_{s}+H_{s} with ∣λ1∣≥∣λ2∣⋯≥∣λl∣\left\lvert\lambda_{1}\right\rvert\geq\left\lvert\lambda_{2}\right\rvert\cdots\geq\left\lvert\lambda_{l}\right\rvert. Let u1,u2,⋯ ,ulu_{1},u_{2},\cdots,u_{l} be the corresponding eigenvectors of Ms(t)M^{(t)}_{s}. Using Lemma 2 along with the assumption on ∥Hs∥2\left\lVert H_{s}\right\rVert_{2}, we have: ∣λ2k∣≥3σk∗4|\lambda_{2k}|\geq\frac{3\sigma^{*}_{k}}{4}.

Let UΛVU\Lambda V be the eigen vector decomposition of L(t+1)L^{(t+1)}. Let UsΛsUs⊤U_{s}\Lambda_{s}U_{s}^{\top} to be the eigen vector decomposition of Ls(t+1)L^{(t+1)}_{s}. Then, using Remark 1 we have ∀ i∈[2k]\forall\ i\in[2k]:

As ∣λ2k∣≥3σk∗4\left\lvert\lambda_{2k}\right\rvert\geq\frac{3\sigma^{*}_{k}}{4} and ∥Hs∥2≤14σk∗\left\lVert H_{s}\right\rVert_{2}\leq\frac{1}{4}\sigma_{k}^{*}, we can apply the Taylor’s series expansion to get the following expression for uiu_{i}:

Subtracting Ls∗L^{*}_{s} on both sides and taking operator norm, we get:

We separately bound the first and the second term of RHS. The first term can be bounded as follows:

where (ζ1)(\zeta_{1}) follows Remark 1, (ζ2)(\zeta_{2}) from Lemma 6 and (ζ3)(\zeta_{3}) follows from Claim 1 of Lemma 5.

We now bound second term of RHS of (10) which we again split in two parts. We first bound the terms with s+t>log⁡ns+t>\log n:

where (ζ1)(\zeta_{1}) follows from the second claim of Lemma 5 and noting that ∥Hs∥2=∥H∥2\left\lVert H_{s}\right\rVert_{2}=\left\lVert H\right\rVert_{2} and (ζ2)(\zeta_{2}) follows from assumption on ∥H∥2\left\lVert H\right\rVert_{2} and using the fact that s+t≥log⁡ns+t\geq\log n.

Summing up over all terms with s+t>log⁡ns+t>\log n, we get from 13 and 12:

Now, for terms corresponding to 1≤s+t≤log⁡n1\leq s+t\leq\log n, we have:

where (ζ1)(\zeta_{1}) follows from assumption on HH in the lemma statement, (ζ2)(\zeta_{2}) follows from Claim 2 of Lemma 5.

It now remains to bound the terms, max⁡q1∈[m+n]∥eq1⊤HssUs∗∥2\max\limits_{q_{1}\in[m+n]}\left\lVert e_{q_{1}}^{\top}H_{s}^{s}U_{s}^{*}\right\rVert_{2}. Note from Remark 1.1 that Us∗=12[V∗V∗U∗−U∗]U_{s}^{*}=\frac{1}{\sqrt{2}}\begin{bmatrix}V^{*}&V^{*}\\ U^{*}&-U^{*}\end{bmatrix}. Now, we have the following cases for HssH^{s}_{s}:

This leads to the following 4 cases for max⁡q1∈[m+n]∥eq1⊤HssUs∗∥2\max\limits_{q_{1}\in[m+n]}\left\lVert e_{q_{1}}^{\top}H_{s}^{s}U_{s}^{*}\right\rVert_{2}:

we get the bound on these terms in Lemma 15. Also, note from the Remark 1.2 that ∥Ls∗−Ls(t+1)∥∞=∥L∗−L(t+1)∥∞\left\lVert L^{*}_{s}-L^{(t+1)}_{s}\right\rVert_{\infty}=\left\lVert L^{*}-L^{(t+1)}\right\rVert_{\infty}.

Now, summing up 15 over all 1≤s+t≤log⁡n1\leq s+t\leq\log n and combining with 14 we get the required result. ∎

In the next lemma, we show that with the threshold chosen in the algorithm, we show an improvement in the estimation of S~∗\widetilde{S}^{*} by S~(t+1)\widetilde{S}^{(t+1)}.

In the ttht^{\textit{th}} iterate of the qthq^{\textit{th}} stage, assume the following holds:

∥L∗−L(t)∥∞≤2μ2rm(σk+1∗+(12)zσk∗)\left\lVert L^{*}-L^{(t)}\right\rVert_{\infty}\leq\frac{2\mu^{2}r}{m}\left(\sigma^{*}_{k+1}+\left(\frac{1}{2}\right)^{z}\sigma^{*}_{k}\right)

78(σk+1∗+(12)zσk∗)≤(λk+1+(12)zλk)≤98(σk+1∗+(12)zσk∗)\frac{\displaystyle 7}{\displaystyle 8}\left(\sigma^{*}_{k+1}+\left(\frac{\displaystyle 1}{\displaystyle 2}\right)^{z}\sigma^{*}_{k}\right)\leq\left(\lambda_{k+1}+\left(\frac{\displaystyle 1}{\displaystyle 2}\right)^{z}\lambda_{k}\right)\leq\frac{\displaystyle 9}{\displaystyle 8}\left(\sigma^{*}_{k+1}+\left(\frac{\displaystyle 1}{\displaystyle 2}\right)^{z}\sigma^{*}_{k}\right)

where σk∗\sigma_{k}^{*} and σk+1∗\sigma_{k+1}^{*} are the kk and (k+1)th(k+1)^{\textit{th}} singular values of L∗L^{*}, λk\lambda_{k} and λk+1\lambda_{k+1} are the kk and (k+1)th(k+1)^{\textit{th}} singular values of M(t)M^{(t)} and, rr and μ\mu are the rank and incoherence of the m×nm\times n matrix L∗L^{*} respectively. Then we have

Supp(S~(t))⊆Supp(S~∗)Supp\left(\widetilde{S}^{(t)}\right)\subseteq Supp\left(\widetilde{S}^{*}\right)

∥S~(t)−S~∗∥∞≤8μ2rm(σk+1∗+(12)zσk∗)\left\lVert\widetilde{S}^{(t)}-\widetilde{S}^{*}\right\rVert_{\infty}\leq\frac{8\mu^{2}r}{m}\left(\sigma^{*}_{k+1}+\left(\frac{1}{2}\right)^{z}\sigma^{*}_{k}\right)

We first prove the first claim of the lemma. Consider an index pair (i,j)∉Supp(S~∗)(i,j)\notin Supp(\widetilde{S}^{*}).

where (ζ1)(\zeta_{1}) follows from the second assumption. Hence, we do not threshold any entry that is not corrupted by S~∗\widetilde{S}^{*}.

Now, we prove the second claim of the lemma. Consider an index entry (i,j)∈Supp(S~∗)(i,j)\in Supp(\widetilde{S}^{*}). Here, we consider two cases:

The entry (i,j)∈Supp(S~(t))(i,j)\in Supp(\widetilde{S}^{(t)}): Here the entry (i,j)(i,j) is thresholded. We know that Lij(t)+S~ij(t)=Lij∗+S~ij∗L^{(t)}_{ij}+\widetilde{S}^{(t)}_{ij}=L^{*}_{ij}+\widetilde{S}^{*}_{ij} from which we get

The entry (i,j)∉Supp(S~(t))(i,j)\notin Supp(\widetilde{S}^{(t)}): Here the entry (i,j)(i,j) is not thresholded. We know that ∣Lij∗+S~ij∗−Lij(t)∣≤ζ\left\lvert L^{*}_{ij}+\widetilde{S}^{*}_{ij}-L^{(t)}_{ij}\right\rvert\leq\zeta from which we get

where (ζ2)(\zeta_{2}) follows from the second assumption along with the assumption about η=μ2rm\eta=\frac{\mu^{2}r}{m}.

The above two cases prove the second statement of the lemma. ∎

The number of iteration TT in the inner loop of Algorithm 1 and Algorithm 3 satisfy:

w.p ≥1−n−(10+log⁡α)\geq 1-n^{-(10+\log\alpha)}. Here σ1∗\sigma_{1}^{*} is the highest singular value of L∗L^{*}, rr is it’s rank and μ\mu is it’s incoherence.

where (ζ1)(\zeta_{1}) follows from Lemma 7. ∎

We will now prove Lemma 1 Proof of Lemma 1: Recall the definitions of E1=(S~∗−S~(t))E_{1}=\left(\widetilde{S}^{*}-\widetilde{S}^{(t)}\right) , E2=(L(t)−L∗)E_{2}=\left(L^{(t)}-L^{*}\right) , E3=(I−PΩq,tp)(E2−E1)E_{3}=\left(\mathcal{I}-\frac{\mathcal{P}_{\Omega_{q,t}}}{p}\right)\left(E_{2}-E_{1}\right) and β=2np∥E2−E1∥∞\beta=2\sqrt{\frac{n}{p}}\left\lVert E_{2}-E_{1}\right\rVert_{\infty}. Recall that H≔E1+E3H\coloneqq E_{1}+E_{3} From Lemma 4, we have that 1βE3\frac{1}{\beta}E_{3} satisfies Definition 1. This implies that the matrix 1β(E1+E3)\frac{1}{\beta}\left(E_{1}+E_{3}\right) satisfies the conditions of Lemma 15. Now, we have ∀1≤a≤⌈log⁡n⌉\forall 1\leq a\leq\lceil\log n\rceil and ∀i∈[n]\forall i\in[n]:

where (ζ)(\zeta) follows from the application of Lemma 15 along with the incoherence assumption on U∗U^{*}. The other statements of the lemma can be proved in a similar manner by invocations of the different claims of Lemma 15. □\Box

2 Algorithm PG-RMC

Proof of Theorem 1: From Lemma 11 we know that T≥log⁡(3μ2rσ1∗ϵ)T\geq\log(\frac{3\mu^{2}r\sigma_{1}^{*}}{\epsilon}). Consider the stage qq reached at the termination of the algorithm. We know from Lemma 12 that:

∥E(T)∥∞≤8μ2rm(σkq+1∗+(12)Tσkq∗)≤8μ2rmσkq+1∗+ϵ10n\left\lVert E^{(T)}\right\rVert_{\infty}\leq\frac{8\mu^{2}r}{m}\left(\sigma^{*}_{k_{q}+1}+\left(\frac{1}{2}\right)^{T}\sigma^{*}_{k_{q}}\right)\leq\frac{8\mu^{2}r}{m}\sigma^{*}_{k_{q}+1}+\frac{\epsilon}{10n}

∥L(T)−L∗∥∞≤2μ2rm(σkq+1∗+(12)T∣σkq∗∣)≤2μ2rmσkq+1∗+ϵ10n\left\lVert L^{(T)}-L^{*}\right\rVert_{\infty}\leq\frac{2\mu^{2}r}{m}\left(\sigma^{*}_{k_{q}+1}+\left(\frac{1}{2}\right)^{T}\left\lvert\sigma^{*}_{k_{q}}\right\rvert\right)\leq\frac{2\mu^{2}r}{m}\sigma^{*}_{k_{q}+1}+\frac{\epsilon}{10n}

Combining this with Lemmas 2 and 7, we get:

When the while loop terminates, ησkq+1(M(T))<ϵ2n\eta\sigma_{k_{q}+1}\left(M^{(T)}\right)<\frac{\epsilon}{2n}, which from 16, implies that σkq+1∗<mϵ7nμ2r\sigma_{k_{q}+1}^{*}<\frac{m\epsilon}{7n\mu^{2}r}. So we have:

We will now bound the number of iterations required for the PG-RMC to converge.

From claim 2 of Lemma 13, we have σkq+1∗≤1732σkq−1+1∗∀q≥1\sigma^{*}_{k_{q}+1}\leq\frac{17}{32}\sigma^{*}_{k_{q-1}+1}\quad\forall q\geq 1. By recursively applying this inequality, we get σkq+1∗≤(1732)qσ1∗\sigma^{*}_{k_{q}+1}\leq\left(\frac{17}{32}\right)^{q}\sigma^{*}_{1}. We know that when the algorithm terminates, σkq+1∗<ϵ7μ2r\sigma_{k_{q}+1}^{*}<\frac{\epsilon}{7\mu^{2}r}. Since, (1732)qσ1∗\left(\frac{17}{32}\right)^{q}\sigma^{*}_{1} is an upper bound for σkq+1∗\sigma_{k_{q}+1}^{*}, an upper bound for the number of iterations is 5log⁡(7μ2rσ1∗ϵ)5\log\left(\frac{7\mu^{2}r\sigma_{1}^{*}}{\epsilon}\right). Also, note that an upper bound to this quantity is used to partition the samples provided to the algorithm. This happens with probability ≥1−T2n−(10+log⁡α)≥1−n−log⁡α\geq 1-T^{2}n^{-(10+\log\alpha)}\geq 1-n^{-\log\alpha}. This concludes the proof. □\Box

In the following lemma, we show that we make progress simultaneously in the estimation of both S~∗\widetilde{S}^{*} and L∗L^{*} by S~(t)\widetilde{S}^{(t)} and L(t)L^{(t)}. We make use of Lemmas 9 and 10 to show progress in the estimation of one affects the other alternatively. We also emphasize the roles of the following quantities in enabling us to prove our convergence result:

∥H∥2\left\lVert H\right\rVert_{2} - We use Lemma 7 to bound this quanitity

The analysis of the following 4 quanitities is crucial to obtaining error bounds in ∥∥∞\lVert\rVert_{\infty} norm

Let L∗L^{*}, Ω\Omega, S~∗\widetilde{S}^{*} and S~(t)\widetilde{S}^{(t)} satisfy Assumptions 1,2,3 respectively. Then, in the ttht^{\textrm{th}} iteration of the qthq^{\textrm{th}} stage of Algorithm 1, S~(t)\widetilde{S}^{(t)} and L(t)L^{(t)} satisfy:

with probability ≥1−((q−1)T+t−1)n−(10+log⁡α)\geq 1-((q-1)T+t-1)n^{-(10+\log\alpha)} where TT is the number of iterations in the inner loop.

We prove the lemma by induction on both qq and tt.

Base Case: q=1q=1 and t=0t=0 We begin by first proving an upper bound on ∥L∗∥∞\left\lVert L^{*}\right\rVert_{\infty}. We do this as follows:

where the last inequality follows from Cauchy-Schwartz and the incoherence of U∗U^{*}. This directly proves the third claim of the lemma for the base case. We also note that due to the thresholding step and the incoherence assumption on L∗L^{*}, we have:

∥E(0)∥∞≤8μ2rm(σ2∗+2σ1∗)≤(ζ)8μ2rm(8σk1∗),\mboxand\left\lVert{E}^{(0)}\right\rVert_{\infty}\leq\frac{8\mu^{2}r}{m}\left(\sigma_{2}^{*}+2\sigma_{1}^{*}\right)\overset{(\zeta)}{\leq}\frac{8\mu^{2}r}{m}\left(8\sigma_{k_{1}}^{*}\right),\mbox{ and }

Supp(S~(t))=Supp(S~∗).\textrm{Supp}\left(\widetilde{S}^{(t)}\right)=\textrm{Supp}\left(\widetilde{S}^{*}\right).

where (ζ)(\zeta) follows from Lemma 13. So the base case of induction is satisfied.

Induction over tt We first prove the inductive step over tt (for a fixed qq). By inductive hypothesis we assume that:

∥E(t)∥∞≤8μ2rm(σkq+1∗+(12)t−3σkq∗)\left\lVert{E}^{(t)}\right\rVert_{\infty}\leq\frac{8\mu^{2}r}{m}\left(\sigma_{k_{q}+1}^{*}+\left(\frac{1}{2}\right)^{t-3}\sigma_{k_{q}}^{*}\right)

Supp(S~(t))⊆Supp(S~∗)\textrm{Supp}\left(\widetilde{S}^{(t)}\right)\subseteq\textrm{Supp}\left(\widetilde{S}^{*}\right).

∥L∗−L(t)∥∞≤2μ2rm(σkq+1∗+(12)t−3σkq∗)\left\lVert L^{*}-L^{(t)}\right\rVert_{\infty}\leq\frac{2\mu^{2}r}{m}\left(\sigma_{k_{q}+1}^{*}+\left(\frac{1}{2}\right)^{t-3}\sigma_{k_{q}}^{*}\right)

with probability 1−((q−1)T+t−1)n−(10+log⁡α)1-((q-1)T+t-1)n^{-(10+\log\alpha)}. Then by Lemma 9, we have:

where (ζ1)(\zeta_{1}) follows from our assumptions on ρ\rho and our inductive hypothesis on ∥E(t)∥∞\left\lVert{E}^{(t)}\right\rVert_{\infty} and (ζ2)(\zeta_{2}) follows from our assumption on pp and by noticing that ∥D∥∞≤∥E(t)∥∞+∥L∗−L(t)∥∞\left\lVert D\right\rVert_{\infty}\leq\left\lVert{E}^{(t)}\right\rVert_{\infty}+\left\lVert L^{*}-L^{(t)}\right\rVert_{\infty}. Recall that D=L(t)−L∗+S~(t)−S~∗D=L^{(t)}-L^{*}+\widetilde{S}^{(t)}-\widetilde{S}^{*}.

with probability ≥1−n−(10+log⁡α)\geq 1-n^{-(10+\log\alpha)}. From Equations 19, 18 and 17, we have:

which by union bound holds with probability ≥1−((q−1)T+t)n−(10+log⁡α)\geq 1-((q-1)T+t)n^{-(10+\log\alpha)}. Hence, using Lemma 10 and 8 we have:

∥E(t+1)∥∞≤8μ2rm(σkq+1∗+(12)t−2σkq∗)\left\lVert{E}^{(t+1)}\right\rVert_{\infty}\leq\frac{8\mu^{2}r}{m}\left(\sigma_{k_{q}+1}^{*}+\left(\frac{1}{2}\right)^{t-2}\sigma_{k_{q}}^{*}\right)

Supp(S~(t)t+1)⊆Supp(S~∗)\textrm{Supp}\left(\widetilde{S}^{(t)}{t+1}\right)\subseteq\textrm{Supp}\left(\widetilde{S}^{*}\right).

which also holds with probability ≥1−((q−1)T+t)n−(10+log⁡α)\geq 1-((q-1)T+t)n^{-(10+\log\alpha)}. This concludes the proof for induction over tt.

Induction Over Stages qq We now prove the induction over qq. Suppose the hypothesis holds for stage qq. At the end of stage qq, we have:

∥E(T)∥∞≤8μ2rm(σkq+1∗+(12)Tσkq∗)≤8μ2rσkq+1∗m+ϵ10n\left\lVert{E}^{(T)}\right\rVert_{\infty}\leq\frac{8\mu^{2}r}{m}\left(\sigma_{k_{q}+1}^{*}+\left(\frac{1}{2}\right)^{T}\sigma_{k_{q}}^{*}\right)\leq\frac{8\mu^{2}r\sigma_{k_{q}+1}^{*}}{m}+\frac{\epsilon}{10n}, and

Supp(S~(T))⊆Supp(S~∗)\textrm{Supp}\left(\widetilde{S}^{(T)}\right)\subseteq\textrm{Supp}\left(\widetilde{S}^{*}\right).

with probability ≥1−(qT−1)n−(10+log⁡α)\geq 1-(qT-1)n^{-(10+\log\alpha)}. From Lemmas 2 and 7 we get:

with probability 1−n−(10+log⁡α)1-n^{-(10+\log\alpha)}. We know that ησkq+1(M(t))≥ϵ2n\eta\sigma_{k_{q}+1}\left(M^{(t)}\right)\geq\frac{\epsilon}{2n} which with 20 implies that ∣σkq+1∗∣>mϵ10nμ2r\left\lvert\sigma_{k_{q}+1}^{*}\right\rvert>\frac{m\epsilon}{10n\mu^{2}r}.

where (ζ4)(\zeta_{4}) follows from Lemma 13. By union bound this holds with probability ≥1−qTn−(10+log⁡α)\geq 1-qTn^{-(10+\log\alpha)}.

Now, from 8 and 10, we have through a similar series of arguments as above:

which holds with probability ≥1−qTn−(10+log⁡α)\geq 1-qTn^{-(10+\log\alpha)}. ∎

Suppose at the beginning of the qthq^{\text{th}} stage of algorithm 1:

∥L∗−L(0)∥∞≤2μ2rm(2σkq−1+1∗)\left\lVert L^{*}-L^{(0)}\right\rVert_{\infty}\leq\frac{\displaystyle 2\mu^{2}r}{\displaystyle m}\left(2\sigma_{k_{q-1}+1}^{*}\right)

∥E(0)∥∞≤8μ2rm(2σkq−1+1∗)\left\lVert E^{(0)}\right\rVert_{\infty}\leq\frac{\displaystyle 8\mu^{2}r}{\displaystyle m}\left(2\sigma_{k_{q-1}+1}^{*}\right)

σkq∗≥1532σkq−1+1∗\sigma^{*}_{k_{q}}\geq\frac{15}{32}\sigma^{*}_{k_{q-1}+1}

σkq+1∗≤1732σkq−1+1∗\sigma^{*}_{k_{q}+1}\leq\frac{17}{32}\sigma^{*}_{k_{q-1}+1}

with probability ≥1−n−(10+log⁡α)\geq 1-n^{-(10+\log\alpha)}

Combining the three inequalities, we get:

Applying Lemma 7, we get the first claim of the lemma.

Again, combining the three inequalities, we get:

Another application of Lemma 7 gives the second claim. ∎

3 Algorithm R-RMC

Proof of Theorem 2: From Lemma 11 we know that T≥log⁡(3μ2nrσ1∗mϵ)T\geq\log(\frac{3\mu^{2}nr\sigma_{1}^{*}}{m\epsilon}).

Consider the stage qq reached at the termination of the algorithm. We know from Lemma 14 that:

∥E(T)∥∞≤8μ2rm(σq+1∗+(12)Tσq∗)≤8μ2rmσq+1∗+ϵ10n\left\lVert E^{(T)}\right\rVert_{\infty}\leq\frac{8\mu^{2}r}{m}\left(\sigma^{*}_{q+1}+\left(\frac{1}{2}\right)^{T}\sigma^{*}_{q}\right)\leq\frac{8\mu^{2}r}{m}\sigma^{*}_{q+1}+\frac{\epsilon}{10n}

∥L(T)−L∗∥∞≤2μ2rm(σq+1∗+(12)Tσq∗)≤2μ2rmσq+1∗+ϵ10n\left\lVert L^{(T)}-L^{*}\right\rVert_{\infty}\leq\frac{2\mu^{2}r}{m}\left(\sigma^{*}_{q+1}+\left(\frac{1}{2}\right)^{T}\sigma^{*}_{q}\right)\leq\frac{2\mu^{2}r}{m}\sigma^{*}_{q+1}+\frac{\epsilon}{10n}

Combining this with Lemmas 2 and 7, we get:

When the while loop terminates, ησq+1(M(T))<ϵ2n\eta\sigma_{q+1}\left(M^{(T)}\right)<\frac{\epsilon}{2n}, which from 22, implies that σq+1∗<mϵ7nμ2r\sigma_{q+1}^{*}<\frac{m\epsilon}{7n\mu^{2}r}. So we have:

As in the case of the proof of Theorem 1, the following lemma shows that we simultaneously make progress in both the estimation of L∗L^{*} and S~∗\widetilde{S}^{*} by L(t)L^{(t)} and S~(t)\widetilde{S}^{(t)} respectively. Similar to Lemma 12, we make use of Lemmas 10 and 9 to show how improvement in estimation of one of the quantities affects the other and the other five terms, ∥H∥2\left\lVert H\right\rVert_{2}, max⁡q′∈[n]∥eq′⊤(H⊤H)jV∗∥2\max\limits_{q^{\prime}\in[n]}\left\lVert e_{q^{\prime}}^{\top}\left(H^{\top}H\right)^{j}V^{*}\right\rVert_{2}, max⁡q′∈[m]∥eq′⊤(HH⊤)jU∗∥2\max\limits_{q^{\prime}\in[m]}\left\lVert e_{q^{\prime}}^{\top}\left(HH^{\top}\right)^{j}U^{*}\right\rVert_{2}, max⁡q′∈[n]∥eq′⊤H⊤(HH⊤)jU∗∥2\max\limits_{q^{\prime}\in[n]}\left\lVert e_{q^{\prime}}^{\top}H^{\top}\left(HH^{\top}\right)^{j}U^{*}\right\rVert_{2} and max⁡q′∈[m]∥eq′⊤H(H⊤H)jV∗∥2\max\limits_{q^{\prime}\in[m]}\left\lVert e_{q^{\prime}}^{\top}H\left(H^{\top}H\right)^{j}V^{*}\right\rVert_{2} are analyzed the same way:

Let L∗L^{*}, Ω\Omega, S~∗\widetilde{S}^{*} and S~(t)\widetilde{S}^{(t)} satisfy Assumptions 1,2,3 respectively. Then, in the ttht^{\textrm{th}} iteration of the qthq^{\textrm{th}} stage of Algorithm 3, S~(t)\widetilde{S}^{(t)} and L(t)L^{(t)} satisfy:

with probability ≥1−((q−1)T+t−1)n−(10+log⁡α)\geq 1-((q-1)T+t-1)n^{-(10+\log\alpha)} where TT is the number of iterations in the inner loop.

We prove the lemma by induction on both qq and tt.

Base Case: q=1q=1 and t=0t=0 We begin by first proving an upper bound on ∥L∗∥∞\left\lVert L^{*}\right\rVert_{\infty}. We do this as follows:

where the last inequality follows from Cauchy-Schwartz and the incoherence of U∗U^{*}. This directly proves the third claim of the lemma for the base case. We also note that due to the thresholding step and the incoherence assumption on L∗L^{*}, we have:

∥E(0)∥∞≤8μ2rm(σ2∗+2σ1∗)\left\lVert{E}^{(0)}\right\rVert_{\infty}\leq\frac{8\mu^{2}r}{m}\left(\sigma_{2}^{*}+2\sigma_{1}^{*}\right)

Supp(S~(t))=Supp(S~∗).\textrm{Supp}\left(\widetilde{S}^{(t)}\right)=\textrm{Supp}\left(\widetilde{S}^{*}\right).

So the base case of induction is satisfied.

Induction over tt We first prove the inductive step over tt (for a fixed qq). By inductive hypothesis we assume that:

∥E(t)∥∞≤8μ2rm(∣σq+1∗∣+(12)t−1∣σq∗∣)\left\lVert{E}^{(t)}\right\rVert_{\infty}\leq\frac{8\mu^{2}r}{m}\left(|\sigma_{q+1}^{*}|+\left(\frac{1}{2}\right)^{t-1}|\sigma_{q}^{*}|\right)

Supp(S~(t))⊆Supp(S~∗)\textrm{Supp}\left(\widetilde{S}^{(t)}\right)\subseteq\textrm{Supp}\left(\widetilde{S}^{*}\right).

∥L∗−L(t)∥∞≤2μ2rm(∣σq+1∗∣+(12)t−1∣σq∗∣)\left\lVert L^{*}-L^{(t)}\right\rVert_{\infty}\leq\frac{2\mu^{2}r}{m}\left(|\sigma_{q+1}^{*}|+\left(\frac{1}{2}\right)^{t-1}|\sigma_{q}^{*}|\right)

with probability 1−((q−1)T+t−1)n−(10+log⁡α)1-((q-1)T+t-1)n^{-(10+\log\alpha)}.

where (ζ1)(\zeta_{1}) follows from our assumptions on ρ\rho and our inductive hypothesis on ∥E(t)∥∞\left\lVert{E}^{(t)}\right\rVert_{\infty} and (ζ2)(\zeta_{2}) follows from our assumption on pp and by noticing that ∥D∥∞≤∥E(t)∥∞+∥L∗−L(t)∥∞\left\lVert D\right\rVert_{\infty}\leq\left\lVert{E}^{(t)}\right\rVert_{\infty}+\left\lVert L^{*}-L^{(t)}\right\rVert_{\infty}. Recall that D=L(t)−L∗+S~(t)−S~∗D=L^{(t)}-L^{*}+\widetilde{S}^{(t)}-\widetilde{S}^{*}.

with probability ≥1−n−(10+log⁡α)\geq 1-n^{-(10+\log\alpha)}. From Equations 25, 24 and 23, we have:

which by union bound holds with probability ≥1−((q−1)T+t)n−(10+log⁡α)\geq 1-((q-1)T+t)n^{-(10+\log\alpha)}. Hence, using Lemma 10 and 8 we have:

∥E(t+1)∥∞≤8μ2rm(σq+1∗+(12)tσq∗)\left\lVert{E}^{(t+1)}\right\rVert_{\infty}\leq\frac{8\mu^{2}r}{m}\left(\sigma_{q+1}^{*}+\left(\frac{1}{2}\right)^{t}\sigma_{q}^{*}\right)

Supp(S~(t+1))⊆Supp(S~∗)\textrm{Supp}\left(\widetilde{S}^{(t+1)}\right)\subseteq\textrm{Supp}\left(\widetilde{S}^{*}\right).

which also holds with probability ≥1−((q−1)T+t)n−(10+log⁡α)\geq 1-((q-1)T+t)n^{-(10+\log\alpha)}. This concludes the proof for induction over tt.

Induction Over Stages qq We now prove the induction over qq. Suppose the hypothesis holds for stage qq. At the end of stage qq, we have:

∥E(T)∥∞≤8μ2rm(σq+1∗+(12)Tσq∗)≤8μ2rσq+1∗m+ϵ10n\left\lVert{E}^{(T)}\right\rVert_{\infty}\leq\frac{8\mu^{2}r}{m}\left(\sigma_{q+1}^{*}+\left(\frac{1}{2}\right)^{T}\sigma_{q}^{*}\right)\leq\frac{8\mu^{2}r\sigma_{q+1}^{*}}{m}+\frac{\epsilon}{10n}

Supp(S~(T))⊆Supp(S~∗)\textrm{Supp}\left(\widetilde{S}^{(T)}\right)\subseteq\textrm{Supp}\left(\widetilde{S}^{*}\right).

with probability ≥1−(qT−1)n−(10+log⁡α)\geq 1-(qT-1)n^{-(10+\log\alpha)}.

with probability 1−n−(10+log⁡α)1-n^{-(10+\log\alpha)}. We know that ησq+1(M(t))≥ϵ2n\eta\sigma_{q+1}\left(M^{(t)}\right)\geq\frac{\epsilon}{2n} which with 26 implies that σq+1∗>mϵ10nμ2r\sigma_{q+1}^{*}>\frac{m\epsilon}{10n\mu^{2}r}.

By union bound this holds with probability ≥1−qTn−(10+log⁡α)\geq 1-qTn^{-(10+\log\alpha)}.

Now, from 8 and 10, we have through a similar series of arguments as above:

which holds with probability ≥1−qTn−(10+log⁡α)\geq 1-qTn^{-(10+\log\alpha)}. ∎

4 Proof of a generalized form of Lemma 1

with probability n−2log⁡c4+4n^{-2\log{\frac{c}{4}}+4}.

Similar to [JN15], we will prove the statement for q=1q=1 and it can be proved for q∈[n]q\in[n] by taking a union bound over all qq. For the sake of brevity, we will prove only the inequality:

The rest of the lemma follows by applying similar arguments to the appropriate quantities.

Let ω:[2a]→{1,2}\omega:[2a]\rightarrow\{1,2\} be a function used to index a single term in the expansion of (H⊤H)a(H^{\top}H)^{a}. We express the term as follows:

We will now fix one such term ω\omega and then bound the length of the following random vector:

Let α\alpha be used to denote a tuple (i,j)(i,j) of integers used to index entries in a matrix. Let T(i)T(i) be used to denote the parity function computed on ii, i.e, if ii is divisible by 22 and 11 otherwise. This function indicates if the matrix in the expansion is transposed or not. We now introduce B(i,j),(k,l)p,q, p∈{1,2}, q∈{0,1}{B_{(i,j),(k,l)}^{p,q},\,p\in\{1,2\},\,q\in\{0,1\}} and A(i,j)p, p∈{1,2}A_{(i,j)}^{p},\,p\in\{1,2\} which are defined as follows:

where δi,j=1\delta_{i,j}=1 if i=ji=j and 0 otherwise. We will subsequently write the random vector vωv_{\omega} in terms of the individual entries of the matrices. The role of B(i,j),(k,l)p,qB_{(i,j),(k,l)}^{p,q} and A(i,j)pA_{(i,j)}^{p} is to ensure consistency in the terms used to describe vωv_{\omega}. We will use hi,αh_{i,\alpha} to refer to (Hi)α(H_{i})_{\alpha}.

With this notation in hand, we are ready to describe vωv_{\omega}.

We now write the squared length of vωv_{\omega} as follows:

We can see from the above equations that the entries used to represent vωv_{\omega} are defined with respect to paths in a bipartite graph. In the following, we introduce notations to represent entire paths rather than just individual edges:

Let α≔(α1,…,α2a)\bm{\alpha}\coloneqq(\alpha_{1},\ldots,\alpha_{2a}) and

Calculating the kthk^{\text{th}} moment expansion of XωX_{\omega} for some number kk, we obtain:

We now show how to bound the above moment effectively. Notice that the moment is defined with respect to a collection of 2k2k paths. We denote this collection by Δ≔(α1,…,α2k)\Delta\coloneqq(\bm{\alpha^{1}},\ldots,\bm{\alpha^{2k}}). For each such collection, we define a partition Γ(Δ)\Gamma(\Delta) of the index set {(s,l):s∈[2k],l∈[2a]}\{(s,l):s\in[2k],l\in[2a]\} where (s,l)(s,l) and (s′,l′)(s^{\prime},l^{\prime}) are in the same equivalence class if ω(l)=ω(l′)=1\omega(l)=\omega(l^{\prime})=1 and αls=αl′s′\alpha^{s}_{l}=\alpha^{s^{\prime}}_{l^{\prime}}. Additionally, each (s,l)(s,l) such that ω(l)=2\omega(l)=2 is in a separate equivalence class.

We bound the expression in (28) by partitioning all possible collections of 2k2k paths based on the partitions defined by them in the above manner. We then proceed to bound the contribution of any one specific path to (28) following a particular partition Γ\Gamma, the number of paths satisfying that particular partition and finally, the total number of partitions. Since, H1H_{1} is a matrix with mean, any equivalence class containing an index (s,l)(s,l) such that ω(l)=1\omega(l)=1 contains at least two elements.

We proceed to bound (28) by taking absolute values:

We now fix one particular partition and bound the contribution to (29) of all collections of paths Δ\Delta that correspond to a valid partition Γ\Gamma.

We construct from Γ\Gamma a directed multigraph GG. The equivalence classes of Γ\Gamma form the vertex set of G, V(G)V(G). There are 4 kinds of edges in GG where each type is indexed by a tuple (p,q)(p,q) where p∈{1,2}, q∈{0,1}p\in\{1,2\},\,q\in\{0,1\}. We denote the edge sets corresponding to these 4 edge types by E(1,0)E_{(1,0)}, E(1,1)E_{(1,1)}, E(2,0)E_{(2,0)} and E(2,1)E_{(2,1)} respectively. An edge of type (p,q)(p,q) exists from equivalence class γ1\gamma_{1} to equivalence class γ2\gamma_{2} if there exists (s,l)∈γ1(s,l)\in\gamma_{1} and (s′,l′)∈γ2(s^{\prime},l^{\prime})\in\gamma_{2} such that l′=l+1l^{\prime}=l+1, s=s′s=s^{\prime}, ω(s′)=p\omega(s^{\prime})=p and T(l′)=qT(l^{\prime})=q.

The summation in 29 can be written as follows:

where (ζ1)(\zeta_{1}) follows from the moment conditions on H1H_{1}. V1(G)V_{1}(G) and V2(G)V_{2}(G) are the vertices in the graph corresponding to tuples (i,j)(i,j) such that ω(j)=1\omega(j)=1 and ω(j)=2\omega(j)=2 respectively and w1=∣V1(G)∣w_{1}=\left\lvert V_{1}(G)\right\rvert, w2=∣V2(G)∣w_{2}=\left\lvert V_{2}(G)\right\rvert.

We first consider an equivalence class γ1\gamma_{1} such that there exists an index (s,l)∈γ1(s,l)\in\gamma_{1} and l=1l=1. We form a spanning tree T1T_{1} of all the nodes reachable from γ1\gamma_{1} with γ1\gamma_{1} as root. We then remove the nodes V(T1)V(T_{1}) from the graph GG and repeat this procedure until we obtain a set of ll trees T1,…,TlT_{1},\dots,T_{l} with roots γ1,…,γl\gamma_{1},\dots,\gamma_{l} such that ⋃i=1lV(Gi)=V(G)\bigcup\limits_{i=1}^{l}V(G_{i})=V(G). This happens because every node is reachable from some equivalence class which contains an index of the form (s,1)(s,1). Also, each of these trees Ti, ∀ i∈[l]T_{i},\ \forall\,i\in[l] is disjoint in their vertex sets. Given this decomposition, we can factorize the above product as follows:

For a single connected component, we can compute the summation bottom up from the leaves. First, notice that:

Where the first two follow from the sparsity of H2H_{2}. Every node in the tree TjT_{j} with the exception of the root has a single incoming edge. For the root, γj\gamma_{j}, we have:

From the above two observations, we have:

where wk,jw_{k,j} represents the number of vertices in the jthj^{th} component which contain tuples (i,j)(i,j) such that ω(j)=k\omega(j)=k for k∈{1,2}k\in\{1,2\}.

Now, we bound the probability that X^ω\hat{X}_{\omega} is too large. Choosing k=⌈log⁡na1⌉k=\left\lceil\frac{\log n}{a_{1}}\right\rceil and applying the kthk^{th} moment Markov inequality, we obtain:

Taking a union bound over all the 2a2^{a} possible ω\omega, over values of aa from 11 to log⁡n\log n and over the nn values of qq, we get the required result. ∎

5 Additional Experimental Results

We detail some additional experiments performed with Algorithm 1 in this section. The experiments were performed on synthetic data and real world data sets.

Foreground-background separation. We present results for one more real world data set in this section. We applied our PG-RMC method (with varying pp) to the Escalator video. Figure 4 (a) shows one frame from the video. Figure 4 (b) shows the extracted background from the video by using our method (PG-RMC , Algorithm 1) with probability of sampling p=0.05p=0.05. Figure 4 (c) compares objective function value for different pp values.