Fast Algorithms for Robust PCA via Gradient Descent

Xinyang Yi, Dohyung Park, Yudong Chen, Constantine Caramanis

Introduction

While there have been recent results developing provably robust algorithms for PCA (e.g., ), the running times range from O(r2d2)\mathcal{O}(r^{2}d^{2}) to O(d3)\mathcal{O}(d^{3})For precise dependence on error and other factors, please see details below. and hence are significantly worse than SVD. Meanwhile, the literature developing sub-quadratic algorithms for PCA (e.g., ) seems unable to guarantee robustness to outliers or missing data.

Our contribution lies precisely in this area: provably robust algorithms for PCA with improved run-time. Specifically, we provide an efficient algorithm with running time that matches SVD while nearly matching the best-known robustness guarantees. In the case where rank is small compared to dimension, we develop an algorithm with running time that is nearly linear in the dimension. This last algorithm works by subsampling the data, and therefore we also show that our algorithm solves the Robust PCA problem with partial observations (a generalization of matrix completion and Robust PCA).

Provable solutions for this model are first provided in the works of and . They propose to solve this problem by convex relaxation:

where ∣ ⁣∣ ⁣∣M∣ ⁣∣ ⁣∣\mboxnuc|\!|\!|M|\!|\!|_{{\mbox{\tiny{nuc}}}} denotes the nuclear norm of MM. Despite analyzing the same method, the corruption models in and differ. In , the authors consider the setting where the entries of M∗M^{*} are corrupted at random with probability α\alpha. They show their method succeeds in exact recovery with α\alpha as large as 0.10.1, which indicates they can tolerate a constant fraction of corruptions. Work in considers a deterministic corruption model, where nonzero entries of S∗S^{*} can have arbitrary position, but the sparsity of each row and column does not exceed αd\alpha d. They prove that for exact recovery, it can allow α=O(1/(μrd))\alpha=\mathcal{O}(1/(\mu r\sqrt{d})). This was subsequently further improved to α=O(1/(μr))\alpha=\mathcal{O}(1/(\mu r)), which is in fact optimal . Here, μ\mu represents the incoherence of M∗M^{*} (see Section 2 for details). In this paper, we follow this latter line and focus on the deterministic corruption model.

The state-of-the-art solver for (1) has time complexity O(d3/ε)\mathcal{O}(d^{3}/\varepsilon) to achieve error ε\varepsilon, and is thus much slower than SVD, and prohibitive for even modest values of dd. Work in considers the deterministic corruption model, and improves this running time without sacrificing the robustness guarantee on α\alpha. They propose an alternating projection (AltProj) method to estimate the low rank and sparse structures iteratively and simultaneously, and show their algorithm has complexity O(r2d2log⁡(1/ε))\mathcal{O}(r^{2}d^{2}\log(1/\varepsilon)), which is faster than the convex approach but still slower than SVD.

After the initial post of our paper on arxiv, Cherapanamjeri et al. posted their paper on solving robust PCA with partial observations by a method modified from AltProj. Their approach is shown to have optimal robustness. The sample complexity they established is O(μ2r2dlog⁡2dlog⁡2(1/ε))\mathcal{O}(\mu^{2}r^{2}d\log^{2}d\log^{2}(1/\varepsilon)), which depends on the estimation error ε\varepsilon, since the analysis of AltProj under partial observations requires sampling splitting. In contrast, the sample complexity O(μ2r2dlog⁡d)\mathcal{O}(\mu^{2}r^{2}d\log d) of our approach (see Corollary 2) does not have such dependence. Moreover, their method requires computing rank-rr SVD of the observed matrices in every round of iterations. Our algorithm only needs to compute rank-rr SVD (approximately) once in the initialization step.

It is worth mentioning other works that obtain provable guarantees of non-convex algorithms or problems including phase retrieval , EM algorithms , tensor decompositions and second order method . It might be interesting to bring robust considerations to these works.

2 Our Contributions

In this paper, we develop efficient non-convex algorithms for robust PCA. We propose a novel algorithm based on the projected gradient method on the factorized space. We also extend it to solve robust PCA in the setting with partial observations, i.e., in addition to gross corruptions, the data matrix has a large number of missing values. Our main contributions are summarized as follows.To ease presentation, the discussion here assumes M∗M^{*} has constant condition number, whereas our results below show the dependence on condition number explicitly.

We propose a novel sparse estimator for the setting of deterministic corruptions. For the low-rank structure to be identifiable, it is natural to assume that deterministic corruptions are “spread out” (no more than some number in each row/column). We leverage this information in a simple but critical algorithmic idea, that is tied to the ultimate complexity advantages our algorithm delivers.

Based on the proposed sparse estimator, we propose a projected gradient method on the matrix factorized space. While non-convex, the algorithm is shown to enjoy linear convergence under proper initialization. Along with a new initialization method, we show that robust PCA can be solved within complexity O(rd2log⁡(1/ε))\mathcal{O}(rd^{2}\log(1/\varepsilon)) while ensuring robustness α=O(1/(μr1.5))\alpha=\mathcal{O}(1/(\mu r^{1.5})). Our algorithm is thus faster than the best previous known algorithm by a factor of rr, and enjoys superior empirical performance as well.

Algorithms for Robust PCA with partial observations still rely on a computationally expensive convex approach, as apparently this problem has evaded treatment by non-convex methods. We consider precisely this problem. In a nutshell, we show that our gradient method succeeds (it is guaranteed to produce the subspace of M∗M^{*}) even when run on no more than O(μ2r2dlog⁡d)\mathcal{O}(\mu^{2}r^{2}d\log d) random entries of YY. The computational cost is O(μ3r4dlog⁡dlog⁡(1/ε))\mathcal{O}(\mu^{3}r^{4}d\log d\log(1/\varepsilon)). When rank rr is small compared to the dimension dd, in fact this dramatically improves on our bound above, as our cost becomes nearly linear in dd. We show, moreover, that this savings and robustness to erasures comes at no cost in the robustness guarantee for the deterministic (gross) corruptions. While this demonstrates our algorithm is robust to both outliers and erasures, it also provides a way to reduce computational costs even in the fully observed setting, when rr is small.

An immediate corollary of the above result provides a guarantee for exact matrix completion, with general rectangular matrices, using O(μ2r2dlog⁡d)\mathcal{O}(\mu^{2}r^{2}d\log d) observed entries and O(μ3r4dlog⁡dlog⁡(1/ε))\mathcal{O}(\mu^{3}r^{4}d\log d\log(1/\varepsilon)) time, thereby improving on existing results in .

3 Organization and Notation

The remainder of this paper is organized as follows. In Section 2, we formally describe our problem and assumptions. In Section 3, we present and describe our algorithms for fully (Algorithm 1) and partially (Algorithm 2) observed settings. In Section 4.1, we establish theoretical guarantees of Algorithm 1. The theory for partially observed setting are presented in Section 4.2. The numerical results are collected in Section 5. Sections 6, 7 and Appendix A contain all the proofs and technical lemmas.

Problem Setup

(ii) The entries of S∗S^{*} are “spread out” – for α∈[0,1)\alpha\in[0,1), we assume S∗∈SαS^{*}\in\mathcal{S}_{\alpha}, where

In other words, S∗S^{*} contains at most α\alpha-fraction nonzero entries per row and column.

Algorithms

For both the full and partial observation settings, our method proceeds in two phases. In the first phase, we use a new sorting-based sparse estimator to produce a rough estimate SinitS_{\text{init}} for S∗S^{*} based on the observed matrix YY, and then find a rank rr matrix factorized as U0V0⊤U_{0}V_{0}^{\top} that is a rough estimate of M∗M^{*} by performing SVD on (Y−SinitY-S_{\text{init}}). In the second phase, given (U0,V0)(U_{0},V_{0}), we perform an iterative method to produce series {(Ut,Vt)}t=0∞\{(U_{t},V_{t})\}_{t=0}^{\infty}. In each step tt, we first apply our sparse estimator to produce a sparse matrix StS_{t} based on (Ut,Vt)(U_{t},V_{t}), and then perform a projected gradient descent step on the low-rank factorized space to produce (Ut+1,Vt+1)(U_{t+1},V_{t+1}). This flow is the same for full and partial observations, though a few details differ. Algorithm 1 gives the full observation algorithm, and Algorithm 2 gives the partial observation algorithm. We now describe the key details of each algorithm.

where A(i,⋅)(k)A_{(i,\cdot)}^{(k)} and A(⋅,j)(k)A_{(\cdot,j)}^{(k)} denote the elements of A(i,⋅)A_{(i,\cdot)} and A(⋅,j)A_{(\cdot,j)} that have the kk-th largest magnitude respectively. In other words, we choose to keep those elements that are simultaneously among the largest α\alpha-fraction entries in the corresponding row and column. In the case of entries having identical magnitude, we break ties arbitrarily. It is thus guaranteed that Tα[A]∈Sα\mathcal{T}_{\alpha}\left[A\right]\in\mathcal{S}_{\alpha}.

Initialization.

In the fully observed setting, we compute SinitS_{\text{init}} based on YY as Sinit=Tα[Y]S_{\text{init}}=\mathcal{T}_{\alpha}\left[Y\right]. In the partially observed setting with sampling rate pp, we let Sinit=T2pα[Y]S_{\text{init}}=\mathcal{T}_{2p\alpha}\left[Y\right]. In both cases, we then set U0=LΣ1/2U_{0}=L\Sigma^{1/2} and V0=RΣ1/2V_{0}=R\Sigma^{1/2}, where LΣR⊤L\Sigma R^{\top} is an SVD of the best rank rr approximation of Y−SinitY-S_{\text{init}}.

Gradient Method on Factorized Space.

After initialization, we proceed by projected gradient descent. To do this, we define loss functions explicitly in the factored space, i.e., in terms of U,VU,V and SS:

Recall that our goal is to recover M∗M^{*} that satisfies the μ\mu-incoherent condition. Given an SVD M∗=L∗ΣR∗⊤M^{*}=L^{*}\Sigma R^{*\top}, we expect that the solution (U,V)(U,V) is close to (L∗Σ1/2,R∗Σ1/2)(L^{*}\Sigma^{1/2},R^{*}\Sigma^{1/2}) up to some rotation. In order to serve such μ\mu-incoherent structure, it is natural to put constraints on the row norms of U,VU,V based on ∣ ⁣∣ ⁣∣M∗∣ ⁣∣ ⁣∣\mboxop|\!|\!|M^{*}|\!|\!|_{{\mbox{\tiny{op}}}}. As ∣ ⁣∣ ⁣∣M∗∣ ⁣∣ ⁣∣\mboxop|\!|\!|M^{*}|\!|\!|_{{\mbox{\tiny{op}}}} is unavailable, given U0,V0U_{0},V_{0} computed in the first phase, we rely on the sets U\mathcal{U}, V\mathcal{V} defined as

Now we consider the following optimization problems with constraints:

The regularization term in the objectives above is used to encourage that UU and VV have the same scale. Given (U0,V0)(U_{0},V_{0}), we propose the following iterative method to produce series {(Ut,Vt)}t=0∞\{(U_{t},V_{t})\}_{t=0}^{\infty} and {St}t=0∞\{S_{t}\}_{t=0}^{\infty}. We give the details for the fully observed case – the partially observed case is similar. For t=0,1,…t=0,1,\ldots, we update StS_{t} using the sparse estimator St=Tγα[Y−UtVt⊤]S_{t}=\mathcal{T}_{\gamma\alpha}\left[Y-U_{t}V_{t}^{\top}\right], followed by a projected gradient update on UtU_{t} and VtV_{t}

Here α\alpha is the model parameter that characterizes the corruption fraction, γ\gamma and η\eta are algorithmic tunning parameters, which we specify in our analysis. Essentially, the above algorithm corresponds to applying projected gradient method to optimize (8), where SS is replaced by the aforementioned sparse estimator in each step.

Main Results

In this section, we establish theoretical guarantees for Algorithm 1 in the fully observed setting and for Algorithm 2 in the partially observed setting.

We begin with some definitions and notation. It is important to define a proper error metric because the optimal solution corresponds to a manifold and there are many distinguished pairs (U,V)(U,V) that minimize (8). Given the SVD of the true low-rank matrix M∗=L∗Σ∗R∗⊤M^{*}=L^{*}\Sigma^{*}R^{*\top}, we let U∗:=L∗Σ∗1/2U^{*}:=L^{*}\Sigma^{*1/2} and V∗:=R∗Σ∗1/2V^{*}:=R^{*}\Sigma^{*1/2}. We also let σ1∗≥σ2∗≥…≥σr∗\sigma_{1}^{*}\geq\sigma_{2}^{*}\geq\ldots\geq\sigma_{r}^{*} be sorted nonzero singular values of M∗M^{*}, and denote the condition number of M∗M^{*} by κ\kappa, i.e., κ:=σ1∗/σr∗\kappa:=\sigma_{1}^{*}/\sigma_{r}^{*}. We define estimation error d(U,V;U∗,V∗)d(U,V;U^{*},V^{*}) as the minimal Frobenius norm between (U,V)(U,V) and (U∗,V∗)(U^{*},V^{*}) with respect to the optimal rotation, namely

when d(U,V;U∗,V∗)≤σ1∗d(U,V;U^{*},V^{*})\leq\sqrt{\sigma_{1}^{*}}. We denote the local region around the optimum (U∗,V∗)(U^{*},V^{*}) with radius ω\omega as

The next two theorems provide guarantees for the initialization phase and gradient iterations, respectively, of Algorithm 1. The proofs are given in Sections 6.1 and 6.2.

Consider the paired (U0,V0)(U_{0},V_{0}) produced in the first phase of Algorithm 1. If α≤1/(16κμr)\alpha\leq 1/(16\kappa\mu r), we have

Therefore, using proper initialization and step size, the gradient iteration converges at a linear rate with a constant contraction factor 1−O(1/κ)1-\mathcal{O}(1/\kappa). To obtain relative precision ε\varepsilon compared to the initial error, it suffices to perform O(κlog⁡(1/ε))O(\kappa\log(1/\varepsilon)) iterations. Note that the step size is chosen according to 1/σ1∗1/\sigma_{1}^{*}. When α≲1/(μκr3)\alpha\lesssim 1/(\mu\sqrt{\kappa r^{3}}), Theorem 1 and the inequality (11) together imply that ∣ ⁣∣ ⁣∣U0V0⊤−M∗∣ ⁣∣ ⁣∣\mboxop≤12σ1∗|\!|\!|U_{0}V_{0}^{\top}-M^{*}|\!|\!|_{{\mbox{\tiny{op}}}}\leq\frac{1}{2}\sigma_{1}^{*}. Hence we can set the step size as η=O(1/σ1(U0V0⊤))\eta=\mathcal{O}(1/\sigma_{1}(U_{0}V_{0}^{\top})) using being the top singular value σ1(U0V0⊤)\sigma_{1}(U_{0}V_{0}^{\top}) of the matrix U0V0⊤U_{0}V_{0}^{\top}

Combining Theorems 1 and 2 implies the following result, proved in Section 6.3, that provides an overall guarantee for Algorithm 1.

for some constant cc. Then for any ε∈(0,1)\varepsilon\in(0,1), Algorithm 1 with T=O(κlog⁡(1/ε))T=\mathcal{O}(\kappa\log(1/\varepsilon)) outputs a pair (UT,VT)(U_{T},V_{T}) that satisfies

For simplicity we assume d1=d2=dd_{1}=d_{2}=d. Our sparse estimator (4) can be implemented by finding the top αd\alpha d elements of each row and column via partial quick sort, which has running time O(d2log⁡(αd))\mathcal{O}({d^{2}\log(\alpha d)}). Performing rank-rr SVD in the first phase and computing the gradient in each iteration both have complexity O(rd2)\mathcal{O}(rd^{2}).In fact, it suffices to compute the best rank-rr approximation with running time independent of the eigen gap. Algorithm 1 thus has total running time O(κrd2log⁡(1/ε))\mathcal{O}(\kappa rd^{2}\log(1/\varepsilon)) for achieving an ϵ\epsilon accuracy as in (12). We note that when κ=O(1)\kappa=\mathcal{O}(1), our algorithm is orderwise faster than the AltProj algorithm in , which has running time O(r2d2log⁡(1/ε))\mathcal{O}(r^{2}d^{2}\log(1/\varepsilon)). Moreover, our algorithm only requires computing one singular value decomposition.

Assuming κ=O(1)\kappa=\mathcal{O}(1), our algorithm can tolerate corruption at a sparsity level up to α=O(1/(μrr))\alpha=\mathcal{O}(1/(\mu r\sqrt{r})). This is worse by a factor r\sqrt{r} compared to the optimal statistical guarantee 1/(μr)1/(\mu r) obtained in . This looseness is a consequence of the condition for (U0,V0)(U_{0},V_{0}) in Theorem 2. Nevertheless, when μr=O(1)\mu r=\mathcal{O}(1), our algorithm can tolerate a constant α\alpha fraction of corruptions.

Notably, we show that gradient descent works in the case of α=O(1/(μr))\alpha=\mathcal{O}(1/(\mu r)) if initialization is sufficiently close. Accordingly, to provide an algorithm with optimal robustness, it is straightforward to use a more complicated initial method such as AltProj that can tolerate 1/(μr)1/(\mu r) fraction of corruptions while satisfying our initial condition. As gradient descent provides a simple and efficient way to successively refine the estimation, such combination still gives a better running time than prior arts.

2 Analysis of Algorithm 2

We now move to the guarantees of Algorithm 2. We show here that not only can we handle partial observations, but in fact subsampling the data in the fully observed case can significantly reduce the time complexity from the guarantees given in the previous section without sacrificing robustness. In particular, for smaller values of rr, the complexity of Algorithm 2 has near linear dependence on the dimension dd, instead of quadratic.

In the following discussion, we let d:=max⁡{d1,d2}d:=\max\{d_{1},d_{2}\}. The next two results, proved in Sections 6.4 and 6.5, control the quality of the initialization step, and then the gradient iterations.

Suppose the observed indices Φ\Phi follow the Bernoulli model given in (2). Consider the pair (U0,V0)(U_{0},V_{0}) produced in the first phase of Algorithm 2. There exist constants {ci}i=13\{c_{i}\}_{i=1}^{3} such that for any ϵ∈(0,r/(8c1κ))\epsilon\in(0,\sqrt{r}/(8c_{1}\kappa)), if

with probability at least 1−c3d−11-c_{3}d^{-1}.

Suppose the observed indices Φ\Phi follow the Bernoulli model given in (2). Consider the second phase of Algorithm 2. Suppose we choose γ=3\gamma=3, and η=c/(μrσ1∗)\eta=c/(\mu r\sigma_{1}^{*}) for a sufficiently small constant cc. There exist constants {ci}i=14\{c_{i}\}_{i=1}^{4} such that if

then with probability at least 1−c3d−11-c_{3}d^{-1}, the iterates {(Ut,Vt)}t=0∞\{(U_{t},V_{t})\}_{t=0}^{\infty} satisfy

The above result ensures linear convergence to (U∗,V∗)(U^{*},V^{*}) (up to rotation) even when the gradient iterations are computed using partial observations. Note that setting p=1p=1 recovers Theorem 2 up to an additional factor μr\mu r in the contraction factor. For achieving ε\varepsilon relative accuracy, now we need O(μrκlog⁡(1/ε))\mathcal{O}(\mu r\kappa\log(1/\varepsilon)) iterations.

Putting Theorems 3 and 4 together, we have the following overall guarantee, proved in Section 6.6, for Algorithm 2.

for some constants c,c′c,c^{\prime}. With probability at least 1−O(d−1)1-\mathcal{O}(d^{-1}), for any ε∈(0,1)\varepsilon\in(0,1), Algorithm 2 with T=O(μrκlog⁡(1/ε))T=\mathcal{O}(\mu r\kappa\log(1/\varepsilon)) outputs a pair (UT,VT)(U_{T},V_{T}) that satisfies

This result shows that partial observations do not compromise robustness to sparse corruptions: as long as the observation probability pp satisfies the condition in Corollary 2, Algorithm 2 enjoys the same robustness guarantees as the method using all entries. Below we provide two remarks on the sample and time complexity. For simplicity, we assume d1=d2=dd_{1}=d_{2}=d, κ=O(1)\kappa=\mathcal{O}(1).

Using the lower bound on pp, it is sufficient to have O(μ2r2dlog⁡d)\mathcal{O}(\mu^{2}r^{2}d\log d) observed entries. In the special case S∗=0S^{*}=0, our partial observation model is equivalent to the model of exact matrix completion (see, e.g., ). We note that our sample complexity (i.e., observations needed) matches that of completing a positive semidefinite (PSD) matrix by gradient descent as shown in , and is better than the non-convex matrix completion algorithms in and . Accordingly, our result reveals the important fact that we can obtain robustness in matrix completion without deterioration of our statistical guarantees. It is known that that any algorithm for solving exact matrix completion must have sample size Ω(μrdlog⁡d)\Omega(\mu rd\log d) , and a nearly tight upper bound O(μrdlog⁡2d)O(\mu rd\log^{2}d) is obtained in by convex relaxation. While sub-optimal by a factor μr\mu r, our algorithm is much faster than convex relaxation as shown below.

Our sparse estimator on the sparse matrix with support Φ\Phi can be implemented via partial quick sort with running time O(pd2log⁡(αpd))\mathcal{O}({pd^{2}\log(\alpha pd)}). Computing the gradient in each step involves the two terms in the objective function (9). Computing the gradient of the first term L~\widetilde{\mathcal{L}} takes time O(r∣Φ∣)\mathcal{O}(r|\Phi|), whereas the second term takes time O(r2d)\mathcal{O}(r^{2}d). In the initialization phase, performing rank-rr SVD on a sparse matrix with support Φ\Phi can be done in time O(r∣Φ∣)\mathcal{O}(r|\Phi|). We conclude that when ∣Φ∣=O(μ2r2dlog⁡d)|\Phi|=\mathcal{O}(\mu^{2}r^{2}d\log d), Algorithm 2 achieves the error bound (15) with running time O(μ3r4dlog⁡dlog⁡(1/ε))\mathcal{O}(\mu^{3}r^{4}d\log d\log(1/\varepsilon)). Therefore, in the small rank setting with r≪d1/3r\ll d^{1/3}, even when full observations are given, it is better to use Algorithm 2 by subsampling the entries of YY.

Numerical Results

In this section, we provide numerical results and compare the proposed algorithms with existing methods, including the inexact augmented lagrange multiplier (IALM) approach for solving the convex relaxation (1) and the alternating projection (AltProj) algorithm proposed in . All algorithms are implemented in MATLAB Our code is available at https://www.yixinyang.org/code/RPCA_GD.zip., and the codes for existing algorithms are obtained from their authors. SVD computation in all algorithms uses the PROPACK library.http://sun.stanford.edu/~rmunk/PROPACK/ We ran all simulations on a machine with Intel 32-core Xeon (E5-2699) 2.3GHz with 240GB RAM.

The results are summarized in Figure 1. Figure 1(a) shows the convergence of our algorithms for different random instances with different sub-sampling rate pp (note that p=1p=1 corresponds to the fully observed setting). As predicted by Theorems 2 and 4, our gradient method converges geometrically with a contraction factor nearly independent of pp. Figure 1(b) shows the running time of our algorithm with partially observed data. We see that the running time scales linearly with dd, again consistent with the theory. We note that our algorithm is memory-efficient: in the large scale setting with d=2×105d=2\times 10^{5}, using approximately 0.1%0.1\% entries is sufficient for the successful recovery. In contrast, AltProj and IALM are designed to manipulate the entire matrix with d2=4×1010d^{2}=4\times 10^{10} entries, which is prohibitive on a single machine. Figure 1(c) compares our algorithms with AltProj and IALM by showing reconstruction error versus real running time. Our algorithm requires significantly less computation to achieve the same accuracy level, and using only a subset of the entries provides additional speed-up.

2 Foreground-background Separation

We apply our method to the task of foreground-background (FB) separation in a video. We use two public benchmarks, the Restaurant and ShoppingMall datasets.http://perception.i2r.a-star.edu.sg/bk_model/bk_index.html Each dataset contains a video with static background. By vectorizing and stacking the frames as columns of a matrix YY, the FB separation problem can be cast as RPCA, where the static background corresponds to a low rank matrix M∗M^{*} with identical columns, and the moving objects in the video can be modeled as sparse corruptions S∗S^{*}. Figure 2 shows the output of different algorithms on two frames from the dataset. Our algorithms require significantly less running time than both AltProj and IALM. Moreover, even with 20% sub-sampling, our methods still appear to achieve better separation quality (note that in each of the frames our algorithms remove a person that is not identified by the other algorithms).

Figure 3 shows recovery results for several more frames. Again, our algorithms enjoy better running time and outperform AltProj and IALM in separating persons from the background images. In Appendix B, we describe the detailed parameter settings for our algorithm.

Proofs

In this section we provide the proofs for our main theoretical results in Theorems 1–4 and Corollaries 1–2.

Let \makebox[0.0pt][l]Y:=Y−Sinit\makebox[0.0pt][l]{\hskip 2.05pt\rule[8.12498pt]{5.67886pt}{0.43057pt}}{Y}:=Y-S_{\text{init}}. As Y=M∗+S∗Y=M^{*}+S^{*}, we have \makebox[0.0pt][l]Y−M∗=S∗−Sinit\makebox[0.0pt][l]{\hskip 2.05pt\rule[8.12498pt]{5.67886pt}{0.43057pt}}{Y}-M^{*}=S^{*}-S_{\text{init}}. We obtain \makebox[0.0pt][l]Y−M∗∈S2α\makebox[0.0pt][l]{\hskip 2.05pt\rule[8.12498pt]{5.67886pt}{0.43057pt}}{Y}-M^{*}\in\mathcal{S}_{2\alpha} because S∗,Sinit∈SαS^{*},S_{\text{init}}\in\mathcal{S}_{\alpha}.

We claim that ∥\makebox[0.0pt][l]Y−M∗∥∞≤2∥M∗∥∞\|\makebox[0.0pt][l]{\hskip 2.05pt\rule[8.12498pt]{5.67886pt}{0.43057pt}}{Y}-M^{*}\|_{\infty}\leq 2\|M^{*}\|_{\infty}. Denote the support of S∗,SinitS^{*},S_{\text{init}} by Ω∗\Omega^{*} and Ω\Omega respectively. Since \makebox[0.0pt][l]Y−M∗\makebox[0.0pt][l]{\hskip 2.05pt\rule[8.12498pt]{5.67886pt}{0.43057pt}}{Y}-M^{*} is supported on Ω∪Ω∗\Omega\cup\Omega^{*}, to prove the claim it suffices to consider the following three cases.

For (i,j)∈Ω∗∩Ω(i,j)\in\Omega^{*}\cap\Omega, due to rule of sparse estimation, we have =0=0.

For (i,j)∈Ω∗∖Ω(i,j)\in\Omega^{*}\setminus\Omega, we must have ∣∣≤2∥M∗∥∞||\leq 2\|M^{*}\|_{\infty}. Otherwise, we have ∣∣=∣∣>∥M∗∥∞||=||>\|M^{*}\|_{\infty}. So ∣∣|| is larger than any uncorrupted entries in its row and column. Since there are at most α\alpha fraction corruptions per row and column, we have ∈Ω\in\Omega, which violates the prior condition (i,j)∈Ω∗∖Ω(i,j)\in\Omega^{*}\setminus\Omega.

For the last case (i,j)∈Ω∖Ω∗(i,j)\in\Omega\setminus\Omega^{*}, since ==, trivially we have ∣∣≤∥M∗∥∞||\leq\|M^{*}\|_{\infty}.

The following result, proved in Section 7.1, relates the operator norm of \makebox[0.0pt][l]Y−M∗\makebox[0.0pt][l]{\hskip 2.05pt\rule[8.12498pt]{5.67886pt}{0.43057pt}}{Y}-M^{*} to its infinite norm.

In the last step, we use the fact that M∗M^{*} satisfies the μ\mu-incoherent condition, which leads to

We denote the ii-th largest singular value of \makebox[0.0pt][l]Y\makebox[0.0pt][l]{\hskip 2.05pt\rule[8.12498pt]{5.67886pt}{0.43057pt}}{Y} by σi\sigma_{i}. By Weyl’s theorem, we have ∣σi∗−σi∣≤∣ ⁣∣ ⁣∣\makebox[0.0pt][l]Y−M∗∣ ⁣∣ ⁣∣\mboxop|\sigma_{i}^{*}-\sigma_{i}|\leq|\!|\!|\makebox[0.0pt][l]{\hskip 2.05pt\rule[8.12498pt]{5.67886pt}{0.43057pt}}{Y}-M^{*}|\!|\!|_{{\mbox{\tiny{op}}}} for all i∈[d1∧d2]i\in[d_{1}\wedge d_{2}]. Since σr+1∗=0\sigma_{r+1}^{*}=0, we have σr+1≤∣ ⁣∣ ⁣∣\makebox[0.0pt][l]Y−M∗∣ ⁣∣ ⁣∣\mboxop\sigma_{r+1}\leq|\!|\!|\makebox[0.0pt][l]{\hskip 2.05pt\rule[8.12498pt]{5.67886pt}{0.43057pt}}{Y}-M^{*}|\!|\!|_{{\mbox{\tiny{op}}}}. Recall that U0V0⊤U_{0}V_{0}^{\top} is the best rank rr approximation of \makebox[0.0pt][l]Y\makebox[0.0pt][l]{\hskip 2.05pt\rule[8.12498pt]{5.67886pt}{0.43057pt}}{Y}. Accordingly, we have

Under condition αμr≤116κ\alpha\mu r\leq\frac{1}{16\kappa}, we obtain ∣ ⁣∣ ⁣∣U0V0⊤−M∗∣ ⁣∣ ⁣∣\mboxop≤12σr∗|\!|\!|U_{0}V_{0}^{\top}-M^{*}|\!|\!|_{{\mbox{\tiny{op}}}}\leq\frac{1}{2}\sigma_{r}^{*}. Applying Lemma 5.14 in (we provide it as Lemma 15 for the sake of completeness), we obtain

Plugging the upper bound of ∣ ⁣∣ ⁣∣U0V0⊤−M∗∣ ⁣∣ ⁣∣\mboxop|\!|\!|U_{0}V_{0}^{\top}-M^{*}|\!|\!|_{{\mbox{\tiny{op}}}} into the above inequality completes the proof.

2 Proof of Theorem 2

We essentially follow the general framework developed in for analyzing the behaviors of gradient descent in factorized low-rank optimization. But it is worth to note that only studies the symmetric and positive semidefinite setting, while we avoid such constraint on M∗M^{*}. The techniques for analyzing general asymmetric matrix in factorized space is inspired by the recent work on solving low-rank matrix equations. In our setting, the technical challenge is to verify the local descent condition of the loss function (8), which not only has a bilinear dependence on UU and VV, but also involves our sparse estimator (4).

We begin with some notations. Define the equivalent set of optimal solution as

As a result, for U,V\mathcal{U},\mathcal{V} constructed according to (7), we have

For L(U,V;S)\mathcal{L}(U,V;S), we denote the gradient with respect to MM by ∇ML(U,V;S)\nabla_{M}\mathcal{L}(U,V;S), i.e. ∇ML(U,V;S)=UV⊤+S−Y\nabla_{M}\mathcal{L}(U,V;S)=UV^{\top}+S-Y.

The local descent property is implied by combining the following two results, which are proved in Section 6.7 and 6.8 respectively.

Here ΔU:=U−Uπ∗\Delta_{U}:=U-U_{\pi^{*}}, ΔV:=V−Vπ∗\Delta_{V}:=V-V_{\pi^{*}}, δ:=∣ ⁣∣ ⁣∣ΔU∣ ⁣∣ ⁣∣\mboxF2+∣ ⁣∣ ⁣∣ΔV∣ ⁣∣ ⁣∣\mboxF2\delta:=|\!|\!|\Delta_{U}|\!|\!|_{{\mbox{\tiny{F}}}}^{2}+|\!|\!|\Delta_{V}|\!|\!|_{{\mbox{\tiny{F}}}}^{2}, and ν:=9(β+6)αμr+5β−1\nu:=9(\beta+6)\alpha\mu r+5\beta^{-1}.

where δ\delta is defined according to Lemma 2.

As another key ingredient, we establish the following smoothness condition, proved in Section 6.9, which indicates that the Frobenius norm of gradient decreases as (U,V)(U,V) approaches the optimal manifold.

With the above results in hand, we are ready to prove Theorem 2.

For (Ut,Vt)(U_{t},V_{t}), let (Uπ∗t,Vπ∗t):=arg⁡ ⁣min⁡(A,B)∈E(M∗)∣ ⁣∣ ⁣∣Ut−A∣ ⁣∣ ⁣∣\mboxF2+∣ ⁣∣ ⁣∣Vt−B∣ ⁣∣ ⁣∣\mboxF2(U_{\pi^{*}}^{t},V_{\pi^{*}}^{t}):=\arg\!\min_{(A,B)\in\mathcal{E}(M^{*})}|\!|\!|U_{t}-A|\!|\!|_{{\mbox{\tiny{F}}}}^{2}+|\!|\!|V_{t}-B|\!|\!|_{{\mbox{\tiny{F}}}}^{2}. Define ΔUt:=Ut−Uπ∗t\Delta_{U}^{t}:=U_{t}-U_{\pi^{*}}^{t}, ΔVt:=Vt−Vπ∗t\Delta_{V}^{t}:=V_{t}-V_{\pi^{*}}^{t}.

where the second step follows from the non-expansion property of projection onto U,V\mathcal{U},\mathcal{V}, which is implied by E(M∗)⊆U×V\mathcal{E}(M^{*})\subseteq\mathcal{U}\times\mathcal{V} shown in (19). Since ∇ULt=[∇MLt]V\nabla_{U}\mathcal{L}_{t}=\left[\nabla_{M}\mathcal{L}_{t}\right]V and ∇VLt=[∇MLt]⊤U\nabla_{V}\mathcal{L}_{t}=\left[\nabla_{M}\mathcal{L}_{t}\right]^{\top}U, we have

Combining Lemma 2 and 3, under condition δt<σr∗\delta_{t}<\sigma_{r}^{*}, we have that

By the assumption η=c/σ1∗\eta=c/\sigma_{1}^{*} for any constant c≤1/36c\leq 1/36, we thus have

In Lemma 2, choosing β=320κ\beta=320\kappa and assuming α≲1/(κ2μr)\alpha\lesssim 1/(\kappa^{2}\mu r), we can have ν≤1/(32κ)\nu\leq 1/(32\kappa). Assuming δt≲σr∗/κ\delta_{t}\lesssim\sigma_{r}^{*}/\kappa leads to 14σ1∗δt3≤116σr∗δt14\sqrt{\sigma_{1}^{*}\delta_{t}^{3}}\leq\frac{1}{16}\sigma_{r}^{*}\delta_{t}. We thus obtain

Under initial condition δ0≲σr∗/κ\delta_{0}\lesssim\sigma_{r}^{*}/\kappa, we obtain that such condition holds for all tt since estimation error decays geometrically after each iteration. Then applying (25) for all iterations, we conclude that for all t=0,1,…t=0,1,\ldots,

3 Proof of Corollary 1

We need α≲1κ2μr\alpha\lesssim\frac{1}{\kappa^{2}\mu r} due to the condition of Theorem 2. In order to ensure the linear convergence happens, it suffices to let the initial error shown in Theorem 1 be less than the corresponding condition in Theorem 2. Accordingly, we need

which leads to α≲1μrκ3\alpha\lesssim\frac{1}{\mu\sqrt{r\kappa}^{3}}.

Using the conclusion that gradient descent has linear convergence, choosing T=O(κlog⁡(1/ε))T=\mathcal{O}(\kappa\log(1/\varepsilon)), we have

Finally, applying the relationship between d(UT,VT;U∗,V∗)d(U_{T},V_{T};U^{*},V^{*}) and ∣ ⁣∣ ⁣∣UTVT⊤−M∗∣ ⁣∣ ⁣∣\mboxF|\!|\!|U_{T}V_{T}^{\top}-M^{*}|\!|\!|_{{\mbox{\tiny{F}}}} shown in (11), we complete the proof.

4 Proof of Theorem 3

Let \makebox[0.0pt][l]Y:=1p(Y−Sinit)\makebox[0.0pt][l]{\hskip 2.05pt\rule[8.12498pt]{5.67886pt}{0.43057pt}}{Y}:=\frac{1}{p}(Y-S_{\text{init}}). Similar to the proof of Theorem 1, we first establish an upper bound on ∣ ⁣∣ ⁣∣\makebox[0.0pt][l]Y−M∗∣ ⁣∣ ⁣∣\mboxop|\!|\!|\makebox[0.0pt][l]{\hskip 2.05pt\rule[8.12498pt]{5.67886pt}{0.43057pt}}{Y}-M^{*}|\!|\!|_{{\mbox{\tiny{op}}}}. We have that

For the first term, we have \makebox[0.0pt][l]Y−1pΠΦM∗=1p(ΠΦ(S∗)−Sinit)\makebox[0.0pt][l]{\hskip 2.05pt\rule[8.12498pt]{5.67886pt}{0.43057pt}}{Y}-\frac{1}{p}\Pi_{\Phi}M^{*}=\frac{1}{p}(\Pi_{\Phi}(S^{*})-S_{\text{init}}) because Y=ΠΦ(M∗+S∗)Y=\Pi_{\Phi}(M^{*}+S^{*}). Lemma 10 shows that under condition p≳log⁡dα(d1∧d2)p\gtrsim\frac{\log d}{\alpha(d_{1}\wedge d_{2})}, there are at most 32pα\frac{3}{2}p\alpha-fraction nonzero entries in each row and column of ΠΦ(S∗)\Pi_{\Phi}(S^{*}) with high probability. Since Sinit∈S2pαS_{\text{init}}\in\mathcal{S}_{2p\alpha}, we have

Denote the support of ΠΦ(S∗)\Pi_{\Phi}(S^{*}) and SinitS_{\text{init}} by Ωo∗\Omega_{o}^{*} and Ω\Omega. For (i,j)∈Ωo∗∩Ω(i,j)\in\Omega_{o}^{*}\cap\Omega and (i,j)∈Ω∖Ωo∗(i,j)\in\Omega\setminus\Omega_{o}^{*}, we have =0=0 and =−=-, respectively. To prove the claim, it remains to show that for (i,j)∈Ωo∗∖Ω(i,j)\in\Omega_{o}^{*}\setminus\Omega, ∣∣<2∥M∗∥∞||<2\|M^{*}\|_{\infty}. If this is not true, then we must have ∣∣>∥M∗∥∞||>\|M^{*}\|_{\infty}. Accordingly, ∣∣|| is larger than the magnitude of any uncorrupted entries in its row and column. Note that on the support Φ\Phi, there are at most 32pα\frac{3}{2}p\alpha corruptions per row and column, we have (i,j)∈Ω(i,j)\in\Omega, which violates our prior condition (i,j)∈Ωo∗∖Ω(i,j)\in\Omega_{o}^{*}\setminus\Omega.

Using these two properties (27), (28) and applying Lemma 1, we have

For the second term in (26), we use the following lemma proved in .

Given the SVD M∗=L∗ΣR∗⊤M^{*}=L^{*}\Sigma R^{*\top}, for any i∈[d1]i\in[d_{1}], we have

We can bound ∣ ⁣∣ ⁣∣M∗⊤∣ ⁣∣ ⁣∣2,∞|\!|\!|M^{*\top}|\!|\!|_{{2,\infty}} similarly. Lemma 5 leads to

under condition p≥4μr2log⁡dϵ2(d1∧d2)p\geq\frac{4\mu r^{2}\log d}{\epsilon^{2}(d_{1}\wedge d_{2})}.

Putting (29) and (30) together, we obtain

Then using the fact that U0V0⊤U_{0}V_{0}^{\top} is the best rank rr approximation of \makebox[0.0pt][l]Y\makebox[0.0pt][l]{\hskip 2.05pt\rule[8.12498pt]{5.67886pt}{0.43057pt}}{Y} and applying Wely’s theorem (see the proof of Theorem 1 for a detailed argument), we have

Under our assumptions, we have 16αμrσ1∗+2c′ϵσ1∗/r≤12σr∗16\alpha\mu r\sigma_{1}^{*}+2c^{\prime}\epsilon\sigma_{1}^{*}/\sqrt{r}\leq\frac{1}{2}\sigma_{r}^{*}. Accordingly, Lemma 15 gives

We complete the proof by combining the above two inequalities.

5 Proof of Theorem 4

In this section, we turn to prove Theorem 4. Similar to the proof of Theorem 2, we rely on establishing the local descent and smoothness conditions. Compared to the full observation setting, we replace L\mathcal{L} by L~\widetilde{\mathcal{L}} given in (6), while the regularization term G~(U,V):=164∣ ⁣∣ ⁣∣U⊤U−V⊤V∣ ⁣∣ ⁣∣\mboxF2\widetilde{\mathcal{G}}(U,V):=\frac{1}{64}|\!|\!|U^{\top}U-V^{\top}V|\!|\!|_{{\mbox{\tiny{F}}}}^{2} merely differs from G(U,V)\mathcal{G}(U,V) given in (20) by a constant factor. It is thus sufficient to analyze the properties of L~\widetilde{\mathcal{L}}.

Define E(M∗)\mathcal{E}(M^{*}) according to (18). Under the initial condition, we still have

We prove the next two lemmas in Section 6.10 and 6.11 respectively. In both lemmas, for any (U,V)∈U×V(U,V)\in\mathcal{U}\times\mathcal{V}, we use shorthands

ΔU:=U−Uπ∗\Delta_{U}:=U-U_{\pi^{*}}, ΔV:=V−Vπ∗\Delta_{V}:=V-V_{\pi^{*}}, and δ:=∣ ⁣∣ ⁣∣ΔU∣ ⁣∣ ⁣∣\mboxF2+∣ ⁣∣ ⁣∣ΔV∣ ⁣∣ ⁣∣\mboxF2\delta:=|\!|\!|\Delta_{U}|\!|\!|_{{\mbox{\tiny{F}}}}^{2}+|\!|\!|\Delta_{V}|\!|\!|_{{\mbox{\tiny{F}}}}^{2}. Recall that d:=max⁡{d1,d2}d:=\max\{d_{1},d_{2}\}.

Suppose U,V\mathcal{U},\mathcal{V} satisfy (31). Suppose we let

where we choose γ=3\gamma=3. For any β>0\beta>0 and ϵ∈(0,14)\epsilon\in(0,\frac{1}{4}), we define ν:=(14β+81)αμr+26ϵ+18β−1\nu:=(14\beta+81)\alpha\mu r+26\sqrt{\epsilon}+18\beta^{-1}. There exist constants {ci}i=12\{c_{i}\}_{i=1}^{2} such that if

then with probability at least 1−c2d−11-c_{2}d^{-1},

In the remainder of this section, we condition on the events in Lemma 6 and 7. Now we are ready to prove Theorem 4.

We essentially follow the process for proving Theorem 2. Let the following shorthands be defined in the same fashion: δt\delta_{t}, (Uπ∗t,Vπ∗t)(U_{\pi^{*}}^{t},V_{\pi^{*}}^{t}), (ΔUt,ΔVt)(\Delta_{U}^{t},\Delta_{V}^{t}), L~t\widetilde{\mathcal{L}}_{t}, G~t\widetilde{\mathcal{G}}_{t}.

Here we show error decays in one step of iteration. The induction process is the same as the proof of Theorem 2, and is thus omitted. For any t≥0t\geq 0, similar to (24) we have that

which can be lower bounded by Lemma 6. Note that G~\widetilde{\mathcal{G}} differs from G\mathcal{G} by a constant, we can still leverage Lemma 3. Hence, we obtain that

where cc is a constant, and the last step is implied by Lemma 4 and Lemma 7.

By the assumption η=c′/[μrσ1∗]\eta=c^{\prime}/[\mu r\sigma_{1}^{*}] for sufficiently small constant c′c^{\prime}, we thus have

Recall that ν:=(14β+81)αμr+26ϵ+18β−1\nu:=(14\beta+81)\alpha\mu r+26\sqrt{\epsilon}+18\beta^{-1}. By letting β=c1κ\beta=c_{1}\kappa, ϵ=c2/κ2\epsilon=c_{2}/\kappa^{2} and assuming α≤c3/(μrκ2)\alpha\leq c_{3}/(\mu r\kappa^{2}) and δt≤c4σr∗/κ\delta_{t}\leq c_{4}\sigma_{r}^{*}/\kappa for some sufficiently small constants {ci}i=14\{c_{i}\}_{i=1}^{4}, we can have −2η(W1+W2)+η2(W3+W4)≤−164ησr∗δt-2\eta(W_{1}+W_{2})+\eta^{2}(W_{3}+W_{4})\leq-\frac{1}{64}\eta\sigma_{r}^{*}\delta_{t}, which implies that

6 Proof of Corollary 2

We need α≲1μκ2r\alpha\lesssim\frac{1}{\mu\kappa^{2}r} due to the condition of Theorem 4. Letting the initial error provided in Theorem 3 be less than the corresponding condition in Theorem 4, we have

Plugging the above two upper bounds into the second term in (13), it suffices to have

Comparing the above bound with the second term in (14) completes the proof.

7 Proof of Lemma 2

Plugging it back into the left hand side of (21), we obtain

Next we derive upper bounds of T1T_{1} and T2T_{2} respectively.

We denote the support of SS, S∗S^{*} by Ω\Omega and Ω∗\Omega^{*} respectively. Since S−S∗S-S^{*} is supported on Ω∗∪Ω\Omega^{*}\cup\Omega, we have

Recall that for any (i,j)∈Ω(i,j)\in\Omega, we have ==. Accordingly, we have

Now we turn to bound W2W_{2}. Since S(i,j)=0S_{(i,j)}=0 for any (i,j)∈Ω∗∖Ω(i,j)\in\Omega^{*}\setminus\Omega, we have

Let uiu_{i} be the ii-th row of M−M∗M-M^{*}, and vjv_{j} be the jj-th column of M−M∗M-M^{*}. For any k∈[d2]k\in[d_{2}], we let ui(k)u_{i}^{(k)} denote the element of uiu_{i} that has the kk-th largest magnitude. Similarly, for any k∈[d1]k\in[d_{1}], we let vj(k)v_{j}^{(k)} denote the element of vjv_{j} that has the kk-th largest magnitude.

From the design of sparse estimator (4), we have that for any (i,j)∈Ω∗∖Ω(i,j)\in\Omega^{*}\setminus\Omega, ∣∣|| is either smaller than the γαd2\gamma\alpha d_{2}-th largest entry of the ii-th row of M∗+S∗−MM^{*}+S^{*}-M or smaller than the γαd1\gamma\alpha d_{1}-th largest entry of the jj-th column of M∗+S∗−MM^{*}+S^{*}-M. Note that S∗S^{*} only contains at most α\alpha-fraction nonzero entries per row and column. As a result, ∣∣|| has to be less than the magnitude of ui(γαd2−αd2)u_{i}^{(\gamma\alpha d_{2}-\alpha d_{2})} or vj(γαd1−αd1)v_{j}^{(\gamma\alpha d_{1}-\alpha d_{1})}. Formally, we have for (i,j)∈Ω∗∖Ω(i,j)\in\Omega^{*}\setminus\Omega,

Meanwhile, for any (i,j)∈Ω∗∖Ω(i,j)\in\Omega^{*}\setminus\Omega, we have

where β\beta in the last step can be any positive number. Combining (38) and (39) leads to

We introduce shorthand δ:=∣ ⁣∣ ⁣∣ΔU∣ ⁣∣ ⁣∣\mboxF2+∣ ⁣∣ ⁣∣ΔV∣ ⁣∣ ⁣∣\mboxF2\delta:=|\!|\!|\Delta_{U}|\!|\!|_{{\mbox{\tiny{F}}}}^{2}+|\!|\!|\Delta_{V}|\!|\!|_{{\mbox{\tiny{F}}}}^{2}. We prove the following inequality in the end of this section.

where the last step follows from Lemma 14 by noticing that ΠΩ(M−M∗)\Pi_{\Omega}(M-M^{*}) has at most γα\gamma\alpha-fraction nonzero entries per row and column.

To ease notation, we let C:=M+S−M∗−S∗C:=M+S-M^{*}-S^{*}. We observe that CC is supported on Ωc\Omega^{c}, we have

where the last step follows from (42) and ∣ ⁣∣ ⁣∣ΔU∣ ⁣∣ ⁣∣\mboxF∣ ⁣∣ ⁣∣ΔV∣ ⁣∣ ⁣∣\mboxF≤δ/2|\!|\!|\Delta_{U}|\!|\!|_{{\mbox{\tiny{F}}}}|\!|\!|\Delta_{V}|\!|\!|_{{\mbox{\tiny{F}}}}\leq\delta/2.

It remains to bound W4W_{4}. By Cauchy-Swartz inequality, we have

where step (a)(a) is from (37), step (b)(b) follows from (38), and step (c)(c) follows from (6.7). Combining the upper bounds of W3W_{3} and W4W_{4}, we obtain

Combining pieces.

Now we choose γ=2\gamma=2. Then inequality (43) implies that

Plugging the above two inequalities into (35) completes the proof.

where the first step follows from the upper bound of ∣ ⁣∣ ⁣∣M−M∗∣ ⁣∣ ⁣∣\mboxF|\!|\!|M-M^{*}|\!|\!|_{{\mbox{\tiny{F}}}} shown in Lemma 12, and the second step follows from the assumption ∣ ⁣∣ ⁣∣ΔU∣ ⁣∣ ⁣∣\mboxF,∣ ⁣∣ ⁣∣ΔV∣ ⁣∣ ⁣∣\mboxF≤σ1∗|\!|\!|\Delta_{U}|\!|\!|_{{\mbox{\tiny{F}}}},|\!|\!|\Delta_{V}|\!|\!|_{{\mbox{\tiny{F}}}}\leq\sqrt{\sigma_{1}^{*}}.

8 Proof of Lemma 3

where the last step follows from ΔU⊤Uπ∗−ΔV⊤Vπ∗=U⊤Uπ∗−V⊤Vπ∗\Delta_{U}^{\top}U_{\pi^{*}}-\Delta_{V}^{\top}V_{\pi^{*}}=U^{\top}U_{\pi^{*}}-V^{\top}V_{\pi^{*}} since Uπ∗⊤Uπ∗=Vπ∗⊤Vπ∗U_{\pi^{*}}^{\top}U_{\pi^{*}}=V_{\pi^{*}}^{\top}V_{\pi^{*}}. Note that

where we use Uπ∗⊤Uπ∗=Vπ∗⊤Vπ∗U_{\pi^{*}}^{\top}U_{\pi^{*}}=V_{\pi^{*}}^{\top}V_{\pi^{*}} again in the last step. Furthermore, since U⊤U−V⊤VU^{\top}U-V^{\top}V is symmetric, we have

Using these arguments, for the second term in (45), denoted by T2T_{2}, we have

It remains to find a lower bound of ∣ ⁣∣ ⁣∣U⊤U−V⊤V∣ ⁣∣ ⁣∣\mboxF|\!|\!|U^{\top}U-V^{\top}V|\!|\!|_{{\mbox{\tiny{F}}}}. The following inequality, which we turn to prove later, is true:

Proceeding with the first term in (45) by using (47), we get

Introduce ΔF:=F−Fπ∗\Delta_{F}:=F-F_{\pi^{*}}. Recall that δ:=∣ ⁣∣ ⁣∣ΔU∣ ⁣∣ ⁣∣\mboxF2+∣ ⁣∣ ⁣∣ΔV∣ ⁣∣ ⁣∣\mboxF2\delta:=|\!|\!|\Delta_{U}|\!|\!|_{{\mbox{\tiny{F}}}}^{2}+|\!|\!|\Delta_{V}|\!|\!|_{{\mbox{\tiny{F}}}}^{2}. Equivalently δ=∣ ⁣∣ ⁣∣ΔF∣ ⁣∣ ⁣∣\mboxF2\delta=|\!|\!|\Delta_{F}|\!|\!|_{{\mbox{\tiny{F}}}}^{2}. We have

For the cross term, by the following result, proved in (we also provide a proof in Section 7.5 for the sake of completeness), we have ⟨ ⁣⟨ΔFFπ∗⊤,  Fπ∗ΔF⊤⟩ ⁣⟩≥0\langle\!\langle{\Delta_{F}F^{\top}_{\pi*}},\;{F_{\pi^{*}}\Delta_{F}^{\top}}\rangle\!\rangle\geq 0.

When ∣ ⁣∣ ⁣∣F−Fπ∗∣ ⁣∣ ⁣∣\mboxop<2σr∗|\!|\!|F-F_{\pi^{*}}|\!|\!|_{{\mbox{\tiny{op}}}}<\sqrt{2\sigma_{r}^{*}}, we have that ΔF⊤Fπ∗\Delta_{F}^{\top}F_{\pi^{*}} is symmetric.

Accordingly, we have ∣ ⁣∣ ⁣∣FF⊤−Fπ∗Fπ∗⊤∣ ⁣∣ ⁣∣\mboxF≥2σr∗δ−δ≥σr∗δ|\!|\!|FF^{\top}-F_{\pi^{*}}F^{\top}_{\pi^{*}}|\!|\!|_{{\mbox{\tiny{F}}}}\geq 2\sqrt{\sigma_{r}^{*}\delta}-\delta\geq\sqrt{\sigma_{r}^{*}\delta} under condition δ≤σr∗\delta\leq\sigma_{r}^{*}. Plugging this lower bound into (48), we obtain

Putting (45), (46) and the above inequality together completes the proof.

Proof of inequality (47). For the term on the left hand side of (47), it is easy to check that

The property Uπ∗⊤Uπ∗=Vπ∗⊤Vπ∗U_{\pi^{*}}^{\top}U_{\pi^{*}}=V_{\pi^{*}}^{\top}V_{\pi^{*}} implies that ∣ ⁣∣ ⁣∣Uπ∗Uπ∗⊤∣ ⁣∣ ⁣∣\mboxF=∣ ⁣∣ ⁣∣Vπ∗Vπ∗⊤∣ ⁣∣ ⁣∣\mboxF=∣ ⁣∣ ⁣∣Uπ∗Vπ∗⊤∣ ⁣∣ ⁣∣\mboxF|\!|\!|U_{\pi^{*}}U_{\pi^{*}}^{\top}|\!|\!|_{{\mbox{\tiny{F}}}}=|\!|\!|V_{\pi^{*}}V_{\pi^{*}}^{\top}|\!|\!|_{{\mbox{\tiny{F}}}}=|\!|\!|U_{\pi^{*}}V_{\pi^{*}}^{\top}|\!|\!|_{{\mbox{\tiny{F}}}}. Therefore, expanding those quadratic terms on the right hand side of (47), one can show that it is equal to

Comparing inequalities (49) and (50), it thus remains to show that

Equivalently, we always have ∣ ⁣∣ ⁣∣Uπ∗⊤U−Vπ∗⊤V∣ ⁣∣ ⁣∣\mboxF2≥0|\!|\!|U_{\pi^{*}}^{\top}U-V_{\pi^{*}}^{\top}V|\!|\!|_{{\mbox{\tiny{F}}}}^{2}\geq 0, and thus prove (47).

9 Proof of Lemma 4

Now we turn to prove (22). We observe that

where we let M:=UV⊤M:=UV^{\top}. We denote the support of S,S∗S,S^{*} by Ω\Omega and Ω∗\Omega^{*} respectively. Based on the sparse estimator (4) for computing SS, ∇ML(U,V;S)\nabla_{M}\mathcal{L}(U,V;S) is only supported on Ωc\Omega^{c}. We thus have

It remains to upper bound the second term on the right hand side. Following (37) and (38), we have

where the last step is proved in (6.7). By choosing γ=2\gamma=2, we thus conclude that

10 Proof of Lemma 6

We denote the support of ΠΦ(S∗)\Pi_{\Phi}(S^{*}), SS by Ωo∗\Omega^{*}_{o} and Ω\Omega. We always have Ωo∗⊆Φ\Omega^{*}_{o}\subseteq\Phi and Ω⊆Φ\Omega\subseteq\Phi.

In the sequel, we establish several results that characterize the properties of Φ\Phi. The first result, proved in Section 7.2, shows that the Frobenius norm of any incoherent matrix whose row (or column) space are equal to L∗L^{*} (or R∗R^{*}) is well preserved under partial observations supported on Φ\Phi.

We need the next result, proved in Section 7.3, to control the number of nonzero entries per row and column in Ωo∗\Omega^{*}_{o} and Φ\Phi.

If p≥563log⁡dα(d1∧d2)p\geq\frac{56}{3}\frac{\log d}{\alpha(d_{1}\wedge d_{2})}, then with probability at least 1−6d−11-6d^{-1}, we have

The next lemma, proved in Section 7.4, can be used to control the projection of small matrices to Φ\Phi.

In the remainder of this section, we condition on the events in Lemmas 9, 10 and 11. Now we are ready to prove Lemma 6.

Plugging it back into the left hand side of (21), we obtain

Next we derive lower bounds of T1T_{1}, upper bounds of T2T_{2} and T3T_{3} respectively.

We observe that M−M∗=Uπ∗∗ΔV⊤+ΔUVπ∗⊤+ΔUΔV⊤M-M^{*}=U_{\pi^{*}}^{*}\Delta_{V}^{\top}+\Delta_{U}V_{\pi^{*}}^{\top}+\Delta_{U}\Delta_{V}^{\top}. By triangle inequality, we have

Note that when c≥a−bc\geq a-b for a,b≥0a,b\geq 0, we always have c2≥12a2−b2c^{2}\geq\frac{1}{2}a^{2}-b^{2}. We thus have

where the second step is implied by Lemma 9, the third step follows from (51) in Lemma 11 by noticing that ∣ ⁣∣ ⁣∣ΔU∣ ⁣∣ ⁣∣2,∞≤3μrσ1∗/d1|\!|\!|\Delta_{U}|\!|\!|_{{2,\infty}}\leq 3\sqrt{\mu r\sigma_{1}^{*}/d_{1}} and ∣ ⁣∣ ⁣∣ΔV∣ ⁣∣ ⁣∣2,∞≤3μrσ1∗/d1|\!|\!|\Delta_{V}|\!|\!|_{{2,\infty}}\leq 3\sqrt{\mu r\sigma_{1}^{*}/d_{1}}, which is further implied by (31).

Since S−S∗S-S^{*} is supported on Ω0∗∪Ω\Omega^{*}_{0}\cup\Omega, we have

For any (i,j)∈Ω(i,j)\in\Omega, we have (S−S∗)(i,j)=(M∗−M)(i,j)(S-S^{*})_{(i,j)}=(M^{*}-M)_{(i,j)}. Therefore, for the second term on the right hand side, we have

where the last inequality follows from Lemma 14 and the fact that ∣Ω(i,⋅)∣≤γpαd2|\Omega_{(i,\cdot)}|\leq\gamma p\alpha d_{2}, ∣Ω(⋅,j)∣≤γpαd1|\Omega_{(\cdot,j)}|\leq\gamma p\alpha d_{1} for all i∈[d1]i\in[d_{1}], j∈[d2]j\in[d_{2}].

We denote the ii-th row of ΠΦ(M−M∗)\Pi_{\Phi}(M-M^{*}) by uiu_{i}, and we denote the jj-th column of ΠΦ(M−M∗)\Pi_{\Phi}(M-M^{*}) by vjv_{j}. We let ui(k)u_{i}^{(k)} denote the element of uiu_{i} that has the kk-th largest magnitude. We let vj(k)v_{j}^{(k)} denote the element of vjv_{j} that has the kk-th largest magnitude.

For the first term on the right hand side of (55), we first observe that for (i,j)∈Ωo∗∖Ω(i,j)\in\Omega^{*}_{o}\setminus\Omega, ∣(M∗+S∗−M)(i,j)∣|(M^{*}+S^{*}-M)_{(i,j)}| is either less than the γpαd2\gamma p\alpha d_{2}-th largest element in the ii-th row of ΠΦ(M∗+S∗−M)\Pi_{\Phi}(M^{*}+S^{*}-M), or less than γpαd1\gamma p\alpha d_{1}-th largest element in the jj-th row of ΠΦ(M∗+S∗−M)\Pi_{\Phi}(M^{*}+S^{*}-M). Based on Lemma 10, ΠΦ(S∗)\Pi_{\Phi}(S^{*}) has at most 3pαd2/23p\alpha d_{2}/2 nonzero entries per row and at most 3pαd1/23p\alpha d_{1}/2 nonzero entries per column. Therefore, we have

where the second step holds for any β>0\beta>0 and the last step follows from Lemma 14 under the size constraints of Ωo∗\Omega^{*}_{o} shown in Lemma 10. For the second term in (58), using (57), we have

where the second step follows from Lemma 9 and inequality (51) in Lemma 11. Putting (55)-(60) together, we obtain

where we use (51) in Lemma 11 in the second step.

We observe that ΠΦ(M−M∗+S−S∗)\Pi_{\Phi}(M-M^{*}+S-S^{*}) is supported on Φ∖Ω\Phi\setminus\Omega. Therefore, we have

where the third step follows from (6.10), and the last step is from (60). Under assumptions γ=3\gamma=3, ϵ≤1/4\epsilon\leq 1/4 and δ≤σ1∗\delta\leq\sigma_{1}^{*}, we have

Combining pieces.

Under the aforementioned assumptions, putting all pieces together leads to

11 Proof of Lemma 7

Conditioning on the event in Lemma 11, since (U,V)∈\makebox[0.0pt][l]U×\makebox[0.0pt][l]V(U,V)\in\makebox[0.0pt][l]{\hskip 2.05pt\rule[8.12498pt]{5.17749pt}{0.43057pt}}{\mathcal{U}}\times\makebox[0.0pt][l]{\hskip 2.05pt\rule[8.12498pt]{5.17749pt}{0.43057pt}}{\mathcal{V}}, inequalities (52) and (53) imply that

It remains to bound the term ∣ ⁣∣ ⁣∣ΠΦ(M+S−M∗−S∗)∣ ⁣∣ ⁣∣\mboxF2|\!|\!|\Pi_{\Phi}\left(M+S-M^{*}-S^{*}\right)|\!|\!|_{{\mbox{\tiny{F}}}}^{2}. Let Ωo∗\Omega^{*}_{o} and Ω\Omega be the support of ΠΦ(S∗)\Pi_{\Phi}(S^{*}) and SS respectively. We observe that

In the proof of Lemma 6, it is shown in (6.10) that

We thus finish proving our conclusion by combining all pieces and noticing that γ=3\gamma=3 and ϵ≤1/4\epsilon\leq 1/4.

Proofs for Technical Lemmas

In this section, we prove several technical lemmas that are used in the proofs of our main theorems.

It is thus implied that ∣ ⁣∣ ⁣∣A∣ ⁣∣ ⁣∣\mboxop≤12α(β−1d2+βd1)∥A∥∞|\!|\!|A|\!|\!|_{{\mbox{\tiny{op}}}}\leq\frac{1}{2}\alpha(\beta^{-1}d_{2}+\beta d_{1})\|A\|_{\infty}. Choosing β=d2/d1\beta=\sqrt{d_{2}/d_{1}} completes the proof.

2 Proof of Lemma 9

holds with probability at least 1−2d−31-2d^{-3}.

In our setting, by restricting X=L∗A⊤+BR∗⊤X=L^{*}A^{\top}+BR^{*\top}, we have ΠKX=X\Pi_{\mathcal{K}}X=X. Therefore, (61) implies that

For ∣ ⁣∣ ⁣∣ΠΦX∣ ⁣∣ ⁣∣\mboxF2|\!|\!|\Pi_{\Phi}X|\!|\!|_{{\mbox{\tiny{F}}}}^{2}, we have

Combining the above two inequalities, we complete the proof.

3 Proof of Lemma 10

We observe that ∣Φ(i,⋅)∣|\Phi_{(i,\cdot)}| is a summation of d2d_{2} i.i.d. binary random variables with mean pp and variance p(1−p)p(1-p). By Bernstein’s inequality, for any i∈[d1]i\in[d_{1}],

where the last inequality holds by assuming p≥563log⁡dd2p\geq\frac{56}{3}\frac{\log d}{d_{2}}.

The term ∣Ωo(i,⋅)∗∣|\Omega^{*}_{o(i,\cdot)}| is a summation of at most αd2\alpha d_{2} i.i.d. binary random variables with mean pp and variance p(1−p)p(1-p). Again, applying Bernstein’s inequality leads to

Accordingly, by the assumption p≥563log⁡dαd2p\geq\frac{56}{3}\frac{\log d}{\alpha d_{2}}, we obtain

The proofs for ∣Φ(⋅,j)∣|\Phi_{(\cdot,j)}| and ∣Ωo(⋅,j)∗∣|\Omega^{*}_{o(\cdot,j)}| follow the same idea.

4 Proof of Lemma 11

By the assumption p≳μ2r2log⁡dϵ2(d1∧d2)p\gtrsim\frac{\mu^{2}r^{2}\log d}{\epsilon^{2}(d_{1}\wedge d_{2})}, we finish proving (51).

According to the proof of Lemma 10, if p≥clog⁡dd1∧d2p\geq c\frac{\log d}{d_{1}\wedge d_{2}}, with probability at least 1−O(d−1)1-\mathcal{O}(d^{-1}), we have ∣Φ(i,⋅)∣≤32pd2|\Phi_{(i,\cdot)}|\leq\frac{3}{2}pd_{2} and ∣Φ(⋅,j)∣≤32pd1|\Phi_{(\cdot,j)}|\leq\frac{3}{2}pd_{1} for all i∈[d1]i\in[d_{1}] and j∈[d2]j\in[d_{2}]. Conditioning on this event, we have

We thus finish proving (52). Inequality (53) can be proved in the same way.

5 Proof of Lemma 8

where pip_{i} is the ii-th column of Q1Q_{1} and qiq_{i} is the ii-th column of Q⊤Q2Q^{\top}Q_{2}. Hence, Tr(F⊤F∗Q)≤∑i∈[r]\text{Tr}(F^{\top}F^{*}Q)\leq\sum_{i\in[r]} and the equality holds if and only if pi=qip_{i}=q_{i} for all i∈[r]i\in[r] since every >0>0. We have Q1=Q⊤Q2Q_{1}=Q^{\top}Q_{2} and thus finish proving the argument.

In the second step, we use the fact that the singular values of Fπ∗F_{\pi^{*}} are equal to the diagonal terms of 2Σ∗1/2\sqrt{2}\Sigma^{*1/2}. Hence, F⊤Fπ∗F^{\top}F_{\pi^{*}} has full rank. Furthermore, it implies that F⊤F∗F^{\top}F^{*} has full rank and only contains positive singular values.

Proceeding with the proved argument, we have

which implies that F⊤Fπ∗F^{\top}F_{\pi^{*}} is symmetric. Accordingly, we have (F−Fπ∗)⊤Fπ∗(F-F_{\pi^{*}})^{\top}F_{\pi^{*}} is also symmetric.

Acknowledgment

Y. Chen acknowledges support from the School of Operations Research and Information Engineering, Cornell University.

References

Appendix A Supporting Lemmas

In this section, we provide several technical lemmas used for proving our main results.

where ΔU:=U−U∗\Delta_{U}:=U-U^{*}, ΔV:=V−V∗\Delta_{V}:=V-V^{*}.

We observe that UV⊤−U∗V∗⊤=U∗ΔV⊤+ΔUV∗⊤+ΔUΔV⊤UV^{\top}-U^{*}V^{*\top}=U^{*}\Delta_{V}^{\top}+\Delta_{U}V^{*\top}+\Delta_{U}\Delta_{V}^{\top}. Hence,

Furthermore, assuming (U,V)∈U×V(U,V)\in\mathcal{U}\times\mathcal{V}, where U\mathcal{U} and V\mathcal{V} satisfy the conditions in (19), we have the next result.

For any (i,j)∈[d1]×[d2](i,j)\in[d_{1}]\times[d_{2}], we have

Lemma 13 can be used to prove the following result.

For any α∈\alpha\in, suppose Ω⊆[d1]×[d2]\Omega\subseteq[d_{1}]\times[d_{2}] satisfies ∣Ω(i,⋅)∣≤αd2|\Omega_{(i,\cdot)}|\leq\alpha d_{2} for all i∈[d1]i\in[d_{1}] and ∣Ω(⋅,j)∣≤αd1|\Omega_{(\cdot,j)}|\leq\alpha d_{1} for all j∈[d2]j\in[d_{2}]. Then we have

Using Lemma 13 for bounding each entry of UV⊤−U∗V∗⊤UV^{\top}-U^{*}V^{*\top}, we have that

Denote the ii-th largest singular value of matrix MM by σi(M)\sigma_{i}(M).

Appendix B Parameter Settings for FB Separation Experiments

We approximate the FB separation problem by the RPCA framework with r=10r=10, α=0.2\alpha=0.2, μ=10\mu=10. Our algorithmic parameters are set as γ=1\gamma=1, η=1/(2σ^1∗)\eta=1/(2\hat{\sigma}_{1}^{*}), where σ^1∗\hat{\sigma}_{1}^{*} is an estimate of σ1∗\sigma_{1}^{*} obtained from the initial SVD. The parameters of AltProj are kept as provided in the default setting. For IALM, we use the tradeoff paramter λ=1/d1\lambda=1/\sqrt{d_{1}}, where d1d_{1} is the number of pixels in each frame (the number of rows in YY).

Note that both IALM and AltProj use the stopping criterion

Our algorithm for the partial observation setting never explicitly forms the d1d_{1}-by-d2d_{2} matrix Mt=UtVt⊤M_{t}=U_{t}V_{t}^{\top}, which is favored in large scale problems, but also renders the above criterion inapplicable. Instead, we use the following stopping criterion

This rule checks whether the iterates corresponding to low-rank factors become stable. In fact, our stopping criterion seems more natural and practical because in most real applications, matrix YY cannot be strictly decomposed into low-rank MM and sparse SS that satisfy Y=M+SY=M+S. Instead of forcing M+SM+S to be close to YY, our rule relies on seeking a robust subspace that captures the most variance of YY.