Fast Algorithms for Robust PCA via Gradient Descent

Abstract: We consider the problem of Robust PCA in the fully and partially observed settings. Without corruptions, this is the well-known matrix completion problem. From a statistical standpoint this problem has been recently well-studied, and conditions on when recovery is possible (how many observations do we need, how many corruptions can we tolerate) via polynomial-time algorithms is by now understood. This paper presents and analyzes a non-convex optimization approach that greatly reduces the computational complexity of the above problems, compared to the best available algorithms. In particular, in the fully observed case, with $r$ denoting rank and $d$ dimension, we reduce the complexity from $\mathcal{O}(r^2d^2\log(1/\varepsilon))$ to $\mathcal{O}(rd^2\log(1/\varepsilon))$ -- a big savings when the rank is big. For the partially observed case, we show the complexity of our algorithm is no more than $\mathcal{O}(r^4d \log d \log(1/\varepsilon))$. Not only is this the best-known run-time for a provable algorithm under partial observation, but in the setting where $r$ is small compared to $d$, it also allows for near-linear-in-$d$ run-time that can be exploited in the fully-observed case as well, by simply running our algorithm on a subset of the observations.

\section{Introduction}
{\em Principal component analysis} (PCA) aims to find a low rank subspace that best-approximates a data matrix $Y \in \ensuremath{\mathbb{R}}^{d_1 \times d_2}$. The simple and standard method of PCA by {\em singular value decomposition} (SVD) fails in many modern data problems due to missing and corrupted entries, as well as sheer scale of the problem. Indeed, SVD is highly sensitive to outliers by virtue of the squared-error criterion it minimizes. Moreover, its running time scales as $\ensuremath{\mathcal{O}}(r d^2)$ to recover a rank $r$ approximation of a $d$-by-$d$ matrix.

While there have been recent results developing provably robust algorithms for PCA (e.g., \cite{candes2011robust,Xu2012RPCA}), the running times range from $\ensuremath{\mathcal{O}}(r^2 d^2)$ to $\ensuremath{\mathcal{O}}(d^3)$\footnote{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., \cite{frieze2004fast, clarkson2013low,bhojanapalli2015tighter}) 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).

\subsection{The Model and Related Work}
We consider the following setting for robust PCA. Suppose we are given a matrix $Y \in \ensuremath{\mathbb{R}}^{d_1 \times d_2}$ that has decomposition $Y = \ensuremath{ M^{*} } + \ensuremath{ S^{*} }$, where $\ensuremath{ M^{*} }$ is  a rank $r$ matrix and $\ensuremath{ S^{*} }$ is a sparse corruption matrix containing entries with arbitrary magnitude. The goal is to recover $\ensuremath{ M^{*} }$ and $\ensuremath{ S^{*} }$ from $Y$. To ease notation, we let $d_1 = d_2 = d$ in the remainder of this section.

Provable solutions for this model are first provided in the works of \cite{chandrasekaran2011rank} and \cite{candes2011robust}. They propose to solve this problem by {\em convex relaxation}:
\begin{equation} \label{eq:convex_rpca}
\min_{M, S} \nucnorm{M} + \lambda \|S\|_1, ~\text{s.t.}~ Y = M + S,
\end{equation}
where $\nucnorm{M}$ denotes the nuclear norm of $M$. Despite analyzing the same method, the corruption models in \cite{candes2011robust} and \cite{chandrasekaran2011rank} differ. In \cite{candes2011robust}, the authors consider the setting where the entries of $\ensuremath{ M^{*} }$ are corrupted at random with probability $\alpha$. They show their method succeeds in exact recovery with $\alpha$ as large as $0.1$, which indicates they can tolerate a constant fraction of corruptions. Work in \cite{chandrasekaran2011rank} considers a {\em deterministic corruption model}, where nonzero entries of $\ensuremath{ S^{*} }$ can have arbitrary position, but the sparsity of each row and column does not exceed $\alpha d$. They prove that for exact recovery, it can allow $\alpha = \ensuremath{\mathcal{O}}(1/(\mu r\sqrt{d}))$. This was subsequently further improved to $\alpha = \ensuremath{\mathcal{O}}(1/(\mu r))$, which is in fact optimal  \cite{chen2011LSarxiv, hsu2011robust}. Here, $\mu$ represents the incoherence of $\ensuremath{ M^{*} }$ (see Section \ref{sec:problem_setup} for details). In this paper, we follow this latter line and focus on the deterministic corruption model.

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

Non-convex approaches have recently seen numerous developments for applications in low-rank estimation, including alternating minimization (see e.g.\ \cite{jain2013low,hardt2014understanding,gu2016low}) and gradient descent  (see e.g.\ \cite{bhojanapalli2015dropping,chen2015fast, sun2015guaranteed,tu2015low,zhao2015nonconvex,zheng2015convergent}). These works have fast running times, yet do not provide robustness guarantees. One exception is \cite{chen2015fast}, where the authors analyze a row-wise $\ell_1$ projection method for recovering $\ensuremath{ S^{*} }$. Their analysis hinges on positive semidefinite $\ensuremath{ M^{*} }$, and the algorithm requires prior knowledge of the $\ell_1$ norm of every row of $\ensuremath{ S^{*} }$ and is thus prohibitive in practice. Another exception is work \cite{gu2016low}, which analyzes alternating minimization plus an overall sparse projection. Their algorithm is shown to tolerate at most a fraction of $\alpha = \ensuremath{\mathcal{O}}{(1/(\mu^{2/3} r^{2/3} d))}$ corruptions. As we discuss in Section \ref{sec:contribution}, we can allow $\ensuremath{ S^{*} }$ to have much higher sparsity $\alpha = \ensuremath{\mathcal{O}}{(1/(\mu r^{1.5}))}$, which is close to optimal.

After the initial post of our paper on arxiv, Cherapanamjeri et~al.\ \cite{cherapanamjeri2016nearly} 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 $\ensuremath{\mathcal{O}}(\mu^2r^2d\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 $\ensuremath{\mathcal{O}}(\mu^2r^2d\log d)$ of our approach (see Corollary \ref{cor:rpca_partial}) does not have such dependence. Moreover, their method requires computing rank-$r$ SVD of the observed matrices in every round of iterations. Our algorithm only needs to compute rank-$r$ 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 \cite{candes2015phase,chen2015solving,zhang2016provable}, EM algorithms \cite{balakrishnan2014statistical, wang2015high, yi2015regularized}, tensor decompositions \cite{anandkumar2014tensor} and second order method \cite{sun2015nonconvex}. It might be interesting to bring robust considerations to these works.

\subsection{Our Contributions} \label{sec:contribution}
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.\footnote{To ease presentation, the discussion here assumes $\ensuremath{ M^{*} }$ has constant condition number, whereas our results below show the dependence on condition number explicitly.}
\begin{enumerate}
\item 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.

\item 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 $\ensuremath{\mathcal{O}}(rd^2\log(1/\varepsilon))$ while ensuring robustness $\alpha = \ensuremath{\mathcal{O}}(1/(\mu r^{1.5}))$. Our algorithm is thus faster than the best previous known algorithm by a factor of $r$, and enjoys superior empirical performance as well.

\item 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 $\ensuremath{ M^{*} }$) even when run on no more than $\ensuremath{\mathcal{O}}(\mu^2r^2d\log d)$ random entries of $Y$. The computational cost is $\ensuremath{\mathcal{O}}(\mu^3r^4d\log d \log(1/\varepsilon))$. When rank $r$ is small compared to the dimension $d$, in fact this dramatically improves on our bound above, as our cost becomes nearly linear in $d$. We show, moreover, that this savings and robustness to erasures comes at {\em 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 $r$ is small.

\item An immediate corollary of the above result provides a guarantee for exact matrix completion, with general rectangular matrices, using $\ensuremath{\mathcal{O}}(\mu^2r^2d\log d)$ observed entries and $\ensuremath{\mathcal{O}}(\mu^3r^4d\log d \log(1/\varepsilon))$ time, thereby improving on existing results in~\cite{chen2015fast,sun2015guaranteed}.
\end{enumerate}

\subsection{Organization and Notation}
The remainder of this paper is organized as follows. In Section \ref{sec:problem_setup}, we formally describe our problem and assumptions. In Section \ref{sec:algorithm}, we present and describe our algorithms for fully (Algorithm \ref{alg:rpca}) and partially (Algorithm \ref{alg:rpca_partial}) observed settings. In Section \ref{sec:analyze_rpca}, we establish theoretical guarantees of Algorithm \ref{alg:rpca}. The theory for partially observed setting are presented in Section \ref{sec:rpca_partial}. The numerical results are collected in Section \ref{sec:num_results}. Sections \ref{sec:proofs}, \ref{sec:proof_lemmas} and Appendix \ref{sec:supporting_lemma} contain all the proofs and technical lemmas.

For any index set $\Omega \subseteq [d_1]\times [d_2]$, we let $\row{\Omega}{i} := \left\{(i, j) \in \Omega ~\big|~ j \in [d_2]\right\}$, $\column{\Omega}{j} := \left\{(i, j) \in \Omega ~\big|~ i \in [d_1]\right\}$. For any matrix $A \in \ensuremath{\mathbb{R}}^{d_1 \times d_2}$, we denote its projector onto support $\Omega$ by $\ensuremath{\Pi}_{\Omega}\left(A\right)$, i.e., the $(i,j)$-th entry of $\ensuremath{\Pi}_{\Omega}\left(A\right)$ is equal to $A$ if $(i,j) \in \Omega$ and zero otherwise. The $i$-th row and $j$-th column of $A$ are denoted by $\row{A}{i}$ and $\column{A}{j}$. The $(i,j)$-th entry is denoted as $\entry{A}{i}{j}$. Operator norm of $A$ is $\opnorm{A}$. Frobenius norm of $A$ is $\frobnorm{A}$. The $\ell_a/\ell_b$ norm of $A$ is denoted by $\matsnorm{A}{b,a}$, i.e., the $\ell_a$ norm of the vector formed by the $\ell_b$ norm of every row. For instance, $\twoinfnorm{A}$ stands for $\max_{i \in [d_1]}\twonorm{\row{A}{i}}$.

\section{Problem Setup} \label{sec:problem_setup}
We consider the problem where we observe a matrix $\ensuremath{Y} \in \ensuremath{\mathbb{R}}^{d_1\times d_2}$ that satisfies $\ensuremath{Y} = \ensuremath{ M^{*} } + \ensuremath{ S^{*} }$, where $\ensuremath{ M^{*} }$ has rank $r$, and $\ensuremath{ S^{*} }$ is corruption matrix with sparse support. Our goal is to recover $\ensuremath{ M^{*} }$ and $\ensuremath{ S^{*} }$. In the partially observed setting, in addition to sparse corruptions, we have erasures. We assume that each entry of $\ensuremath{ M^{*} } + \ensuremath{ S^{*} }$ is revealed independently with probability $p \in (0,1)$. In particular, for any $(i,j) \in [d_1]\times [d_2]$, we consider the Bernoulli model where
\begin{equation} \label{eq:Y}
Y_{(i,j)} = \begin{cases}
(\ensuremath{ M^{*} } + \ensuremath{ S^{*} })_{(i,j)}, & ~ \text{with probability}~ p; \\
*,&  ~\text{otherwise}.
\end{cases}
\end{equation}
We denote the support of $\ensuremath{Y}$ by $\Phi = \{(i,j) ~|~ Y_{(i,j)} \neq *\}$. Note that we assume $\ensuremath{ S^{*} }$ is not adaptive to $\Phi$. As is well-understood thanks to work in matrix completion, this task is impossible in general -- we need to guarantee that $\ensuremath{ M^{*} }$ is not both low-rank and sparse. To avoid such identifiability issues, we make the following standard assumptions on $\ensuremath{ M^{*} }$ and $\ensuremath{ S^{*} }$: (i) $\ensuremath{ M^{*} }$ is not near-sparse or ``spiky.'' We impose this by requiring $\ensuremath{ M^{*} }$ to be $\mu$-incoherent -- given a singular value decomposition (SVD)\footnote{Throughout this paper, we refer to SVD of rank $r$ matrix by form $\ensuremath{L} \Sigma \ensuremath{R}^{\top}$ where $\Sigma \in \ensuremath{\mathbb{R}}^{r \times r}$.}  $\ensuremath{ M^{*} } = \ensuremath{L}^* \ensuremath{\Sigma}^* \ensuremath{R}^{*\top}$, we assume that
\[
\twoinfnorm{\ensuremath{L}^*} \leq \sqrt{\frac{\mu r}{d_1}}, ~~ \twoinfnorm{\ensuremath{R}^*} \leq \sqrt{\frac{\mu r}{d_2}}.
\]
(ii) The entries of $\ensuremath{ S^{*} }$ are ``spread out'' -- for $\alpha \in [0,1)$, we assume $\ensuremath{ S^{*} } \in \Sparset{\alpha}$, where
\begin{equation} \label{eq:S_alpha}
\Sparset{\alpha} := \left\{ A \in \ensuremath{\mathbb{R}}^{d_1 \times d_2} ~\big|~ \zeronorm{\row{A}{i}} \leq \alpha d_2~\text{for all}~ i \in [d_1]~; \zeronorm{\column{A}{j}} \leq \alpha d_1~\text{for all}~ j \in [d_2]  \right\}.
\end{equation}
In other words, $\ensuremath{ S^{*} }$ contains at most $\alpha$-fraction nonzero entries per row and column.

\section{Algorithms} \label{sec:algorithm}
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 $\ensuremath{S_{\text{init}}}$ for $S^*$ based on the observed matrix $\ensuremath{Y}$, and then find a rank $r$ matrix factorized as $\ensuremath{U}_{0}\ensuremath{V}_{0}^{\top}$ that is a rough estimate of $\ensuremath{ M^{*} }$ by performing SVD on ($\ensuremath{Y} - \ensuremath{S_{\text{init}}}$). In the second phase, given $(\ensuremath{U}_{0}, \ensuremath{V}_{0})$, we perform an iterative method to produce series $\{(\ensuremath{U}_t, \ensuremath{V}_t)\}_{t=0}^{\infty}$. In each step $t$, we first apply our sparse estimator to produce a sparse matrix $\ensuremath{S}_{t}$ based on $(\ensuremath{U}_{t}, \ensuremath{V}_{t})$, and then perform a projected gradient descent step on the low-rank factorized space to produce $(\ensuremath{U}_{t+1}, \ensuremath{V}_{t+1})$. This flow is the same for full and partial observations, though a few details differ. Algorithm \ref{alg:rpca} gives the full observation algorithm, and Algorithm \ref{alg:rpca_partial} gives the partial observation algorithm. We now describe the key details of each algorithm.

\paragraph{Sparse Estimation.} A natural idea is to keep those entries of residual matrix $\ensuremath{Y} - \ensuremath{M}$ that have large magnitude. At the same time, we need to make use of the dispersed property of $\Sparset{\alpha}$ that every column and row contain at most $\alpha$-fraction of nonzero entries. Motivated by these two principles, we introduce the following sparsification operator: For any matrix $A \in \ensuremath{\mathbb{R}}^{d_1 \times d_2}$: for all $(i,j) \in [d_1]\times [d_2]$, we let
\begin{equation} \label{eq:sparse_estimator}
\UnifSparse{A}{\alpha} := \begin{cases} \entry{A}{i}{j}, & \text{if}~ \abs{\entry{A}{i}{j}} \geq \abs{\row{A}{i}^{(\alpha d_2)}} ~\text{and}~ \abs{\entry{A}{i}{j}} \geq \abs{\column{A}{j}^{(\alpha d_1)}} \\
0, & \text{otherwise}\end{cases},
\end{equation}
where $\row{A}{i}^{(k)}$ and  $\column{A}{j}^{(k)}$ denote the elements of $\row{A}{i}$ and $\column{A}{j}$ that have the $k$-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 $\UnifSparse{A}{\alpha} \in \Sparset{\alpha}$.

\begin{algorithm}[H]
	\caption{Fast RPCA}
	\label{alg:rpca}
	\begin{algorithmic}[1]
		\INPUT Observed matrix $\ensuremath{Y}$ with rank $r$ and corruption fraction $\alpha$; parameters $\gamma, \eta$; number of iterations $T$.
		\vskip .05in
		\REMARK  {\em Phase I: Initialization.}
		\State $\ensuremath{S_{\text{init}}} \leftarrow \UnifSparse{Y}{\alpha}$ \hskip .3in  {\em// see \eqref{eq:sparse_estimator} for the definition of $\UnifSparse{\cdot}{\alpha}$.}
		\State $[\ensuremath{L}, \ensuremath{\Sigma}, \ensuremath{R}] \leftarrow \text{SVD}_{r}[Y - \ensuremath{S_{\text{init}}}]$ \footnotemark
		\State $\ensuremath{U}_0 \leftarrow \ensuremath{L}\ensuremath{\Sigma}^{1/2}$, $\ensuremath{V}_0 \leftarrow \ensuremath{R}\ensuremath{\Sigma}^{1/2}$. Let $\ensuremath{\mathcal{U}}, \ensuremath{\mathcal{V}}$ be defined according to \eqref{eq:set}.
		\vskip .05in
		\REMARK {\em Phase II: Gradient based iterations.}
		\State $\ensuremath{U}_0 \leftarrow \ensuremath{\Pi}_{\ensuremath{\mathcal{U}}}\left(\ensuremath{U}_0\right)$, $\ensuremath{V}_0 \leftarrow \ensuremath{\Pi}_{\ensuremath{\mathcal{V}}}\left(\ensuremath{V}_0\right)$
		\For{$t = 0,1,\ldots, T-1$}
		\State $\ensuremath{S}_{t} \leftarrow \UnifSparse{Y - \ensuremath{U}_{t}\ensuremath{V}_{t}^{\top}}{\gamma \alpha}$
		\State $\ensuremath{U}_{t+1} \leftarrow \ensuremath{\Pi}_{\ensuremath{\mathcal{U}}}\left(\ensuremath{U}_{t} - \eta \nabla_{\ensuremath{U}}\ensuremath{\mathcal{L}}(\ensuremath{U}_{t}, \ensuremath{V}_{t}; \ensuremath{S}_{t}) - \frac{1}{2}\eta\ensuremath{U}_{t}(\ensuremath{U}_{t}^{\top}\ensuremath{U}_{t} - \ensuremath{V}_{t}^{\top}\ensuremath{V}_{t})\right)$
		\State $\ensuremath{V}_{t+1} \leftarrow \ensuremath{\Pi}_{\ensuremath{\mathcal{V}}}\left(\ensuremath{V}_{t} - \eta \nabla_{\ensuremath{V}}\ensuremath{\mathcal{L}}(\ensuremath{U}_{t}, \ensuremath{V}_{t}; \ensuremath{S}_{t}) - \frac{1}{2}\eta\ensuremath{V}_{t}(\ensuremath{V}_{t}^{\top}\ensuremath{V}_{t} - \ensuremath{U}_{t}^{\top}\ensuremath{U}_{t})\right)$
		\EndFor
		\OUTPUT $(\ensuremath{U}_T, \ensuremath{V}_T)$
	\end{algorithmic}
\end{algorithm}
\footnotetext{$\text{SVD}_r[A]$ stands for computing a rank-$r$ SVD of matrix $A$, i.e., finding the top $r$ singular values and vectors of $A$. Note that we only need to compute rank-$r$ SVD approximately (see the initialization error requirement in Theorem \ref{thm:initialization}) so that we can leverage fast iterative approaches such as block power method and Krylov subspace methods \cite{saad2011numerical}.}

\paragraph{Initialization.} In the {\em fully observed} setting, we compute $\ensuremath{S_{\text{init}}}$ based on $\ensuremath{Y}$ as $\ensuremath{S_{\text{init}}} = \UnifSparse{Y}{\alpha}$. In the {\em partially observed} setting with sampling rate $p$, we let $\ensuremath{S_{\text{init}}} = \UnifSparse{Y}{2p\alpha}$. In both cases, we then set $\ensuremath{U}_{0} = \ensuremath{L} \ensuremath{\Sigma}^{1/2}$ and $\ensuremath{V}_{0} = \ensuremath{R} \ensuremath{\Sigma}^{1/2}$, where $\ensuremath{L} \ensuremath{\Sigma} \ensuremath{R}^{\top}$ is an SVD of the best rank $r$ approximation of $Y - \ensuremath{S_{\text{init}}}$.

\paragraph{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 $\ensuremath{U}, \ensuremath{V}$ and $\ensuremath{S}$:
\begin{eqnarray}
\label{eq:loss}
\ensuremath{\mathcal{L}}(\ensuremath{U}, \ensuremath{V}; \ensuremath{S}) &:=& \frac{1}{2}\frobnorm{\ensuremath{U}\ensuremath{V}^{\top} + \ensuremath{S} - \ensuremath{Y}}^2, \qquad \mbox{(fully observed)} \\
\label{eq:loss_partial}
\widetilde{\ensuremath{\mathcal{L}}}(\ensuremath{U}, \ensuremath{V}; \ensuremath{S}) &:=& \frac{1}{2p}\frobnorm{\ensuremath{\Pi}_{\Phi}\left(\ensuremath{U}\ensuremath{V}^{\top} + \ensuremath{S} - \ensuremath{Y}\right)}^2. \qquad \mbox{(partially observed)}
\end{eqnarray}
Recall that our goal is to recover $\ensuremath{ M^{*} }$ that satisfies the $\mu$-incoherent condition. Given an SVD $\ensuremath{ M^{*} } = \ensuremath{L}^* \ensuremath{\Sigma} \ensuremath{R}^{*\top}$, we expect that the solution $(\ensuremath{U}, \ensuremath{V})$ is close to $(\ensuremath{L}^*\ensuremath{\Sigma}^{1/2}, \ensuremath{R}^*\ensuremath{\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 $\ensuremath{U}, \ensuremath{V}$ based on $\opnorm{\ensuremath{ M^{*} }}$. As $\opnorm{\ensuremath{ M^{*} }}$ is unavailable, given $\ensuremath{U}_0, \ensuremath{V}_0$ computed in the first phase, we rely on the sets $\ensuremath{\mathcal{U}}$, $\ensuremath{\mathcal{V}}$ defined as
\begin{equation} \label{eq:set}
\ensuremath{\mathcal{U}} := \left\{A \in \ensuremath{\mathbb{R}}^{d_1 \times r}~\big|~ \twoinfnorm{A} \leq \sqrt{\frac{2\mu r}{d_1}}\opnorm{\ensuremath{U}_0}\right\}, ~
\ensuremath{\mathcal{V}} := \left\{A \in \ensuremath{\mathbb{R}}^{d_2 \times r}~\big|~ \twoinfnorm{A} \leq \sqrt{\frac{2\mu r}{d_2}}\opnorm{\ensuremath{V}_0}\right\}.
\end{equation}
Now we consider the following optimization problems with constraints:
\begin{eqnarray} \label{eq:opt}
\min_{\ensuremath{U} \in \ensuremath{\mathcal{U}}, \ensuremath{V} \in \ensuremath{\mathcal{V}}, \ensuremath{S} \in \Sparset{\alpha}} && \ensuremath{\mathcal{L}}(\ensuremath{U}, \ensuremath{V}; \ensuremath{S}) + \frac{1}{8}\frobnorm{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}}^2, \qquad \mbox{(fully observed)} \\
\label{eq:opt_partial}
\min_{\ensuremath{U} \in \ensuremath{\mathcal{U}}, \ensuremath{V} \in \ensuremath{\mathcal{V}}, \ensuremath{S} \in \Sparset{p\alpha}} && \widetilde{\ensuremath{\mathcal{L}}}(\ensuremath{U}, \ensuremath{V}; \ensuremath{S}) + \frac{1}{64}\frobnorm{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}}^2. \qquad \mbox{(partially observed)}
\end{eqnarray}
The regularization term in the objectives above is used to encourage that $\ensuremath{U}$ and $\ensuremath{V}$ have the same scale. Given $(\ensuremath{U}_0,\ensuremath{V}_0)$, we propose the following iterative method to produce series $\{(\ensuremath{U}_t,\ensuremath{V}_t)\}_{t=0}^{\infty}$ and $\{\ensuremath{S}_t\}_{t=0}^{\infty}$. We give the details for the fully observed case -- the partially observed case is similar. For $t = 0,1,\ldots$, we update
$ \ensuremath{S}_{t} $ using the sparse estimator $\ensuremath{S}_{t} = \UnifSparse{Y - \ensuremath{U}_{t}\ensuremath{V}_{t}^{\top}}{\gamma \alpha}$, followed by a projected gradient update on $ \ensuremath{U}_{t} $ and $ \ensuremath{V}_t $
\begin{align*}
\ensuremath{U}_{t+1} & = \ensuremath{\Pi}_{\ensuremath{\mathcal{U}}}\left(\ensuremath{U}_{t} - \eta \nabla_{\ensuremath{U}}\ensuremath{\mathcal{L}}(\ensuremath{U}_{t}, \ensuremath{V}_{t}; \ensuremath{S}_{t}) - \frac{1}{2}\eta\ensuremath{U}_{t}(\ensuremath{U}_{t}^{\top}\ensuremath{U}_{t} - \ensuremath{V}_{t}^{\top}\ensuremath{V}_{t})\right), \\
\ensuremath{V}_{t+1} & = \ensuremath{\Pi}_{\ensuremath{\mathcal{V}}}\left(\ensuremath{V}_{t} - \eta \nabla_{\ensuremath{V}}\ensuremath{\mathcal{L}}(\ensuremath{U}_{t}, \ensuremath{V}_{t}; \ensuremath{S}_{t}) - \frac{1}{2}\eta\ensuremath{V}_{t}(\ensuremath{V}_{t}^{\top}\ensuremath{V}_{t} - \ensuremath{U}_{t}^{\top}\ensuremath{U}_{t})\right).
\end{align*}
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 \eqref{eq:opt}, where $S$ is replaced by the aforementioned sparse estimator in each step.

\begin{algorithm}[H]
	\caption{Fast RPCA with partial observations}
	\label{alg:rpca_partial}
	\begin{algorithmic}[1]
		\INPUT Observed matrix $\ensuremath{Y}$ with support $\Phi$; parameters $\tau, \gamma, \eta$; number of iterations $T$.
		\vskip .05in
		\REMARK  {\em Phase I: Initialization.}
		\State $\ensuremath{S_{\text{init}}} \leftarrow \UnifSparse{\ensuremath{\Pi}_{\Phi}(Y)}{2p\alpha}$
		\State $[\ensuremath{L}, \ensuremath{\Sigma}, \ensuremath{R}] \leftarrow \text{SVD}_{r}[\frac{1}{p}(Y - \ensuremath{S_{\text{init}}})]$
		\State $\ensuremath{U}_0 \leftarrow \ensuremath{L}\ensuremath{\Sigma}^{1/2}$, $\ensuremath{V}_0 \leftarrow \ensuremath{R}\ensuremath{\Sigma}^{1/2}$. Let $\ensuremath{\mathcal{U}}, \ensuremath{\mathcal{V}}$ be defined according to \eqref{eq:set}.
		\vskip .05in
		\REMARK {\em Phase II: Gradient based iterations.}
		\State $\ensuremath{U}_0 \leftarrow \ensuremath{\Pi}_{\ensuremath{\mathcal{U}}}\left(\ensuremath{U}_0\right)$, $\ensuremath{V}_0 \leftarrow \ensuremath{\Pi}_{\ensuremath{\mathcal{V}}}\left(\ensuremath{V}_0\right)$
		\For{$t = 0,1,\ldots, T-1$}
		\State $\ensuremath{S}_{t} \leftarrow \UnifSparse{\ensuremath{\Pi}_{\Phi}\left(Y - \ensuremath{U}_{t}\ensuremath{V}_{t}^{\top}\right)}{\gamma p \alpha}$
		\State $\ensuremath{U}_{t+1} \leftarrow \ensuremath{\Pi}_{\ensuremath{\mathcal{U}}}\left(\ensuremath{U}_{t} - \eta \nabla_{\ensuremath{U}}\widetilde{\ensuremath{\mathcal{L}}}(\ensuremath{U}_{t}, \ensuremath{V}_{t}; \ensuremath{S}_{t}) - \frac{1}{16}\eta\ensuremath{U}_{t}(\ensuremath{U}_{t}^{\top}\ensuremath{U}_{t} - \ensuremath{V}_{t}^{\top}\ensuremath{V}_{t})\right)$
		\State $\ensuremath{V}_{t+1} \leftarrow \ensuremath{\Pi}_{\ensuremath{\mathcal{V}}}\left(\ensuremath{V}_{t} - \eta \nabla_{\ensuremath{V}}\widetilde{\ensuremath{\mathcal{L}}}(\ensuremath{U}_{t}, \ensuremath{V}_{t}; \ensuremath{S}_{t}) - \frac{1}{16}\eta\ensuremath{V}_{t}(\ensuremath{V}_{t}^{\top}\ensuremath{V}_{t} - \ensuremath{U}_{t}^{\top}\ensuremath{U}_{t})\right)$
		\EndFor
		\OUTPUT $(\ensuremath{U}_T, \ensuremath{V}_T)$
	\end{algorithmic}
\end{algorithm}

\section{Main Results}

In this section, we establish theoretical guarantees for Algorithm~\ref{alg:rpca} in the fully observed setting and for Algorithm~\ref{alg:rpca_partial} in the partially observed setting.

\subsection{Analysis of Algorithm \ref{alg:rpca}} \label{sec:analyze_rpca}
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 $(\ensuremath{U}, \ensuremath{V})$ that minimize \eqref{eq:opt}. Given the SVD  of the true low-rank matrix $\ensuremath{ M^{*} } = \ensuremath{L}^*\ensuremath{\Sigma}^* \ensuremath{R}^{*\top}$, we let $\ensuremath{U^*} := \ensuremath{L}^*\ensuremath{\Sigma}^{*1/2}$ and $\ensuremath{V^*} := \ensuremath{R}^*\ensuremath{\Sigma}^{*1/2}$.  We also let $\sigma_1^* \geq \sigma_2^* \geq \ldots \geq \sigma_r^*$ be sorted nonzero singular values of $\ensuremath{ M^{*} }$, and denote the condition number of $\ensuremath{ M^{*} }$ by $\kappa$, i.e., $\kappa := \sigma_1^*/\sigma_r^*$. We define estimation error $d(\ensuremath{U}, \ensuremath{V}; \ensuremath{U^*}, \ensuremath{V^*})$ as the minimal Frobenius norm between $(\ensuremath{U}, \ensuremath{V})$ and $(\ensuremath{U^*}, \ensuremath{V^*})$ with respect to the optimal rotation, namely
\begin{equation} \label{eq:error_metric}
d(\ensuremath{U}, \ensuremath{V}; \ensuremath{U^*}, \ensuremath{V^*}) := \min_{\ensuremath{Q} \in \RotateSet{r}} \sqrt{ \frobnorm{\ensuremath{U} - \ensuremath{U^*} \ensuremath{Q}}^2 + \frobnorm{\ensuremath{V} - \ensuremath{V^*} \ensuremath{Q}}^2},
\end{equation}
for $\RotateSet{r}$ the set of $r$-by-$r$ orthonormal matrices. This metric controls reconstruction error, as
\begin{equation} \label{eq:error_relation}
\frobnorm{\ensuremath{U}\ensuremath{V}^{\top} - \ensuremath{ M^{*} }} \lesssim \sqrt{\sigma_1^*}d(\ensuremath{U}, \ensuremath{V}; \ensuremath{U^*}, \ensuremath{V^*}),
\end{equation}
when $d(\ensuremath{U}, \ensuremath{V}; \ensuremath{U^*}, \ensuremath{V^*}) \leq \sqrt{\sigma_1^*}$. We denote the local region around the optimum $(\ensuremath{U^*}, \ensuremath{V^*})$ with radius $\omega$ as
\[
\Neighbor{\omega} := \left\{(\ensuremath{U},\ensuremath{V}) \in \ensuremath{\mathbb{R}}^{d_1\times r} \times \ensuremath{\mathbb{R}}^{d_2 \times r} ~\big|~ d(\ensuremath{U},\ensuremath{V};\ensuremath{U^*}, \ensuremath{V^*}) \leq \omega\right\}.
\]

The next two theorems provide guarantees for the initialization phase and gradient iterations, respectively, of Algorithm \ref{alg:rpca}. The proofs are given in Sections \ref{proof:thm:initialization} and \ref{proof:thm:convergence}.

\begin{theorem}[Initialization]\label{thm:initialization} Consider the paired $(\ensuremath{U}_0,\ensuremath{V}_0)$ produced in the first phase of Algorithm \ref{alg:rpca}. If $\alpha \leq 1/(16\kappa \mu r)$,
	we have
	\[
	d(\ensuremath{U}_0, \ensuremath{V}_0; \ensuremath{U^*}, \ensuremath{V^*}) \leq 28 \sqrt{\kappa} \alpha \mu r\sqrt{r} \sqrt{\sigma_1^*}.
	\]
\end{theorem}

\begin{theorem}[Convergence] \label{thm:convergence}
	Consider the second phase of Algorithm \ref{alg:rpca}. Suppose we choose $\gamma = 2$ and $\eta = c/\sigma_1^*$ for any $c \leq 1/36$. There exist constants $c_1, c_2$ such that when $\alpha \leq c_1/(\kappa^2 \mu r)$,
	given any $(\ensuremath{U}_0, \ensuremath{V}_0) \in \Neighbor{c_2\sqrt{\sigma^*_r/\kappa}}$, the iterates $\{(\ensuremath{U}_t, \ensuremath{V}_t)\}_{t=0}^{\infty}$ satisfy
	\[
	d^2(U_t, V_t; \ensuremath{U^*}, \ensuremath{V^*}) \leq \left(1 - \frac{c}{8\kappa}\right)^td^2(U_0,V_0;\ensuremath{U^*}, \ensuremath{V^*}).
	\]
\end{theorem}
Therefore, using proper initialization and step size, the gradient iteration converges at a linear rate with a constant contraction factor $1 - \ensuremath{\mathcal{O}}(1/\kappa)$. To obtain relative precision $\varepsilon$ compared to the initial error, it suffices to perform $O(\kappa \log(1/\varepsilon))$ iterations. Note that the step size is chosen according to $1/\sigma_1^*$. When $\alpha \lesssim 1/(\mu \sqrt{\kappa r^3})$,  Theorem \ref{thm:initialization} and the inequality \eqref{eq:error_relation} together imply that $\opnorm{\ensuremath{U}_0\ensuremath{V}_0^{\top} - \ensuremath{ M^{*} }} \leq \frac{1}{2}\sigma_1^*$. Hence we can set the step size as $\eta = \ensuremath{\mathcal{O}}(1/\sigma_1(\ensuremath{U}_0\ensuremath{V}_0^{\top}))$ using being the top singular value $ \sigma_1(\ensuremath{U}_0\ensuremath{V}_0^{\top}) $ of the matrix $ \ensuremath{U}_0 \ensuremath{V}_0^\top $

Combining Theorems \ref{thm:initialization} and \ref{thm:convergence} implies the following result, proved in Section \ref{proof:cor:rpca}, that provides an overall guarantee for Algorithm \ref{alg:rpca}.

\begin{corollary} \label{cor:rpca}
	Suppose that
	\[
	\alpha \leq c\min\left\{\frac{1}{\mu\sqrt{\kappa r}^3}, \frac{1}{\mu \kappa^2r}\right\}
	\]
	for some constant $c$. Then for any $\varepsilon \in (0,1)$, Algorithm \ref{alg:rpca} with $T = \ensuremath{\mathcal{O}}(\kappa \log (1/\varepsilon))$ outputs a pair $(\ensuremath{U}_{T}, \ensuremath{V}_{T})$ that satisfies
	\begin{equation} \label{eq:final_error}
	\frobnorm{\ensuremath{U}_{T}\ensuremath{V}_{T}^{\top} - \ensuremath{ M^{*} }} \leq \varepsilon\cdot\sigma_r^*.
	\end{equation}
\end{corollary}

\vskip .05in
\begin{remark}[Time Complexity] \label{remark:complexity} For simplicity we assume $d_1 = d_2 = d$. Our sparse estimator \eqref{eq:sparse_estimator} can be implemented by finding the top $\alpha d$ elements of each row and column via partial quick sort, which has running time $\ensuremath{\mathcal{O}}({d^2\log(\alpha d)})$. Performing rank-$r$ SVD in the first phase and computing the gradient in each iteration both have complexity $\ensuremath{\mathcal{O}}(rd^2)$.\footnote{In fact, it suffices to compute the best rank-$ r $ approximation with running time independent of the eigen gap.} Algorithm \ref{alg:rpca} thus has total running time $\ensuremath{\mathcal{O}}(\kappa rd^2\log(1/\varepsilon))$ for achieving an $ \epsilon $ accuracy as in \eqref{eq:final_error}. We note that when $\kappa = \ensuremath{\mathcal{O}}(1)$, our algorithm is orderwise faster than the AltProj algorithm in \cite{netrapalli2014non}, which has running time $\mathcal{O}(r^2d^2\log(1/\varepsilon))$. Moreover, our algorithm only requires computing one singular value decomposition.
\end{remark}
\vskip .05in
\begin{remark}[Robustness]
	Assuming $\kappa = \ensuremath{\mathcal{O}}(1)$, our algorithm can tolerate corruption at a sparsity level up to $\alpha = \ensuremath{\mathcal{O}}(1/(\mu r \sqrt{r}))$. This is worse by a factor $\sqrt{r}$ compared to the optimal statistical guarantee $1/(\mu r)$ obtained in \cite{chen2011LSarxiv,hsu2011robust,netrapalli2014non}. This looseness is a consequence of the condition for $ (\ensuremath{U}_0, \ensuremath{V}_0) $ in Theorem~\ref{thm:convergence}. Nevertheless, when $ \mu r = \ensuremath{\mathcal{O}}(1) $, our algorithm can tolerate a constant $ \alpha $ fraction of corruptions.

	Notably, we show that gradient descent works in the case of $\alpha = \ensuremath{\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/(\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.

\end{remark}

\subsection{Analysis of Algorithm \ref{alg:rpca_partial}}\label{sec:rpca_partial}
We now move to the guarantees of Algorithm \ref{alg:rpca_partial}. 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 $r$, the complexity of Algorithm~\ref{alg:rpca_partial} has {\em near linear dependence on the dimension} $d$, instead of quadratic.

In the following discussion, we let $d := \max\{d_1, d_2\}$. The next two results, proved in Sections \ref{proof:thm:initial_partial} and \ref{proof:thm:convergence_partial}, control the quality of the initialization step, and then the gradient iterations.

\begin{theorem}[Initialization, partial observations] \label{thm:initial_partial}
	Suppose the observed indices $\Phi$ follow the Bernoulli model given in \eqref{eq:Y}. Consider the pair $(\ensuremath{U}_0,\ensuremath{V}_0)$ produced in the first phase of Algorithm \ref{alg:rpca_partial}. There exist constants $\{c_i\}_{i=1}^3$ such that for any $\epsilon \in (0, \sqrt{r}/(8c_1\kappa))$, if
	\begin{equation} \label{eq:condition_thm_initial_partial}
	\alpha \leq \frac{1}{64\kappa \mu r}, ~~ p \geq c_2\left(\frac{\mu r^2}{\epsilon^2} + \frac{1}{\alpha}\right)\frac{\log d}{d_1 \wedge d_2},
	\end{equation}
	then we have
	\[
	d(\ensuremath{U}_0, \ensuremath{V}_0; \ensuremath{U^*}, \ensuremath{V^*}) \leq 51\sqrt{\kappa} \alpha \mu r\sqrt{r} \sqrt{\sigma_1^*} + 7c_1\epsilon \sqrt{\kappa \sigma_1^*},
	\]
	with probability at least $1 - c_3d^{-1}$.
\end{theorem}

\begin{theorem}[Convergence, partial observations] \label{thm:convergence_partial}
	Suppose the observed indices $\Phi$ follow the Bernoulli model given in \eqref{eq:Y}. Consider the second phase of Algorithm \ref{alg:rpca_partial}. Suppose we choose $\gamma = 3$, and $\eta = c/(\mu r\sigma_1^*)$ for a sufficiently small constant $c$. There exist constants $\{c_i\}_{i=1}^4$ such that if
	\begin{equation} \label{eq:condition_thm_convergence_partial}
	\alpha \leq \frac{c_1}{\kappa^2 \mu r} ~~\text{and}~~ p \geq c_2 \frac{\kappa^4 \mu ^2 r^2\log d}{d_1 \wedge d_2},
	\end{equation}
	then with probability at least $1 - c_3d^{-1}$, the iterates $\{(\ensuremath{U}_t,\ensuremath{V}_t)\}_{t = 0}^{\infty}$ satisfy
	\[
	d^2(U_t, V_t; \ensuremath{U^*}, \ensuremath{V^*}) \leq \left(1 - \frac{c}{64\mu r\kappa}\right)^td^2(U_0,V_0;\ensuremath{U^*}, \ensuremath{V^*})
	\]
	for all $(\ensuremath{U}_0, \ensuremath{V}_0) \in \Neighbor{c_4\sqrt{\sigma^*_r/\kappa}}$.
\end{theorem}
The above result ensures linear convergence to $(\ensuremath{U^*},\ensuremath{V^*})$ (up to rotation) even when the gradient iterations are computed using partial observations. Note that setting $p = 1$ recovers Theorem \ref{thm:convergence} up to an additional factor $\mu r$ in the contraction factor. For achieving $\varepsilon$ relative accuracy, now we need $\ensuremath{\mathcal{O}}(\mu r \kappa\log(1/\varepsilon))$ iterations.

Putting Theorems \ref{thm:initial_partial} and \ref{thm:convergence_partial} together, we have the following overall guarantee, proved in Section \ref{proof:cor:rpca_partial}, for Algorithm \ref{alg:rpca_partial}.
\begin{corollary} \label{cor:rpca_partial}
	Suppose that
	\[
	\alpha \leq c\min\left\{\frac{1}{\mu\sqrt{\kappa r}^3}, \frac{1}{\mu \kappa^2r}\right\}, ~~ p \geq c'\frac{\kappa^4\mu^2 r^2\log d}{d_1 \wedge d_2},
	\]
	for some constants $c, c'$. With probability at least $1 - \ensuremath{\mathcal{O}}(d^{-1})$, for any $\varepsilon \in (0,1)$, Algorithm \ref{alg:rpca_partial} with $T = \ensuremath{\mathcal{O}}(\mu r\kappa \log (1/\varepsilon))$ outputs a pair $(\ensuremath{U}_{T}, \ensuremath{V}_{T})$ that satisfies
	\begin{equation} \label{eq:final_error_partial}
	\frobnorm{\ensuremath{U}_{T}\ensuremath{V}_{T}^{\top} - \ensuremath{ M^{*} }} \leq \varepsilon\cdot\sigma_r^*.
	\end{equation}
\end{corollary}
This result shows that partial observations do not compromise robustness to sparse corruptions: as long as the observation probability $ p $ satisfies the condition in Corollary~\ref{cor:rpca_partial}, Algorithm \ref{alg:rpca_partial} 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 $d_1 = d_2 = d$, $\kappa = \ensuremath{\mathcal{O}}(1)$.
\vskip .05in
\begin{remark}[Sample complexity and matrix completion] Using the lower bound on $p$, it is sufficient to have $\ensuremath{\mathcal{O}}(\mu^2 r^2 d\log d)$  observed entries. In the special case $S^* = 0$, our partial observation model is equivalent to the model of exact matrix completion (see, e.g., \cite{candes2010power}). 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 \cite{chen2015fast}, and is better than the non-convex matrix completion algorithms in \cite{jain2013low} and \cite{sun2015guaranteed}. 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 $\Omega(\mu rd\log d)$~\cite{candes2010power}, and a nearly tight upper bound $O(\mu rd\log^2d)$ is obtained in \cite{DBLP:journals/tit/Chen15} by convex relaxation. While sub-optimal by a factor $\mu r$, our algorithm is much faster than convex relaxation as shown below.
\end{remark}
\vskip .05in
\begin{remark}[Time complexity]
	Our sparse estimator on the sparse matrix with support $\Phi$ can be implemented  via partial quick sort with running time $\ensuremath{\mathcal{O}}({pd^2\log(\alpha p d)})$. Computing the gradient in each step involves the two terms in the objective function \eqref{eq:opt_partial}. Computing the gradient of the first term $\widetilde{\mathcal{L}}$ takes time $\ensuremath{\mathcal{O}}(r|\Phi|)$, whereas the second term takes time $\ensuremath{\mathcal{O}}(r^2d)$. In the initialization phase, performing rank-$r$ SVD on a sparse matrix with support $\Phi$ can be done in time $\ensuremath{\mathcal{O}}(r|\Phi|)$. We conclude that when $|\Phi| = \ensuremath{\mathcal{O}}(\mu^2 r^2 d\log d)$, Algorithm \ref{alg:rpca_partial} achieves the error bound \eqref{eq:final_error_partial} with running time $\ensuremath{\mathcal{O}}(\mu^3r^4d\log d \log(1/\varepsilon))$. Therefore, in the small rank setting with $r \ll d^{1/3}$, even when full observations are given, it is better to use Algorithm \ref{alg:rpca_partial} by subsampling the entries of $ Y $.
\end{remark}

\section{Numerical Results} \label{sec:num_results}

In this section, we provide numerical results and compare the proposed algorithms with existing methods, including the inexact augmented lagrange multiplier (IALM) approach \cite{LinChenMa10} for solving the convex relaxation \eqref{eq:convex_rpca}  and the alternating projection (AltProj) algorithm proposed in \cite{netrapalli2014non}. All algorithms are implemented in MATLAB \footnote{Our code is available at \url{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.\footnote{\url{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.

\subsection{Synthetic Datasets}

 We generate a squared data matrix $Y = \ensuremath{ M^{*} } + \ensuremath{ S^{*} } \in \mathbb{R}^{d \times d}$ as follows. The low-rank part $\ensuremath{ M^{*} }$ is given by $M^* = A B^\top$, where $A,B \in \mathbb{R}^{d \times r}$ have entries drawn independently from a zero mean Gaussian distribution with variance $1/d$. For a given sparsity parameter $\alpha$, each entry of $\ensuremath{ S^{*} }$ is set to be nonzero with probability $\alpha$, and the values of the nonzero entries are sampled uniformly from $[-5r/d, 5r/d]$.
\begin{figure}[t]
	\centering
	\begin{subfigure}[t]{0.32\textwidth}
		\centering
		\includegraphics[width=\textwidth]{conv_rate}
		\caption{}
		\label{fig:conv_rate}
	\end{subfigure}%
	~
	\begin{subfigure}[t]{0.32\textwidth}
		\centering
		\includegraphics[width=\textwidth]{time_scale}
		\caption{}
		\label{fig:time_scale}
	\end{subfigure}
	~
	\begin{subfigure}[t]{0.32\textwidth}
		\centering
		\includegraphics[width=\textwidth]{real_time}
		\caption{}
		\label{fig:real_time}
	\end{subfigure}
	\caption{Results on synthetic data. (a) Plot of log estimation error versus number of iterations when using gradient descent (GD) with varying sub-sampling rate $p$. It is conducted using $d = 5000, r = 10, \alpha = 0.1$. (b) Plot of running time of GD versus dimension $d$ with $r = 10, \alpha = 0.1, p = 0.15r^2\log d/d$. The low-rank matrix is recovered in all instances, and the line has slope approximately one. (c) Plot of log estimation error versus running time for different algorithms in problem with $d = 5000, r = 10, \alpha = 0.1$.}
	\label{fig:exp1}
\end{figure}

The results are summarized in Figure \ref{fig:exp1}. Figure \ref{fig:conv_rate} shows the convergence of our algorithms for different random instances with different sub-sampling rate $p$ (note that $p = 1$ corresponds to the fully observed setting). As predicted by Theorems \ref{thm:convergence} and \ref{thm:convergence_partial},  our gradient method converges geometrically with a contraction factor nearly independent of $p$. Figure \ref{fig:time_scale} shows the running time of our algorithm with partially observed data. We see that the running time scales linearly with $d$, again consistent with the theory. We note that our algorithm is memory-efficient: in the large scale setting with $d = 2 \times 10^5$, using approximately $0.1\%$ entries is sufficient for the successful recovery. In contrast, AltProj and IALM are designed to manipulate the entire matrix with $d^2 = 4 \times 10^{10}$ entries, which is prohibitive on a single machine. Figure \ref{fig:real_time} 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.

\begin{figure}
	\centering
	\begin{tabular}{c}
		\includegraphics[scale=0.75, clip, trim = 0 0 0 0]{restaurant} \\
		%
		\includegraphics[scale=0.75, clip, trim = 0 0 0 0]{shoppingmall}
	\end{tabular}
	\caption{Foreground-background separation in \emph{Restaurant} and \emph{ShoppingMall} videos. In each line, the leftmost image is an original frame, and the other four are the separated background obtained from our algorithms with $p = 1$, $p = 0.2$, AltProj, and IALM. The running time required by each algorithm is shown in the title.}
	\label{fig:exp2}
\end{figure}

\subsection{Foreground-background Separation}

We apply our method to the task of foreground-background (FB) separation in a video. We use two public benchmarks, the \emph{Restaurant} and \emph{ShoppingMall} datasets.\footnote{\url{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 $Y$, the FB separation problem can be cast as RPCA, where the static background corresponds to a low rank matrix $M^*$ with identical columns, and the moving objects in the video can be modeled as sparse corruptions $S^*$. Figure \ref{fig:exp2} 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 \ref{fig:exp3} 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 \ref{sec:fb_more}, we describe the detailed parameter settings for our algorithm.

\begin{figure}[!htp]
	\begin{center}
		\includegraphics[width=\textwidth]{supp_restaurant}\\
		\includegraphics[width=\textwidth]{supp_mall}\\
		\caption{More results of FB separation in \emph{Restaurant} and \emph{ShoppingMall} videos. The leftmost images are original frames. The right four columns show the results from our algorithms with $p = 1, p = 0.2$, AltProj \cite{netrapalli2014non}, and IALM \cite{LinChenMa10}. The runtime of each algorithm is written in the title.}
		\label{fig:exp3}
	\end{center}
\end{figure}

\section{Proofs} \label{sec:proofs}

In this section we provide the proofs for our main theoretical results in Theorems~\ref{thm:initialization}--\ref{thm:convergence_partial} and Corollaries~\ref{cor:rpca}--\ref{cor:rpca_partial}.
\subsection{Proof of Theorem \ref{thm:initialization}} \label{proof:thm:initialization}

Let $\widebar{Y} := Y - \ensuremath{S_{\text{init}}}$. As $Y = \ensuremath{ M^{*} } + \ensuremath{ S^{*} }$, we have $\widebar{Y} - \ensuremath{ M^{*} } = \ensuremath{ S^{*} } - \ensuremath{S_{\text{init}}}$. We obtain $\widebar{Y} - \ensuremath{ M^{*} } \in \mathcal{S}_{2\alpha}$ because $\ensuremath{ S^{*} }, \ensuremath{S_{\text{init}}} \in \mathcal{S}_{\alpha}$.

We claim that $\infnorm{\widebar{Y} - \ensuremath{ M^{*} }} \leq 2\infnorm{\ensuremath{ M^{*} }}$. Denote the support of $\ensuremath{ S^{*} }, \ensuremath{S_{\text{init}}}$ by $\Omega^*$ and $\Omega$ respectively. Since $\widebar{Y} - \ensuremath{ M^{*} }$ is supported on $\Omega \cup \Omega^*$, to prove the claim it suffices to consider the following three cases.
\begin{itemize}
	\item For $(i,j) \in \Omega^*\cap \Omega$, due to rule of sparse estimation, we have $\entry{(\ensuremath{ S^{*} } - \ensuremath{S_{\text{init}}})}{i}{j} = 0$.
	\item For $(i,j) \in \Omega^* \setminus \Omega$, we must have $|\entry{\ensuremath{ S^{*} }}{i}{j}| \leq 2\infnorm{\ensuremath{ M^{*} }}$. Otherwise, we have $|\entry{Y}{i}{j}| = |\entry{(\ensuremath{ S^{*} } + \ensuremath{ M^{*} })}{i}{j}| > \infnorm{\ensuremath{ M^{*} }}$. So $|\entry{Y}{i}{j}|$ 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 $\entry{Y}{i}{j} \in \Omega$, which violates the prior condition $(i,j) \in \Omega^* \setminus \Omega$.
	\item For the last case $(i,j) \in \Omega \setminus \Omega^*$, since $\entry{(\ensuremath{S_{\text{init}}})}{i}{j} = \entry{\ensuremath{ M^{*} }}{i}{j}$, trivially we have $|\entry{(\ensuremath{S_{\text{init}}})}{i}{j}| \leq \infnorm{\ensuremath{ M^{*} }}$.
\end{itemize}

The following result, proved in  Section \ref{proof:lem:op_to_inf}, relates the operator norm of $\widebar{Y} - \ensuremath{ M^{*} }$ to its infinite norm.
\begin{lemma} \label{lem:op_to_inf}
	For any matrix $A \in \ensuremath{\mathbb{R}}^{d_1\times d_2}$ that belongs to $\Sparset{\alpha}$ given in \eqref{eq:S_alpha}, we have
	\[
	\opnorm{A} \leq   \alpha\sqrt{d_1 d_2}\infnorm{A}.
	\]
\end{lemma}

We thus obtain
\begin{equation} \label{eq:tmp_opnorm_bound}
\opnorm{\widebar{Y} - \ensuremath{ M^{*} }} \leq 2\alpha \sqrt{d_1d_2} \infnorm{\widebar{Y} - \ensuremath{ M^{*} }} \leq  4\alpha \sqrt{d_1d_2} \infnorm{\ensuremath{ M^{*} }} = 4\alpha \mu r \sigma_1^*.
\end{equation}
In the last step, we use the fact that $\ensuremath{ M^{*} }$ satisfies the $\mu$-incoherent condition, which leads to
\begin{equation} \label{eq:top_element}
\infnorm{\ensuremath{ M^{*} }} \leq \opnorm{\ensuremath{ M^{*} }}\twoinfnorm{\ensuremath{L}^*}\twoinfnorm{\ensuremath{R}^*} \leq \frac{\mu r}{\sqrt{d_1d_2}}\opnorm{\ensuremath{ M^{*} }}.
\end{equation}

We denote the $i$-th largest singular value of $\widebar{Y}$ by $\sigma_i$. By Weyl's theorem, we have $\abs{\sigma_i^* - \sigma_i} \leq \opnorm{\widebar{Y}-\ensuremath{ M^{*} }}$ for all $i \in [d_1\wedge d_2]$. Since $\sigma_{r+1}^* = 0$, we have $\sigma_{r+1} \leq \opnorm{\widebar{Y}-\ensuremath{ M^{*} }}$. Recall that $\ensuremath{U}_0\ensuremath{V}_0^{\top}$ is the best rank $r$ approximation of $\widebar{Y}$. Accordingly, we have
\begin{align*}
\opnorm{\ensuremath{U}_0\ensuremath{V}_0^{\top} - \ensuremath{ M^{*} }} & \leq \opnorm{\ensuremath{U}_0\ensuremath{V}_0^{\top} - \widebar{Y}} + \opnorm{ \widebar{Y} - \ensuremath{ M^{*} }} \\
& = \sigma_{r+1} + \opnorm{ \widebar{Y} - \ensuremath{ M^{*} }} \leq 2\opnorm{\widebar{Y} - \ensuremath{ M^{*} }} \leq 8\alpha\mu r \sigma_1^*.
\end{align*}

Under condition $\alpha \mu r \leq \frac{1}{16\kappa}$, we obtain $\opnorm{\ensuremath{U}_0\ensuremath{V}_0^{\top} - \ensuremath{ M^{*} }} \leq \frac{1}{2}\sigma_r^*$. Applying Lemma 5.14 in \cite{tu2015low} (we provide it as Lemma \ref{lem:tu_lemma} for the sake of completeness), we obtain
\[
d^2(\ensuremath{U}_0, \ensuremath{V}_0; \ensuremath{U^*}, \ensuremath{V^*}) \leq \frac{2}{\sqrt{2} - 1}\frac{\frobnorm{\ensuremath{U}_0\ensuremath{V}_0^{\top} - \ensuremath{ M^{*} }}^2}{\sigma_r^*} \leq \frac{10r\opnorm{\ensuremath{U}_0\ensuremath{V}_0^{\top} - \ensuremath{ M^{*} }}^2}{\sigma_r^*}.
\]
Plugging the upper bound of $\opnorm{\ensuremath{U}_0\ensuremath{V}_0^{\top} - \ensuremath{ M^{*} }}$ into the above inequality completes the proof.

\subsection{Proof of Theorem \ref{thm:convergence}} \label{proof:thm:convergence}
We essentially follow the general framework developed in \cite{chen2015fast} for analyzing the behaviors of gradient descent in factorized low-rank optimization. But it is worth to note that \cite{chen2015fast} only studies the symmetric and positive semidefinite setting, while we avoid such constraint on $\ensuremath{ M^{*} }$. The techniques for analyzing general asymmetric matrix in factorized space is inspired by the recent work \cite{tu2015low} on solving low-rank matrix equations. In our setting, the technical challenge is to verify the local descent condition of the loss function \eqref{eq:opt}, which not only has a bilinear dependence on $\ensuremath{U}$ and $\ensuremath{V}$, but also involves our sparse estimator \eqref{eq:sparse_estimator}.

We begin with some notations. Define the equivalent set of optimal solution as
\begin{equation} \label{eq:Eq_set}
\ensuremath{\mathcal{E}}(\ensuremath{ M^{*} }) := \left\{(A,B) \in \ensuremath{\mathbb{R}}^{d_1 \times r} \times \ensuremath{\mathbb{R}}^{d_2 \times r} ~\big|~ A = \ensuremath{L}^*\ensuremath{\Sigma}^{*1/2}\ensuremath{Q},  B = \ensuremath{R}^*\ensuremath{\Sigma}^{*1/2}\ensuremath{Q}, ~\text{where}~\ensuremath{Q} \in \RotateSet{r} \right\}.
\end{equation}

Given $(\ensuremath{U}_0, \ensuremath{V}_0) \in \Neighbor{c_2\sqrt{\sigma^*_r/\kappa}}$, by \eqref{eq:error_relation}, we have $\opnorm{\ensuremath{U}_0\ensuremath{V}_0^{\top} - \ensuremath{ M^{*} }} \leq \frac{1}{2}\sigma_r^*$ when $c_2$ is sufficiently small. By Weyl's theorem We thus have \[ \sqrt{\sigma_1^*/2}  \leq \opnorm{\ensuremath{U}_0} \leq \sqrt{3\sigma_1^*/2}, ~\text{and}~  \sqrt{\sigma_1^*/2} \leq \opnorm{\ensuremath{V}_0} \leq \sqrt{3\sigma_1^*/2}.
\]
As a result, for $\ensuremath{\mathcal{U}}, \ensuremath{\mathcal{V}}$ constructed according to \eqref{eq:set}, we have
\begin{equation} \label{eq:uvcondition}
\ensuremath{\mathcal{E}}(\ensuremath{ M^{*} }) \subseteq \ensuremath{\mathcal{U}} \times \ensuremath{\mathcal{V}}, ~\text{and}~  \ensuremath{\mathcal{U}} \subseteq \widebar{\ensuremath{\mathcal{U}}},  \ensuremath{\mathcal{V}} \subseteq \widebar{\ensuremath{\mathcal{V}}},
\end{equation}
where
\begin{equation*}
\widebar{\ensuremath{\mathcal{U}}} := \left\{A \in \ensuremath{\mathbb{R}}^{d_1 \times r} ~\big|~ \twoinfnorm{A} \leq \sqrt{\frac{3\mu r\sigma_1^*}{d_1}}\right\}, ~
\widebar{\ensuremath{\mathcal{V}}} := \left\{A \in \ensuremath{\mathbb{R}}^{d_2 \times r}~\big|~ \twoinfnorm{A} \leq \sqrt{\frac{3\mu r\sigma_1^*}{d_2}}\right\}.
\end{equation*}

We let
\begin{equation} \label{eq:G}
\mathcal{G}(\ensuremath{U}, \ensuremath{V}) : = \frac{1}{8}\frobnorm{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}}^2.
\end{equation}
For $\ensuremath{\mathcal{L}}(\ensuremath{U}, \ensuremath{V}; \ensuremath{S})$, we denote the gradient with respect to $M$ by $\nabla_{M}\ensuremath{\mathcal{L}}(\ensuremath{U},\ensuremath{V};\ensuremath{S})$, i.e. $\nabla_{M}\ensuremath{\mathcal{L}}(\ensuremath{U},\ensuremath{V};\ensuremath{S}) = \ensuremath{U}\ensuremath{V}^{\top} + S - Y$.

The  local descent property is implied by combining the following two results, which are proved in Section \ref{proof:lem:local_descent_L} and \ref{proof:lem:local_descent_g} respectively.
\begin{lemma}[Local descent property of $\ensuremath{\mathcal{L}}$] \label{lem:local_descent_L}
	Suppose $\ensuremath{\mathcal{U}}, \ensuremath{\mathcal{V}}$ satisfy \eqref{eq:uvcondition}. For any $(U,V) \in (\ensuremath{\mathcal{U}} \times \ensuremath{\mathcal{V}}) \cap \mathbb{B}_2(\sqrt{\sigma_1^*})$, we let $S = \UnifSparse{Y - \ensuremath{U}\ensuremath{V}^{\top}}{\gamma \alpha}$, where we choose $\gamma = 2$. Then we have that for $(\ensuremath{U_{\pi^*}}, \ensuremath{V_{\pi^*}}) \in \arg\!\min_{(A, B) \in \ensuremath{\mathcal{E}}(\ensuremath{ M^{*} })} \frobnorm{\ensuremath{U} - A}^2 + \frobnorm{\ensuremath{V} - B}^2$ and  $\beta > 0$,
	\begin{align}\label{eq:local_descent}
	& \trinprod{\nabla_{M} \ensuremath{\mathcal{L}}(\ensuremath{U}, \ensuremath{V}; S)}{\ensuremath{U} \ensuremath{V}^{\top} - \ensuremath{U_{\pi^*}} \ensuremath{V_{\pi^*}}^{\top} + \ensuremath{\Delta_{U}} \ensuremath{\Delta_{V}}^{\top}}  \geq \frobnorm{\ensuremath{U} \ensuremath{V}^{\top} - \ensuremath{U_{\pi^*}} \ensuremath{V_{\pi^*}}^{\top}}^2 -  \nu\sigma_1^*\delta - 3\sqrt{\sigma_1^*\delta^3}.
	\end{align}
	Here $\ensuremath{\Delta_{U}} := \ensuremath{U} - \ensuremath{U_{\pi^*}}$, $\ensuremath{\Delta_{V}} := \ensuremath{V} - \ensuremath{V_{\pi^*}}$, $\delta := \frobnorm{\ensuremath{\Delta_{U}}}^2 + \frobnorm{\ensuremath{\Delta_{V}}}^2$, and $\nu := 9(\beta+6)\alpha \mu r + 5\beta^{-1}$.
\end{lemma}

\vskip 0.1in
\begin{lemma}[Local descent property of $\mathcal{G}$] \label{lem:local_descent_g}
	For any $(\ensuremath{U},\ensuremath{V}) \in \mathbb{B}_2(\sqrt{\sigma_r^*})$ and
	\[
	(\ensuremath{U_{\pi^*}}, \ensuremath{V_{\pi^*}}) \in \arg\!\min_{(A, B) \in \ensuremath{\mathcal{E}}(\ensuremath{ M^{*} })} \frobnorm{\ensuremath{U} - A}^2 + \frobnorm{\ensuremath{V} - B}^2,
	\]
	we have
	\begin{align*}
	& \trinprod{\nabla_{\ensuremath{U}} \mathcal{G}(\ensuremath{U}, \ensuremath{V})}{U -  \ensuremath{U_{\pi^*}}} + \trinprod{\nabla_{\ensuremath{V}} \mathcal{G}(\ensuremath{U}, \ensuremath{V})}{V -  \ensuremath{V_{\pi^*}}} \\
	& \geq \frac{1}{8}\frobnorm{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}}^2 + \frac{1}{8}\sigma_r^*\delta -  \sqrt{\frac{\sigma_1^*\delta^3}{2}} -\frac{1}{2}\frobnorm{\ensuremath{U} \ensuremath{V}^{\top} - \ensuremath{U_{\pi^*}}\ensuremath{V_{\pi^*}}^{\top}}^2,
	\end{align*}
	where $\delta$ is defined according to Lemma \ref{lem:local_descent_L}.
\end{lemma}

As another key ingredient, we establish the following smoothness condition, proved in Section \ref{proof:lem:smoothness}, which indicates that the Frobenius norm of gradient decreases as $(\ensuremath{U}, \ensuremath{V})$ approaches the optimal manifold.

\begin{lemma}[Smoothness] \label{lem:smoothness}
	For any $(\ensuremath{U}, \ensuremath{V}) \in \mathbb{B}_2(\sqrt{\sigma_1^*})$, we let $S = \UnifSparse{Y - \ensuremath{U}\ensuremath{V}^{\top}}{\gamma\alpha}$, where we choose $\gamma = 2$. We have that
	\begin{equation} \label{eq:smooth_L}
	\frobnorm{\nabla_{M}\ensuremath{\mathcal{L}}(\ensuremath{U},\ensuremath{V};\ensuremath{S})}^2 \leq  6\frobnorm{\ensuremath{U}\ensuremath{V}^{\top} - \ensuremath{ M^{*} }}^2,
	\end{equation}
	and
	\begin{equation} \label{eq:smooth_G}
	\frobnorm{\nabla_{\ensuremath{U}} \mathcal{G}(\ensuremath{U}, \ensuremath{V})}^2 + \frobnorm{\nabla_{\ensuremath{V}} \mathcal{G}(\ensuremath{U}, \ensuremath{V})}^2 \leq 2\sigma_1^*\frobnorm{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}}^2.
	\end{equation}
\end{lemma}
With the above results in hand, we are ready to prove Theorem \ref{thm:convergence}.
\begin{proof}[Proof of Theorem \ref{thm:convergence}.]
	We use shorthands
	\[
	\delta_t := d^2(U_t, V_t; \ensuremath{U^*}, \ensuremath{V^*}), ~\ensuremath{\mathcal{L}}_t := \ensuremath{\mathcal{L}}(\ensuremath{U}_t,\ensuremath{V}_t; \ensuremath{S}_t),~ \text{and}~~ \mathcal{G}_t := \mathcal{G}(\ensuremath{U}_t,\ensuremath{V}_t).
	\]
	For $(\ensuremath{U}_t, \ensuremath{V}_t)$, let $(\ensuremath{U_{\pi^*}}^{t}, \ensuremath{V_{\pi^*}}^{t}) := \arg\!\min_{(A, B) \in \ensuremath{\mathcal{E}}(\ensuremath{ M^{*} })} \frobnorm{\ensuremath{U}_t - A}^2 + \frobnorm{\ensuremath{V}_t - B}^2$. Define $\Delta_{\ensuremath{U}}^t := \ensuremath{U}_t - \ensuremath{U_{\pi^*}}^{t}$, $\Delta_{\ensuremath{V}}^t := \ensuremath{V}_t - \ensuremath{V_{\pi^*}}^{t}$.

	We prove Theorem \ref{thm:convergence} by induction. It is sufficient to consider one step of the iteration. For any $t \geq 0$, under the induction hypothesis $(\ensuremath{U}_t, \ensuremath{V}_t) \in \Neighbor{c_2\sqrt{\sigma^*_r/\kappa}}$. We find that
	\begin{align}
	\delta_{t+1} & \leq \frobnorm{U_{t+1} - \ensuremath{U_{\pi^*}}^t}^2 + \frobnorm{V_{t+1} - \ensuremath{V_{\pi^*}}^t}^2 \notag\\
	& \leq \frobnorm{U_t - \eta \nabla_{\ensuremath{U}}\ensuremath{\mathcal{L}}_t - \eta\nabla_{\ensuremath{U}}\mathcal{G}_t - \ensuremath{U_{\pi^*}}^t}^2  + \frobnorm{V_t - \eta \nabla_{\ensuremath{V}}\ensuremath{\mathcal{L}}_t - \eta\nabla_{\ensuremath{V}}\mathcal{G}_t - \ensuremath{V_{\pi^*}}^t}^2 \notag \\
	& \leq \delta_t - 2\eta\underbrace{\trinprod{\nabla_{\ensuremath{U}}\ensuremath{\mathcal{L}}_t + \nabla_{\ensuremath{U}}\mathcal{G}_t}{\ensuremath{U}_t - \ensuremath{U_{\pi^*}}^t}}_{W_1} - 2\eta\underbrace{\trinprod{\nabla_{\ensuremath{V}}\ensuremath{\mathcal{L}}_t + \nabla_{\ensuremath{V}}\mathcal{G}_t}{\ensuremath{V}_t - \ensuremath{V_{\pi^*}}^t}}_{W_2} \notag\\
	& ~~~~~~~ + \eta^2 \underbrace{\frobnorm{\nabla_{\ensuremath{U}}\ensuremath{\mathcal{L}}_t + \nabla_{\ensuremath{U}}\mathcal{G}_t}^2}_{W_3} + \eta^2\underbrace{\frobnorm{\nabla_{\ensuremath{V}}\ensuremath{\mathcal{L}}_t + \nabla_{\ensuremath{V}}\mathcal{G}_t}^2}_{W_4}, \label{eq:delta_relation}
	\end{align}
	where the second step follows from the non-expansion property of projection onto $\ensuremath{\mathcal{U}}, \ensuremath{\mathcal{V}}$, which is implied by $\ensuremath{\mathcal{E}}(\ensuremath{ M^{*} }) \subseteq \ensuremath{\mathcal{U}} \times \ensuremath{\mathcal{V}}$ shown in \eqref{eq:uvcondition}. Since $\nabla_{\ensuremath{U}}\ensuremath{\mathcal{L}}_t = \left[\nabla_{M}\ensuremath{\mathcal{L}}_t\right] \ensuremath{V}$ and $\nabla_{\ensuremath{V}}\ensuremath{\mathcal{L}}_t = \left[\nabla_{M}\ensuremath{\mathcal{L}}_t\right]^{\top} \ensuremath{U}$, we have
	\[
	\trinprod{\nabla_{\ensuremath{U}}\ensuremath{\mathcal{L}}_t}{\ensuremath{U}_t - \ensuremath{U_{\pi^*}}^t} + \trinprod{\nabla_{\ensuremath{V}}\ensuremath{\mathcal{L}}_t}{\ensuremath{V}_t - \ensuremath{V_{\pi^*}}^t} = \trinprod{\nabla_{M} \ensuremath{\mathcal{L}}_t}{\ensuremath{U}_t \ensuremath{V}_t^{\top} - \ensuremath{U_{\pi^*}}^t \ensuremath{V_{\pi^*}}^{t\top} + \ensuremath{\Delta_{U}}^t \ensuremath{\Delta_{V}}^{t\top}}.
	\]

	Combining Lemma \ref{lem:local_descent_L} and \ref{lem:local_descent_g}, under condition $\delta_t < \sigma_r^*$, we have that
	\[
	W_1 + W_2 \geq \frac{1}{2}\frobnorm{\ensuremath{U}_t \ensuremath{V}_t^{\top} - \ensuremath{ M^{*} }}^2 +  \frac{1}{8}\frobnorm{\ensuremath{U}_t^{\top}\ensuremath{U}_t - \ensuremath{V}_t^{\top}\ensuremath{V}_t}^2 + \frac{1}{8}\sigma_r^*\delta_t -  \nu\sigma_1^*\delta_t - 4\sqrt{\sigma_1^*\delta_t^3}.
	\]

	On the other hand, we have
	\begin{align*}
	W_3 + W_4 & \leq 2\frobnorm{\nabla_{\ensuremath{U}}\ensuremath{\mathcal{L}}_t}^2 + 2\frobnorm{\nabla_{\ensuremath{U}}\mathcal{G}_t}^2 +  2\frobnorm{\nabla_{\ensuremath{V}}\ensuremath{\mathcal{L}}_t}^2 + 2\frobnorm{\nabla_{\ensuremath{V}}\mathcal{G}_t}^2 \\
	& \leq 2(\opnorm{\ensuremath{U}_t}^2 + \opnorm{\ensuremath{V}_t}^2)\frobnorm{\nabla_M \ensuremath{\mathcal{L}}_t}^2 + 2\frobnorm{\nabla_{\ensuremath{U}}\mathcal{G}_t}^2 + 2\frobnorm{\nabla_{\ensuremath{V}}\mathcal{G}_t}^2 \\
	& \leq 36\sigma_1^*\frobnorm{\ensuremath{U}_t\ensuremath{V}_t^{\top} - \ensuremath{ M^{*} }}^2 +  4\sigma_1^*\frobnorm{\ensuremath{U}_t^{\top}\ensuremath{U}_t - \ensuremath{V}_t^{\top}\ensuremath{V}_t}^2,
	\end{align*}
	where the last step is implied by Lemma \ref{lem:smoothness} and the assumption $(\ensuremath{U}_t, \ensuremath{V}_t) \in \Neighbor{c_2\sqrt{\sigma^*_r/\kappa}}$ that leads to $\opnorm{\ensuremath{U}_t} \leq \sqrt{3\sigma_1^*/2}, \opnorm{\ensuremath{V}_t} \leq \sqrt{3\sigma_1^*/2}$.

	By the assumption $\eta = c/\sigma_1^*$ for any constant $c \leq 1/36$, we thus have
	\begin{align*}
	-2\eta(W_1 + W_2) + \eta^2(W_3 + W_4) \leq  - \frac{1}{4}\eta\sigma_r^*\delta_t + 2\eta  \nu\sigma_1^*\delta_t +  8\eta\sqrt{\sigma_1^*\delta_t^3}.
	\end{align*}
	In Lemma \ref{lem:local_descent_L}, choosing $\beta = 320\kappa$ and assuming $\alpha \lesssim 1/(\kappa^2\mu r)$, we can have $\nu \leq 1/(32\kappa)$. Assuming $\delta_t \lesssim \sigma_r^*/\kappa$ leads to $14\sqrt{\sigma_1^*\delta_t^3} \leq \frac{1}{16}\sigma_r^*\delta_t$. We thus obtain
	\begin{equation} \label{eq:one_step}
	\delta_{t+1} \leq \left(1 - \frac{\eta \sigma_r^*}{8}\right)\delta_t.
	\end{equation}
	Under initial condition $\delta_0 \lesssim \sigma_r^*/\kappa$, we obtain that such condition holds for all $t$ since estimation error decays geometrically after each iteration. Then applying \eqref{eq:one_step} for all iterations, we conclude that for all $t = 0,1,\ldots$,
	\[
	\delta_t \leq \left(1 - \frac{\eta \sigma_r^*}{8}\right)^t\delta_0.
	\]
\end{proof}

\subsection{Proof of Corollary \ref{cor:rpca}} \label{proof:cor:rpca}

We need $\alpha \lesssim \frac{1}{\kappa^2 \mu r}$ due to the condition of Theorem \ref{thm:convergence}. In order to ensure the linear convergence happens, it suffices to let the initial error shown in Theorem \ref{thm:initialization} be less than the corresponding condition in Theorem \ref{thm:convergence}. Accordingly, we need
\[
28 \sqrt{\kappa} \alpha \mu r\sqrt{r} \sqrt{\sigma_1^*} \lesssim \sqrt{\sigma^*_r/\kappa},
\]
which leads to $\alpha \lesssim \frac{1}{\mu \sqrt{r\kappa}^3}$.

Using the conclusion that gradient descent has linear convergence, choosing $T = \ensuremath{\mathcal{O}}(\kappa \log(1/\varepsilon))$, we have
\[
d^2(\ensuremath{U}_{T}, \ensuremath{V}_{T}; \ensuremath{U^*}, \ensuremath{V^*}) \leq \varepsilon^2 d^2(\ensuremath{U}_{0}, \ensuremath{V}_{0}; \ensuremath{U^*}, \ensuremath{V^*}) \lesssim \varepsilon^2 \frac{\sigma_r^*}{\kappa}.
\]
Finally, applying the relationship between $d(\ensuremath{U}_{T}, \ensuremath{V}_{T}; \ensuremath{U^*}, \ensuremath{V^*})$ and $\frobnorm{\ensuremath{U}_{T}\ensuremath{V}_{T}^{\top} - \ensuremath{ M^{*} }}$ shown in \eqref{eq:error_relation}, we complete the proof.

\subsection{Proof of Theorem \ref{thm:initial_partial}} \label{proof:thm:initial_partial}

Let $\widebar{Y} := \frac{1}{p}(Y - \ensuremath{S_{\text{init}}})$. Similar to the proof of Theorem \ref{thm:initialization}, we first establish an upper bound on $\opnorm{\widebar{Y} - \ensuremath{ M^{*} }}$. We have that
\begin{equation} \label{eq:op_bound}
\opnorm{\widebar{Y} - \ensuremath{ M^{*} }} \leq \opnorm{\widebar{Y} - \frac{1}{p}\ensuremath{\Pi}_{\Phi}\ensuremath{ M^{*} }} + \opnorm{\frac{1}{p}\ensuremath{\Pi}_{\Phi}(\ensuremath{ M^{*} }) - \ensuremath{ M^{*} }}.
\end{equation}

For the first term, we have $\widebar{Y} - \frac{1}{p}\ensuremath{\Pi}_{\Phi}\ensuremath{ M^{*} } = \frac{1}{p}(\ensuremath{\Pi}_{\Phi}(\ensuremath{ S^{*} }) - \ensuremath{S_{\text{init}}})$ because $Y = \ensuremath{\Pi}_{\Phi}(\ensuremath{ M^{*} } + \ensuremath{ S^{*} })$.  Lemma \ref{lem:size_concentration} shows that under condition $p \gtrsim \frac{\log d}{\alpha (d_1 \wedge d_2)}$, there are at most $\frac{3}{2}p\alpha$-fraction nonzero entries in each row and column of $\ensuremath{\Pi}_{\Phi}(\ensuremath{ S^{*} })$ with high probability. Since $\ensuremath{S_{\text{init}}} \in \Sparset{2p\alpha}$, we have
\begin{equation} \label{eq:property_1}
\ensuremath{\Pi}_{\Phi}(\ensuremath{ S^{*} }) - \ensuremath{S_{\text{init}}} \in \Sparset{4p\alpha}.
\end{equation}
In addition, we prove below that
\begin{equation} \label{eq:property_2}
\infnorm{\ensuremath{\Pi}_{\Phi}(\ensuremath{ S^{*} }) - \ensuremath{S_{\text{init}}}} \leq 2\infnorm{\ensuremath{ M^{*} }}.
\end{equation}
Denote the support of $\ensuremath{\Pi}_{\Phi}(\ensuremath{ S^{*} })$ and $\ensuremath{S_{\text{init}}}$ by $\Omega_o^*$ and $\Omega$. For $(i,j) \in \Omega_o^* \cap \Omega$ and $(i,j) \in \Omega \setminus \Omega_o^*$, we have $\entry{(\ensuremath{\Pi}_{\Phi}(\ensuremath{ S^{*} }) - \ensuremath{S_{\text{init}}})}{i}{j} = 0$ and $\entry{(\ensuremath{\Pi}_{\Phi}(\ensuremath{ S^{*} }) - \ensuremath{S_{\text{init}}})}{i}{j} = -\entry{\ensuremath{ M^{*} }}{i}{j}$, respectively. To prove the claim, it remains to show that for $(i,j) \in \Omega_o^*\setminus \Omega$, $\abs{\entry{\ensuremath{ S^{*} }}{i}{j}} < 2\infnorm{\ensuremath{ M^{*} }}$. If this is not true, then we must have $\abs{\entry{Y}{i}{j}} > \infnorm{\ensuremath{ M^{*} }}$. Accordingly, $\abs{\entry{Y}{i}{j}}$ is larger than the magnitude of any uncorrupted entries in its row and column. Note that on the support $\Phi$, there are at most $\frac{3}{2}p\alpha$ corruptions per row and column, we have $(i,j) \in \Omega$, which violates our prior condition $(i,j) \in \Omega_o^*\setminus \Omega$.

Using these two properties \eqref{eq:property_1}, \eqref{eq:property_2} and applying Lemma \ref{lem:op_to_inf}, we have
\begin{equation} \label{eq:opnorm_bound_1}
\opnorm{\widebar{Y}- \frac{1}{p}\ensuremath{\Pi}_{\Phi}\ensuremath{ M^{*} }} \leq 4\alpha\sqrt{d_1d_2}\infnorm{\ensuremath{\Pi}_{\Phi}(\ensuremath{ S^{*} }) - \ensuremath{S_{\text{init}}}} \leq 8\alpha\sqrt{d_1d_2}\infnorm{\ensuremath{ M^{*} }} \leq 8\alpha \mu r \sigma_1^*,
\end{equation}
where the last step follow from \eqref{eq:top_element}.

For the second term in \eqref{eq:op_bound}, we use the following lemma proved in \cite{DBLP:journals/tit/Chen15}.
\begin{lemma}[Lemma 2 in \cite{DBLP:journals/tit/Chen15}] \label{lem:op_sampling}
	Suppose $A \in \ensuremath{\mathbb{R}}^{d_1 \times d_2}$ is a fixed matrix. We let $d := \max\{d_1, d_2\}$. There exists a constant $c$ such that with probability at least $1 - \ensuremath{\mathcal{O}}(d^{-1})$,
	\[
	\opnorm{\frac{1}{p}\ensuremath{\Pi}_{\Phi}(A) - A} \leq c\left(\frac{\log d}{p}\infnorm{A} + \sqrt{\frac{\log d}{p}} \max\left\{\twoinfnorm{A}, \twoinfnorm{A^{\top}}\right\} \right).
	\]
\end{lemma}
Given the SVD $\ensuremath{ M^{*} } = \ensuremath{L}^*\ensuremath{\Sigma} \ensuremath{R}^{*\top}$, for any $i \in [d_1]$, we have
\[
\twonorm{\row{\ensuremath{ M^{*} }}{i}} = \twonorm{\row{\ensuremath{L}^*}{i}\ensuremath{\Sigma} \ensuremath{R}^{*\top}} \leq \sigma_1^*\twonorm{\row{\ensuremath{L}^*}{i}} \leq \sigma_1^*\sqrt{\frac{\mu r}{d_1}}.
\]
We can bound $\twoinfnorm{M^{*\top}}$ similarly. Lemma \ref{lem:op_sampling} leads to
\begin{equation} \label{eq:opnorm_bound_2}
\opnorm{\frac{1}{p}\ensuremath{\Pi}_{\Phi}(\ensuremath{ M^{*} }) - \ensuremath{ M^{*} }} \leq c'\frac{\log d}{p}\frac{\mu r}{\sqrt{d_1d_2}}\sigma_1^* + c'\sqrt{\frac{\log d}{p}}\sqrt{\frac{\mu r}{d_1 \wedge d_2}}\sigma_1^* \leq c' \epsilon \sigma_1^*/\sqrt{r}
\end{equation}
under condition $p \geq \frac{4\mu r^2\log d}{\epsilon^2(d_1 \wedge d_2)}$.

Putting \eqref{eq:opnorm_bound_1} and \eqref{eq:opnorm_bound_2} together, we obtain
\[
\opnorm{\widebar{Y} - \ensuremath{ M^{*} }} \leq 8\alpha\mu r\sigma_1^* + c'\epsilon\sigma_1^*/\sqrt{r}.
\]
Then using the fact that $\ensuremath{U}_0\ensuremath{V}_0^{\top}$ is the best rank $r$ approximation of $\widebar{Y}$ and applying Wely's theorem (see the proof of Theorem \ref{thm:initialization} for a detailed argument), we have
\begin{align*}
\opnorm{\ensuremath{U}_0\ensuremath{V}_0^{\top} - \ensuremath{ M^{*} }} & \leq \opnorm{\ensuremath{U}_0\ensuremath{V}_0^{\top} - \widebar{Y}} + \opnorm{\widebar{Y} - \ensuremath{ M^{*} }} \\
& \leq 2\opnorm{\widebar{Y} - \ensuremath{ M^{*} }} \leq 16\alpha\mu r\sigma_1^* + 2c'\epsilon\sigma_1^*/\sqrt{r}
\end{align*}

Under our assumptions,  we have $16\alpha\mu r\sigma_1^* + 2c'\epsilon\sigma_1^*/\sqrt{r} \leq \frac{1}{2}\sigma_r^*$. Accordingly, Lemma \ref{lem:tu_lemma} gives
\[
d^2(\ensuremath{U}_0, \ensuremath{V}_0; \ensuremath{U^*}, \ensuremath{V^*}) \leq \frac{2}{\sqrt{2} - 1}\frac{\frobnorm{\ensuremath{U}_0\ensuremath{V}_0^{\top} - \ensuremath{ M^{*} }}^2}{\sigma_r^*} \leq \frac{10r\opnorm{\ensuremath{U}_0\ensuremath{V}_0^{\top} - \ensuremath{ M^{*} }}^2}{\sigma_r^*}.
\]
We complete the proof by combining the above two inequalities.

\subsection{Proof of Theorem \ref{thm:convergence_partial}} \label{proof:thm:convergence_partial}
In this section, we turn to prove Theorem \ref{thm:convergence_partial}. Similar to the proof of Theorem \ref{thm:convergence}, we rely on establishing the local descent and smoothness conditions. Compared to the full observation setting, we replace $\ensuremath{\mathcal{L}}$ by $\widetilde{\ensuremath{\mathcal{L}}}$ given in \eqref{eq:loss_partial}, while the regularization term $\widetilde{\mathcal{G}}(\ensuremath{U},\ensuremath{V}) := \frac{1}{64}\frobnorm{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}}^2$ merely differs from $\mathcal{G}(\ensuremath{U},\ensuremath{V})$ given in \eqref{eq:G} by a constant factor. It is thus sufficient to analyze the properties of $\widetilde{\ensuremath{\mathcal{L}}}$.

Define $\ensuremath{\mathcal{E}}(\ensuremath{ M^{*} })$ according to \eqref{eq:Eq_set}. Under the initial condition, we still have
\begin{equation} \label{eq:uvcondition_partial}
\ensuremath{\mathcal{E}}(\ensuremath{ M^{*} }) \subseteq \ensuremath{\mathcal{U}} \times \ensuremath{\mathcal{V}}, ~\text{and}~  \ensuremath{\mathcal{U}} \subseteq \widebar{\ensuremath{\mathcal{U}}},  \ensuremath{\mathcal{V}} \subseteq \widebar{\ensuremath{\mathcal{V}}}.
\end{equation}

We prove the next two lemmas in Section \ref{proof:lem:local_descent_L_partial} and \ref{proof:lem:smoothness_partial} respectively. In both lemmas, for any $(\ensuremath{U}, \ensuremath{V}) \in \ensuremath{\mathcal{U}} \times \ensuremath{\mathcal{V}}$, we use shorthands
\[
(\ensuremath{U_{\pi^*}}, \ensuremath{V_{\pi^*}}) = \arg\!\min_{(A, B) \in \ensuremath{\mathcal{E}}(\ensuremath{ M^{*} })} \frobnorm{\ensuremath{U} - A}^2 + \frobnorm{\ensuremath{V} - B}^2,
\]
$\ensuremath{\Delta_{U}} := \ensuremath{U} - \ensuremath{U_{\pi^*}}$, $\ensuremath{\Delta_{V}} := \ensuremath{V} - \ensuremath{V_{\pi^*}}$, and $\delta := \frobnorm{\ensuremath{\Delta_{U}}}^2 + \frobnorm{\ensuremath{\Delta_{V}}}^2$. Recall that $d := \max\{d_1, d_2\}$.
\begin{lemma}[Local descent property of $\widetilde{\ensuremath{\mathcal{L}}}$] \label{lem:local_descent_L_partial}
	Suppose $\ensuremath{\mathcal{U}}, \ensuremath{\mathcal{V}}$ satisfy \eqref{eq:uvcondition_partial}. Suppose we let
	\[
	S = \UnifSparse{\ensuremath{\Pi}_{\Phi}\left(Y - \ensuremath{U}\ensuremath{V}^{\top}\right)}{\gamma p \alpha},
	\]
	where we choose $\gamma = 3$. For any $\beta > 0$ and $\epsilon \in (0, \frac{1}{4})$, we define $\nu := (14\beta + 81)\alpha \mu r + 26\sqrt{\epsilon} + 18\beta^{-1}$. There exist constants $\{c_i\}_{i=1}^2$ such that if
	\begin{equation} \label{eq:condition_p}
	p \geq c_1\left(\frac{\mu ^2 r^2}{\epsilon^2} + \frac{1}{\alpha} + 1\right)\frac{\log d}{d_1 \wedge d_2},
	\end{equation}
	then with probability at least $1 - c_2d^{-1}$,
	\begin{equation}\label{eq:local_descent_partial}
	\trinprod{\nabla_{M} \widetilde{\ensuremath{\mathcal{L}}}(\ensuremath{U}, \ensuremath{V}; S)}{\ensuremath{U} \ensuremath{V}^{\top} - \ensuremath{U_{\pi^*}} \ensuremath{V_{\pi^*}}^{\top} + \ensuremath{\Delta_{U}} \ensuremath{\Delta_{V}}^{\top}}
	\geq \frac{3}{16}\frobnorm{\ensuremath{U} \ensuremath{V}^{\top} - \ensuremath{U_{\pi^*}} \ensuremath{V_{\pi^*}}^{\top}}^2 - \nu \sigma_1^*\delta - 10\sqrt{\sigma_1^*\delta^3} - 2\delta^2
	\end{equation}
	for all $(\ensuremath{U}, \ensuremath{V}) \in \left(\ensuremath{\mathcal{U}}\times\ensuremath{\mathcal{V}}\right) \cap \Neighbor{\sqrt{\sigma_1^*}}$.
\end{lemma}

\begin{lemma}[Smoothness of $\widetilde{\ensuremath{\mathcal{L}}}$] \label{lem:smoothness_partial}
	Suppose $\ensuremath{\mathcal{U}}, \ensuremath{\mathcal{V}}$ satisfy \eqref{eq:uvcondition_partial}. Suppose we let $S = \UnifSparse{\ensuremath{\Pi}_{\Phi}\left(Y - \ensuremath{U}\ensuremath{V}^{\top}\right)}{\gamma\alpha p}$ for $\gamma = 3$. There exist constants $\{c_i\}_{i=1}^3$ such that for any $\epsilon \in (0,\frac{1}{4})$, when $p$ satisfies condition \eqref{eq:condition_p}, with probability at least $1 - c_2d^{-1}$, we have that for all $(\ensuremath{U}, \ensuremath{V}) \in (\ensuremath{\mathcal{U}} \times \ensuremath{\mathcal{V}}) \cap \mathbb{B}_2(\sqrt{\sigma_1^*})$,
	\begin{equation} \label{eq:smooth_partial}
	\frobnorm{\nabla_{\ensuremath{U}}\widetilde{\ensuremath{\mathcal{L}}}(\ensuremath{U},\ensuremath{V};\ensuremath{S})}^2 + \frobnorm{\nabla_{\ensuremath{V}}\widetilde{\ensuremath{\mathcal{L}}}(\ensuremath{U},\ensuremath{V};\ensuremath{S})}^2 \leq c_3\left[\mu r \sigma_1^*\frobnorm{\ensuremath{U} \ensuremath{V}^{\top} - \ensuremath{U_{\pi^*}} \ensuremath{V_{\pi^*}}^{\top}}^2 + \mu r\sigma_1^*\delta(\delta + \epsilon\sigma_1^*)\right].
	\end{equation}
\end{lemma}

In the remainder of this section, we condition on the events in Lemma \ref{lem:local_descent_L_partial} and \ref{lem:smoothness_partial}. Now we are ready to prove Theorem \ref{thm:convergence_partial}.
\begin{proof}[Proof of Theorem \ref{thm:convergence_partial}]
	We essentially follow the process for proving Theorem \ref{thm:convergence}. Let the following shorthands be defined in the same fashion: $\delta_t$, $(\ensuremath{U_{\pi^*}}^{t}, \ensuremath{V_{\pi^*}}^{t})$, $(\Delta_{\ensuremath{U}}^t, \Delta_{\ensuremath{V}}^t)$, $\widetilde{\ensuremath{\mathcal{L}}}_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 \ref{thm:convergence}, and is thus omitted. For any $t \geq 0$, similar to \eqref{eq:delta_relation} we have that
	\begin{align*}
	\delta_{t+1}
	& \leq \delta_t - 2\eta\underbrace{\trinprod{\nabla_{\ensuremath{U}}\widetilde{\ensuremath{\mathcal{L}}}_t + \nabla_{\ensuremath{U}}\widetilde{\mathcal{G}}_t}{\ensuremath{U}_t - \ensuremath{U_{\pi^*}}^t}}_{W_1} - 2\eta\underbrace{\trinprod{\nabla_{\ensuremath{V}}\widetilde{\ensuremath{\mathcal{L}}}_t + \nabla_{\ensuremath{V}}\widetilde{\mathcal{G}}_t}{\ensuremath{V}_t - \ensuremath{V_{\pi^*}}^t}}_{W_2} \\
	& ~~~~~~~ + \eta^2 \underbrace{\frobnorm{\nabla_{\ensuremath{U}}\widetilde{\ensuremath{\mathcal{L}}}_t + \nabla_{\ensuremath{U}}\widetilde{\mathcal{G}}_t}^2}_{W_3} + \eta^2\underbrace{\frobnorm{\nabla_{\ensuremath{V}}\widetilde{\ensuremath{\mathcal{L}}}_t + \nabla_{\ensuremath{V}}\widetilde{\mathcal{G}}_t}^2}_{W_4}.
	\end{align*}
	We also have
	\[
	\trinprod{\nabla_{\ensuremath{U}}\widetilde{\ensuremath{\mathcal{L}}}_t}{\ensuremath{U}_t - \ensuremath{U_{\pi^*}}^t} + \trinprod{\nabla_{\ensuremath{V}}\widetilde{\ensuremath{\mathcal{L}}}_t}{\ensuremath{V}_t - \ensuremath{V_{\pi^*}}^t} = \trinprod{\nabla_{M} \widetilde{\ensuremath{\mathcal{L}}}_t}{\ensuremath{U}_t \ensuremath{V}_t^{\top} - \ensuremath{U_{\pi^*}}^t \ensuremath{V_{\pi^*}}^{t\top} + \ensuremath{\Delta_{U}}^t \ensuremath{\Delta_{V}}^{t\top}},
	\]
	which can be lower bounded by Lemma \ref{lem:local_descent_L_partial}. Note that $\widetilde{\mathcal{G}}$ differs from $\mathcal{G}$ by a constant, we can still leverage Lemma \ref{lem:local_descent_g}. Hence, we obtain that
	\[
	W_1 + W_2 \geq \frac{1}{8}\frobnorm{\ensuremath{U}_t \ensuremath{V}_t^{\top} - \ensuremath{ M^{*} }}^2 +  \frac{1}{64}\frobnorm{\ensuremath{U}_t^{\top}\ensuremath{U}_t - \ensuremath{V}_t^{\top}\ensuremath{V}_t}^2 + \frac{1}{64}\sigma_r^*\delta_t -  \nu\sigma_1^*\delta_t - 11\sqrt{\sigma_1^*\delta_t^3} - 2\delta_t^2.
	\]

	On the other hand, we have
	\begin{align*}
	W_3 + W_4 & \leq 2\frobnorm{\nabla_{\ensuremath{U}}\widetilde{\ensuremath{\mathcal{L}}}_t}^2 + 2\frobnorm{\nabla_{\ensuremath{U}}\widetilde{\mathcal{G}}_t}^2 +  2\frobnorm{\nabla_{\ensuremath{V}}\widetilde{\ensuremath{\mathcal{L}}}_t}^2 + 2\frobnorm{\nabla_{\ensuremath{V}}\widetilde{\mathcal{G}}_t}^2 \\
	& \leq c\left[\mu r \sigma_1^*\frobnorm{\ensuremath{U}_t \ensuremath{V}_t^{\top} - \ensuremath{ M^{*} }}^2 + \mu r\sigma_1^*\delta_t(\delta_t + \epsilon\sigma_1^*) + \sigma_1^*\frobnorm{\ensuremath{U}_t^{\top}\ensuremath{U}_t - \ensuremath{V}_t^{\top}\ensuremath{V}_t}^2\right],
	\end{align*}
	where $c$ is a constant, and the last step is implied by Lemma \ref{lem:smoothness} and Lemma \ref{lem:smoothness_partial}.

	By the assumption $\eta = c'/[\mu r\sigma_1^*]$ for sufficiently small constant $c'$, we thus have
	\begin{align*}
	-2\eta(W_1 + W_2) + \eta^2(W_3 + W_4) \leq  - \frac{1}{32}\eta\sigma_r^*\delta_t + 2\eta  \nu\sigma_1^*\delta_t +  22\eta\sqrt{\sigma_1^*\delta_t^3} + 4\eta\delta_t^2.
	\end{align*}
	Recall that $\nu := (14\beta + 81)\alpha \mu r + 26\sqrt{\epsilon} + 18\beta^{-1}$. By letting $\beta = c_1\kappa$, $\epsilon = c_2/\kappa^2$ and assuming $\alpha \leq c_3/(\mu r \kappa^2)$ and $\delta_t \leq c_4\sigma_r^*/\kappa$ for some sufficiently small constants $\{c_i\}_{i=1}^4$, we can have $-2\eta(W_1 + W_2) + \eta^2(W_3 + W_4) \leq -\frac{1}{64}\eta\sigma_r^*\delta_t$, which implies that
	\begin{equation*}
	\delta_{t+1} \leq \left(1 - \frac{\eta \sigma_r^*}{64}\right)\delta_t,
	\end{equation*}
	and thus completes the proof.
\end{proof}

\subsection{Proof of Corollary \ref{cor:rpca_partial}} \label{proof:cor:rpca_partial}
We need $\alpha \lesssim \frac{1}{\mu \kappa^2 r}$ due to the condition of Theorem \ref{thm:convergence_partial}. Letting the initial error provided in Theorem \ref{thm:initial_partial} be less than the corresponding condition in Theorem \ref{thm:convergence_partial}, we have
\[
51\sqrt{\kappa} \alpha \mu r\sqrt{r} \sqrt{\sigma_1^*} + 7c_1\epsilon \sqrt{\kappa \sigma_1^*} \lesssim \sqrt{\sigma_r^*/\kappa},
\]
which leads to
\[
\alpha \lesssim \frac{1}{\mu \sqrt{r^3\kappa^3}}, ~~ \epsilon \lesssim \frac{1}{\sqrt{\kappa^3}}.
\]
Plugging the above two upper bounds into the second term in \eqref{eq:condition_thm_initial_partial}, it suffices to have
\[
p \gtrsim \frac{\kappa^3\mu r^2\log d}{d_1 \wedge d_2}.
\]
Comparing the above bound with the second term in \eqref{eq:condition_thm_convergence_partial} completes the proof.

\subsection{Proof of Lemma \ref{lem:local_descent_L}} \label{proof:lem:local_descent_L}
Let $M := \ensuremath{U}\ensuremath{V}^{\top}$. We observe that
\[
\nabla_{M} \mathcal{L}(\ensuremath{U}, \ensuremath{V}; S) = M + S - \ensuremath{ M^{*} } - \ensuremath{ S^{*} }.
\]
Plugging it back into the left hand side of \eqref{eq:local_descent}, we obtain
\begin{align}
&\trinprod{\nabla_{M} \mathcal{L}(\ensuremath{U}, \ensuremath{V}; S)}{ \ensuremath{U} \ensuremath{V}^{\top} - \ensuremath{U_{\pi^*}}\ensuremath{V_{\pi^*}}^{\top} + \ensuremath{\Delta_{U}} \ensuremath{\Delta_{V}}^{\top} } = \trinprod{M + S - \ensuremath{ M^{*} } - \ensuremath{ S^{*} }}{ M - \ensuremath{ M^{*} } + \ensuremath{\Delta_{U}} \ensuremath{\Delta_{V}}^{\top} }\notag \\
& \geq \frobnorm{M - \ensuremath{ M^{*} }}^2 - \underbrace{\abs{   \trinprod{S - \ensuremath{ S^{*} }}{M - \ensuremath{ M^{*} }}}}_{T_1} - \underbrace{\abs{\trinprod{ M + S - \ensuremath{ M^{*} } - \ensuremath{ S^{*} }}{\ensuremath{\Delta_{U}} \ensuremath{\Delta_{V}}^{\top}}}}_{T_2}. \label{eq:tmp_local_descent}
\end{align}

Next we derive upper bounds of $T_1$ and $T_2$ respectively.

\paragraph{Upper bound of $T_1$.} We denote the support of $S$, $\ensuremath{ S^{*} }$ by $\Omega$ and $\Omega^*$ respectively. Since $S - \ensuremath{ S^{*} }$ is supported on $\Omega^*\cup \Omega$, we have
\[
T_1 \leq \underbrace{\abs{\trinprod{\ensuremath{\Pi}_{\Omega}(S - \ensuremath{ S^{*} })}{M - \ensuremath{ M^{*} }}}}_{W_1} + \underbrace{\abs{\trinprod{\ensuremath{\Pi}_{\Omega^* \setminus \Omega}(S - \ensuremath{ S^{*} })}{M - \ensuremath{ M^{*} }}}}_{W_2}.
\]
Recall that for any $(i,j) \in \Omega$, we have $\entry{S}{i}{j} = \entry{(\ensuremath{ M^{*} } + \ensuremath{ S^{*} } - M)}{i}{j}$. Accordingly, we have
\begin{equation} \label{eq:T1}
W_1 = \frobnorm{\ensuremath{\Pi}_{\Omega}(M - \ensuremath{ M^{*} })}^2.
\end{equation}

Now we turn to bound $W_2$. Since $S_{(i,j)} = 0$ for any $(i,j) \in \Omega^*\setminus \Omega$, we have
\[
W_2 = \abs{\trinprod{ \ensuremath{\Pi}_{\Omega^*\setminus \Omega}\ensuremath{ S^{*} }}{M - \ensuremath{ M^{*} }}}.
\]
Let $u_i$ be the $i$-th row of $M - \ensuremath{ M^{*} }$, and $v_j$ be the $j$-th column of $M - \ensuremath{ M^{*} }$. For any $k \in [d_2]$, we let $u_i^{(k)}$ denote the element of $u_i$ that has the $k$-th largest magnitude. Similarly, for any $k \in [d_1]$, we let $v_j^{(k)}$ denote the element of $v_j$ that has the $k$-th largest magnitude.

From the design of sparse estimator \eqref{eq:sparse_estimator}, we have that for any $(i,j) \in \Omega^*\setminus \Omega$, $\abs{\entry{(\ensuremath{ M^{*} } + \ensuremath{ S^{*} } - M)}{i}{j}}$ is either smaller than the $\gamma \alpha d_2$-th largest entry of the $i$-th row of $\ensuremath{ M^{*} } + \ensuremath{ S^{*} } - M$ or smaller than the $\gamma\alpha d_1$-th largest entry of the $j$-th column of $\ensuremath{ M^{*} } + \ensuremath{ S^{*} } - M$. Note that $\ensuremath{ S^{*} }$ only contains at most $\alpha$-fraction nonzero entries per row and column. As a result, $\abs{\entry{(\ensuremath{ M^{*} } + \ensuremath{ S^{*} } - M)}{i}{j}}$ has to be less than the magnitude of $u_i^{(\gamma \alpha d_2 - \alpha d_2)}$ or $v_j^{(\gamma \alpha d_1 - \alpha d_1)}$. Formally, we have for $(i,j) \in \Omega^*\setminus \Omega$,
\begin{equation} \label{eq:def_b_ij}
\abs{\entry{(\ensuremath{ M^{*} } + \ensuremath{ S^{*} } - M)}{i}{j}} \leq \underbrace{\max\left\{ \abs{u_i^{(\gamma \alpha d_2 - \alpha d_2)}},~ \abs{v_j^{(\gamma \alpha d_1 - \alpha d_1)}}\right\}}_{b_{ij}}.
\end{equation}
Furthermore, we obtain
\begin{equation} \label{eq:b_ij}
b_{ij}^2 \leq  \abs{u_i^{(\gamma \alpha d_2 - \alpha d_2)}}^2 + \abs{v_j^{(\gamma \alpha d_1 - \alpha d_1)}}^2 \leq \frac{\twonorm{u_i}^2}{(\gamma - 1)\alpha d_2} + \frac{\twonorm{v_j}^2}{(\gamma - 1)\alpha d_1}.
\end{equation}
Meanwhile, for any $(i,j) \in \Omega^*\setminus \Omega$, we have
\begin{align}
\abs{\entry{\ensuremath{ S^{*} }}{i}{j} \cdot \entry{(M - \ensuremath{ M^{*} })}{i}{j}} & = \abs{  \entry{(\ensuremath{ M^{*} }  + \ensuremath{ S^{*} } - M - \ensuremath{ M^{*} } + M)}{i}{j} \cdot \entry{(M - \ensuremath{ M^{*} })}{i}{j} } \notag \\
& \leq \abs{\entry{(M - \ensuremath{ M^{*} })}{i}{j}}^2 + \abs{\entry{(\ensuremath{ M^{*} } + \ensuremath{ S^{*} } - M)}{i}{j}}\cdot\abs{\entry{(M - \ensuremath{ M^{*} })}{i}{j}} \notag\\
& \leq  \abs{\entry{(M - \ensuremath{ M^{*} })}{i}{j}}^2 + b_{ij}\cdot\abs{(M - \ensuremath{ M^{*} })_{(i,j)}} \notag \\
& \leq \left(1 + \frac{\beta}{2}\right)\abs{\entry{(M - \ensuremath{ M^{*} })}{i}{j}}^2 + \frac{b_{ij}^2}{2\beta}, \label{eq:inner_product}
\end{align}
where $\beta$ in the last step can be any positive number. Combining \eqref{eq:b_ij} and \eqref{eq:inner_product} leads to
\begin{align}
W_2 & \leq \sum_{(i,j) \in \Omega^*\setminus \Omega}  \abs{\entry{\ensuremath{ S^{*} }}{i}{j} \cdot \entry{(M - \ensuremath{ M^{*} })}{i}{j}} \notag\\
& \leq \left( 1 + \frac{\beta}{2}\right)\frobnorm{\ensuremath{\Pi}_{\Omega^*\setminus \Omega}(M - \ensuremath{ M^{*} })}^2 + \sum_{(i, j) \in \Omega^* \setminus \Omega} \frac{b_{ij}^2}{2\beta} \notag\\
& \leq  \left( 1 + \frac{\beta}{2}\right)\frobnorm{\ensuremath{\Pi}_{\Omega^*\setminus \Omega}(M - \ensuremath{ M^{*} })}^2  + \frac{1}{2\beta} \sum_{(i, j) \in \Omega^* \setminus \Omega} \left( \frac{\twonorm{u_i}^2}{(\gamma - 1)\alpha d_2} + \frac{\twonorm{v_j}^2}{(\gamma - 1)\alpha d_1} \right) \notag\\
& \leq  \left( 1 + \frac{\beta}{2}\right)\frobnorm{\ensuremath{\Pi}_{\Omega^*\setminus \Omega}(M - \ensuremath{ M^{*} })}^2 + \frac{1}{\beta(\gamma - 1)}\frobnorm{M - \ensuremath{ M^{*} }}^2. \label{eq:T2}
\end{align}
In the last step, we use
\begin{align} \label{eq:tmp3}
& \sum_{(i,j) \in \Omega^*\setminus \Omega} \left(\frac{1}{d_2}\twonorm{u_i}^2 + \frac{1}{d_1}\twonorm{v_j}^2 \right) \leq \sum_{(i,j) \in \Omega^*} \left(\frac{1}{d_2}\twonorm{u_i}^2 + \frac{1}{d_1}\twonorm{v_j}^2 \right) \notag \\
& \leq \sum_{i \in [d]}\sum_{j \in \row{\Omega^*}{i}} \frac{1}{d_2}\twonorm{u_i}^2 + \sum_{j \in [d]}\sum_{i \in \column{\Omega^*}{j}} \frac{1}{d_1}\twonorm{v_j}^2 \notag\\
& \leq \alpha \sum_{i \in [d]}\twonorm{u_i}^2 + \alpha \sum_{j \in [d]}\twonorm{v_j}^2 = 2\alpha\frobnorm{M - \ensuremath{ M^{*} }}^2.
\end{align}
We introduce shorthand $\delta := \frobnorm{\ensuremath{\Delta_{U}}}^2 + \frobnorm{\ensuremath{\Delta_{V}}}^2$. We prove the following inequality in the end of this section.
\begin{equation} \label{eq:M_frobnorm_bound}
\frobnorm{M - \ensuremath{ M^{*} }} \leq \sqrt{5\sigma_1^*\delta}.
\end{equation}

Combining \eqref{eq:T1}, \eqref{eq:T2} and \eqref{eq:M_frobnorm_bound} leads to
\begin{align}
T_1 & \leq \frobnorm{\ensuremath{\Pi}_{\Omega}(M - \ensuremath{ M^{*} })}^2 + \left(1 + \frac{\beta}{2}\right)\frobnorm{\ensuremath{\Pi}_{\Omega^*\setminus \Omega}(M - \ensuremath{ M^{*} })}^2 + \frac{5\sigma_1^*\delta}{\beta(\gamma  - 1)} \notag\\
& \leq 9(2\gamma + \beta + 2)\alpha \mu r \sigma_1^*\delta + \frac{5\sigma_1^*\delta}{\beta(\gamma  - 1)}, \label{eq:V1}
\end{align}
where the last step follows from Lemma \ref{lem:projection} by noticing that $\ensuremath{\Pi}_{\Omega}(M - \ensuremath{ M^{*} })$ has at most $\gamma\alpha$-fraction nonzero entries per row and column.

\paragraph{Upper bound of $T_2$.} To ease notation, we let $C := M + S - \ensuremath{ M^{*} } - \ensuremath{ S^{*} }$. We observe that $C$ is supported on $\Omega^c$, we have
\[
T_2 \leq \underbrace{\abs{\trinprod{\ensuremath{\Pi}_{\Omega^{*c} \cap \Omega^c} (M - \ensuremath{ M^{*} })}{ \ensuremath{\Delta_{U}} \ensuremath{\Delta_{V}}^{\top} }}}_{W_3} + \underbrace{\abs{\trinprod{ \ensuremath{\Pi}_{\Omega^{*} \cap \Omega^c}C}{\ensuremath{\Delta_{U}} \ensuremath{\Delta_{V}}^{\top}}}}_{W_4}.
\]
By Cauchy-Swartz inequality, we have
\[
W_3 \leq \frobnorm{\ensuremath{\Pi}_{\Omega^{*c} \cap \Omega^c} (M - \ensuremath{ M^{*} })}\frobnorm{\ensuremath{\Delta_{U}} \ensuremath{\Delta_{V}}^{\top} } \leq \frobnorm{M - \ensuremath{ M^{*} }}\frobnorm{\ensuremath{\Delta_{U}}}\frobnorm{\ensuremath{\Delta_{V}}} \leq \sqrt{5\sigma_1^*\delta^3}/2,
\]
where the last step follows from \eqref{eq:M_frobnorm_bound} and  $\frobnorm{\ensuremath{\Delta_{U}}}\frobnorm{\ensuremath{\Delta_{V}}} \leq \delta/2$.

It remains to bound $W_4$. By Cauchy-Swartz inequality, we have
\begin{align*}
W_4 & \leq \frobnorm{\ensuremath{\Pi}_{\Omega^*\cap \Omega^c}C}\frobnorm{\ensuremath{\Delta_{U}} \ensuremath{\Delta_{V}}^{\top}} \leq \frobnorm{\ensuremath{\Pi}_{\Omega^*\cap \Omega^c}(\ensuremath{ M^{*} } + \ensuremath{ S^{*} } - M)}\frobnorm{\ensuremath{\Delta_{U}} \ensuremath{\Delta_{V}}^{\top}} \\
& \overset{(a)}{\leq} \sqrt{\sum_{(i,j)\in \Omega^*\setminus \Omega} b_{ij}^2} \frobnorm{\ensuremath{\Delta_{U}}}\frobnorm{\ensuremath{\Delta_{V}}} \overset{(b)}{\leq} \left[\sum_{(i,j) \in \Omega^*\setminus \Omega} \frac{\twonorm{u_i}^2}{(\gamma - 1)\alpha d_2} + \frac{\twonorm{v_j}^2}{(\gamma - 1)\alpha d_1} \right]^{1/2}\frobnorm{\ensuremath{\Delta_{U}}}\frobnorm{\ensuremath{\Delta_{V}}}. \\
& \overset{(c)}{\leq} \sqrt{\frac{2}{\gamma - 1}}\frobnorm{M - \ensuremath{ M^{*} }}\frobnorm{\ensuremath{\Delta_{U}}}\frobnorm{\ensuremath{\Delta_{V}}} \leq  \sqrt{\frac{5\sigma_1^*\delta^3}{2(\gamma - 1)}},
\end{align*}
where step $(a)$ is from \eqref{eq:def_b_ij}, step $(b)$ follows from \eqref{eq:b_ij}, and step $(c)$ follows from \eqref{eq:tmp3}. Combining the upper bounds of $W_3$ and $W_4$, we obtain
\begin{equation} \label{eq:V2}
T_2 \leq  \sqrt{5\sigma_1^*\delta^3}/2 + \sqrt{\frac{5\sigma_1^*\delta^3}{2(\gamma - 1)}}.
\end{equation}

\paragraph{Combining pieces.} Now we choose $\gamma = 2$. Then inequality \eqref{eq:V1} implies that
\[
T_1 \leq [9(\beta+6)\alpha \mu r + 5\beta^{-1}]\sigma_1^*\delta.
\]
Inequality \eqref{eq:V2} then implies that
\[
T_2 \leq 3\sqrt{\sigma_1^*\delta^3}.
\]

Plugging the above two inequalities into \eqref{eq:tmp_local_descent} completes the proof.

\vskip .1in
\noindent{\em Proof of inequality \eqref{eq:M_frobnorm_bound}.}
We find that
\begin{align*}
\frobnorm{M - \ensuremath{ M^{*} }}^2 & \leq \left[\sqrt{\sigma_1^*}(\frobnorm{\ensuremath{\Delta_{V}}} + \frobnorm{\ensuremath{\Delta_{U}}}) + \frobnorm{\ensuremath{\Delta_{U}}}\frobnorm{\ensuremath{\Delta_{V}}}\right]^2 \\
& \leq \left[\sqrt{\sigma_1^*}(\frobnorm{\ensuremath{\Delta_{V}}} + \frobnorm{\ensuremath{\Delta_{U}}}) + \frac{1}{2}\sqrt{\sigma_1^*}\frobnorm{\ensuremath{\Delta_{U}}} + \frac{1}{2}\sqrt{\sigma_1^*}\frobnorm{\ensuremath{\Delta_{V}}}\right]^2 \\
& \leq 5\sigma_1^*(\frobnorm{\ensuremath{\Delta_{U}}}^2 + \frobnorm{\ensuremath{\Delta_{V}}}^2),
\end{align*}
where the first step follows from the upper bound of $\frobnorm{M - \ensuremath{ M^{*} }}$ shown in Lemma \ref{lem:delta_norm}, and the second step follows from the assumption $\frobnorm{\ensuremath{\Delta_{U}}}, \frobnorm{\ensuremath{\Delta_{V}}} \leq \sqrt{\sigma_1^*}$.

\subsection{Proof of Lemma \ref{lem:local_descent_g}} \label{proof:lem:local_descent_g}
We first observe that
\[
\nabla_{\ensuremath{U}} \mathcal{G}(\ensuremath{U}, \ensuremath{V}) = \frac{1}{2}\ensuremath{U}(\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}),  ~\nabla_{\ensuremath{V}} \mathcal{G}(\ensuremath{U}, \ensuremath{V}) = \frac{1}{2}\ensuremath{V}(\ensuremath{V}^{\top}\ensuremath{V} - \ensuremath{U}^{\top}\ensuremath{U}),
\]
Therefore, we obtain
\begin{align}
& \trinprod{\nabla_{\ensuremath{U}} \mathcal{G}(\ensuremath{U}, \ensuremath{V})}{U -  \ensuremath{U_{\pi^*}}} + \trinprod{\nabla_{\ensuremath{V}} \mathcal{G}(\ensuremath{U}, \ensuremath{V})}{V -  \ensuremath{V_{\pi^*}}}  \notag\\
& = \frac{1}{2}\trinprod{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}}{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V} - \ensuremath{U}^{\top}\ensuremath{U_{\pi^*}} + \ensuremath{V}^{\top}\ensuremath{V_{\pi^*}}} \notag\\
& = \frac{1}{4}\frobnorm{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}}^2 +  \frac{1}{4}\trinprod{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}}{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V} - 2\ensuremath{U}^{\top}\ensuremath{U_{\pi^*}} + 2\ensuremath{V}^{\top}\ensuremath{V_{\pi^*}}} \notag\\
& =  \frac{1}{4}\frobnorm{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}}^2 +  \frac{1}{4}\trinprod{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}}{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V} - 2\ensuremath{\Delta_{U}}^{\top}\ensuremath{U_{\pi^*}} + 2\ensuremath{\Delta_{V}}^{\top}\ensuremath{V_{\pi^*}}}, \label{eq:lower_G}
\end{align}
where the last step follows from $\ensuremath{\Delta_{U}}^{\top}\ensuremath{U_{\pi^*}} - \ensuremath{\Delta_{V}}^{\top}\ensuremath{V_{\pi^*}} = \ensuremath{U}^{\top}\ensuremath{U_{\pi^*}} - \ensuremath{V}^{\top}\ensuremath{V_{\pi^*}}$ since $\ensuremath{U_{\pi^*}}^{\top}\ensuremath{U_{\pi^*}} = \ensuremath{V_{\pi^*}}^{\top}\ensuremath{V_{\pi^*}}$.
Note that
\begin{align*}
\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V} & = (\ensuremath{U_{\pi^*}} + \ensuremath{\Delta_{U}})^{\top}(\ensuremath{U_{\pi^*}} + \ensuremath{\Delta_{U}}) -  (\ensuremath{V_{\pi^*}} + \ensuremath{\Delta_{V}})^{\top}(\ensuremath{V_{\pi^*}} + \ensuremath{\Delta_{V}})\\
& = \ensuremath{U_{\pi^*}}^{\top}\ensuremath{\Delta_{U}} + \ensuremath{\Delta_{U}}^{\top}\ensuremath{U_{\pi^*}} + \ensuremath{\Delta_{U}}^{\top}\ensuremath{\Delta_{U}} - \ensuremath{V_{\pi^*}}^{\top}\ensuremath{\Delta_{V}} - \ensuremath{\Delta_{V}}^{\top}\ensuremath{V_{\pi^*}} - \ensuremath{\Delta_{V}}^{\top}\ensuremath{\Delta_{V}},
\end{align*}
where we use $\ensuremath{U_{\pi^*}}^{\top}\ensuremath{U_{\pi^*}} = \ensuremath{V_{\pi^*}}^{\top}\ensuremath{V_{\pi^*}}$ again in the last step. Furthermore, since $\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}$ is symmetric, we have
\begin{align*}
& \trinprod{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}}{\ensuremath{U_{\pi^*}}^{\top}\ensuremath{\Delta_{U}} + \ensuremath{\Delta_{U}}^{\top}\ensuremath{U_{\pi^*}} - \ensuremath{V_{\pi^*}}^{\top}\ensuremath{\Delta_{V}} - \ensuremath{\Delta_{V}}^{\top}\ensuremath{V_{\pi^*}}} \\
& = \trinprod{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}}{2\ensuremath{\Delta_{U}}^{\top}\ensuremath{U_{\pi^*}} - 2\ensuremath{\Delta_{V}}^{\top}\ensuremath{V_{\pi^*}}}.
\end{align*}
Using these arguments, for the second term in \eqref{eq:lower_G}, denoted by $T_2$, we have
\[
T_2 = \frac{1}{4}\trinprod{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}}{\ensuremath{\Delta_{U}}^{\top}\ensuremath{\Delta_{U}} - \ensuremath{\Delta_{V}}^{\top}\ensuremath{\Delta_{V}}}.
\]
Furthermore, we have
\begin{align}
4T_2 & \leq \abs{\trinprod{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}}{\ensuremath{\Delta_{U}}^{\top}\ensuremath{\Delta_{U}} - \ensuremath{\Delta_{V}}^{\top}\ensuremath{\Delta_{V}}}} \leq \frobnorm{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}}\left(\frobnorm{\ensuremath{\Delta_{U}}}^2 + \frobnorm{\ensuremath{\Delta_{V}}}^2\right) \notag\\
& \leq \left( \frobnorm{\ensuremath{U}^{\top}\ensuremath{U}  - \ensuremath{U_{\pi^*}}^{\top}\ensuremath{U_{\pi^*}}} + \frobnorm{\ensuremath{V}^{\top}\ensuremath{V}  - \ensuremath{V_{\pi^*}}^{\top}\ensuremath{V_{\pi^*}}}\right) \delta \notag\\
& \leq 2\left(\opnorm{\ensuremath{U_{\pi^*}}}\frobnorm{\ensuremath{\Delta_{U}}} + \opnorm{\ensuremath{V_{\pi^*}}}\frobnorm{\ensuremath{\Delta_{V}}}\right)\delta \leq 2\sqrt{2\sigma_1^*\delta^3}. \label{eq:GT2}
\end{align}

It remains to find a lower bound of $\frobnorm{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}}$. The following inequality, which we turn to prove later, is true:
\begin{equation} \label{eq:it}
\frobnorm{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}}^2 \geq \frobnorm{\ensuremath{U} \ensuremath{U}^{\top} - \ensuremath{U_{\pi^*}} \ensuremath{U_{\pi^*}}^{\top}}^2 + \frobnorm{\ensuremath{V} \ensuremath{V}^{\top} - \ensuremath{V_{\pi^*}} \ensuremath{V_{\pi^*}}^{\top}}^2 - 2\frobnorm{\ensuremath{U} \ensuremath{V}^{\top} - \ensuremath{U_{\pi^*}} \ensuremath{V_{\pi^*}}^{\top}}^2.
\end{equation}
Proceeding with the first term in \eqref{eq:lower_G} by using \eqref{eq:it}, we get
\begin{align}
& \frac{1}{4}\frobnorm{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}}^2 = \frac{1}{8}\frobnorm{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}}^2 + \frac{1}{8}\frobnorm{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}}^2 \notag\\
& \geq \frac{1}{8}\frobnorm{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}}^2 + \frac{1}{8}\frobnorm{\ensuremath{U} \ensuremath{U}^{\top} - \ensuremath{U_{\pi^*}} \ensuremath{U_{\pi^*}}^{\top}}^2 + \frac{1}{8}\frobnorm{\ensuremath{V} \ensuremath{V}^{\top} - \ensuremath{V_{\pi^*}} \ensuremath{V_{\pi^*}}^{\top}}^2 - \frac{1}{4}\frobnorm{\ensuremath{U} \ensuremath{V}^{\top} - \ensuremath{U_{\pi^*}} \ensuremath{V_{\pi^*}}^{\top}}^2 \notag\\
& = \frac{1}{8}\frobnorm{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}}^2 + \frac{1}{8}\frobnorm{FF^{\top} - F_{\pi^*}F^{\top}_{\pi^*}}^2 - \frac{1}{2}\frobnorm{\ensuremath{U} \ensuremath{V}^{\top} - \ensuremath{U_{\pi^*}}\ensuremath{V_{\pi^*}}^{\top}}^2, \label{eq:lower_bound_G}
\end{align}
where we let
\[
F := \begin{bmatrix}
\ensuremath{U} \\ \ensuremath{V}\end{bmatrix}, ~~  F_{\pi^*} := \begin{bmatrix}
\ensuremath{U_{\pi^*}} \\ \ensuremath{V_{\pi^*}}\end{bmatrix}.
\]

Introduce $\Delta_F := F - F_{\pi^*}$. Recall that $\delta := \frobnorm{\ensuremath{\Delta_{U}}}^2 + \frobnorm{\ensuremath{\Delta_{V}}}^2$. Equivalently $\delta = \frobnorm{\Delta_F}^2$. We have
\begin{align*}
\frobnorm{FF^{\top} - F_{\pi^*}F^{\top}_{\pi^*}} & = \frobnorm{\Delta_F F^{\top}_{\pi*} + F_{\pi^*}\Delta_F^{\top} + \Delta_F\Delta_F^{\top}} \\
& \geq \frobnorm{\Delta_F F^{\top}_{\pi*} + F_{\pi^*}\Delta_F^{\top}} - \frobnorm{\Delta_F}^2 = \frobnorm{\Delta_F F^{\top}_{\pi*} + F_{\pi^*}\Delta_F^{\top}} - \delta.
\end{align*}
For the first term, we have
\begin{align*}
& \frobnorm{\Delta_F F^{\top}_{\pi*} + F_{\pi^*}\Delta_F^{\top}}^2  = 2\frobnorm{\Delta_F F^{\top}_{\pi*}}^2 + \trinprod{\Delta_F F^{\top}_{\pi*}}{F_{\pi^*}\Delta_F^{\top}} \\
& \geq  2\sigma_r(F_{\pi^*})^2 \frobnorm{\Delta_F}^2 + \trinprod{\Delta_F F^{\top}_{\pi*}}{F_{\pi^*}\Delta_F^{\top}} = 4\sigma_r^*\frobnorm{\Delta_F}^2 + 	\trinprod{\Delta_F F^{\top}_{\pi*}}{F_{\pi^*}\Delta_F^{\top}}.
\end{align*}
For the cross term, by the following result, proved in \cite{chen2015fast} (we also provide a proof in Section \ref{proof:lem:symm} for the sake of completeness), we have $\trinprod{\Delta_F F^{\top}_{\pi*}}{F_{\pi^*}\Delta_F^{\top}} \geq 0$.
\begin{lemma} \label{lem:symm}
	When $\opnorm{F - F_{\pi^*}} < \sqrt{2\sigma_r^*}$, we have that $\Delta_F^{\top}F_{\pi^*}$ is symmetric.
\end{lemma}
Accordingly, we have $\frobnorm{FF^{\top} - F_{\pi^*}F^{\top}_{\pi^*}} \geq 2\sqrt{\sigma_r^*\delta}- \delta \geq \sqrt{\sigma_r^*\delta}$ under condition $\delta \leq \sigma_r^*$. Plugging this lower bound into \eqref{eq:lower_bound_G}, we obtain
\[
\frac{1}{4}\frobnorm{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}}^2 \geq \frac{1}{8}\frobnorm{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}}^2 + \frac{1}{8}\sigma_r^*\delta - \frac{1}{2}\frobnorm{\ensuremath{U} \ensuremath{V}^{\top} - \ensuremath{U_{\pi^*}}\ensuremath{V_{\pi^*}}^{\top}}^2.
\]
Putting \eqref{eq:lower_G}, \eqref{eq:GT2} and the above inequality together completes the proof.

\vskip 0.1in
\noindent{\em Proof of inequality \eqref{eq:it}.} For the term on the left hand side of \eqref{eq:it}, it is easy to check that
\begin{equation} \label{eq:tmp_local_G}
\frobnorm{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}}^2 = \frobnorm{\ensuremath{U}\PIU^{\top}}^2 + \frobnorm{\ensuremath{V}\PIV^{\top}}^2 - 2\frobnorm{\ensuremath{U}\ensuremath{V}^{\top}}^2.
\end{equation}
The property $\ensuremath{U_{\pi^*}}^{\top} \ensuremath{U_{\pi^*}} = \ensuremath{V_{\pi^*}}^{\top} \ensuremath{V_{\pi^*}}$ implies that $\frobnorm{\ensuremath{U_{\pi^*}}\UniqueU^{\top}} = \frobnorm{\ensuremath{V_{\pi^*}}\UniqueV^{\top}} = \frobnorm{\ensuremath{U_{\pi^*}}\ensuremath{V_{\pi^*}}^{\top}}$. Therefore, expanding those quadratic terms on the right hand side of \eqref{eq:it}, one can show that it is equal to
\begin{equation} \label{eq:tmp_local_G_1}
\frobnorm{\ensuremath{U}\PIU^{\top}}^2 + \frobnorm{\ensuremath{V}\PIV^{\top}}^2 - 2\frobnorm{\ensuremath{U_{\pi^*}}^{\top}\ensuremath{U}}^2 - 2\frobnorm{\ensuremath{V_{\pi^*}}^{\top}\ensuremath{V}}^2 + 4\trinprod{\ensuremath{U_{\pi^*}}^{\top}\ensuremath{U}}{\ensuremath{V_{\pi^*}}^{\top}\ensuremath{V}} - 2\frobnorm{\ensuremath{U}\ensuremath{V}^{\top}}^2.
\end{equation}
Comparing inequalities \eqref{eq:tmp_local_G} and \eqref{eq:tmp_local_G_1}, it thus remains to show that
\[
-2\frobnorm{\ensuremath{U_{\pi^*}}^{\top}\ensuremath{U}}^2 - 2\frobnorm{\ensuremath{V_{\pi^*}}^{\top}\ensuremath{V}}^2 + 4\trinprod{\ensuremath{U_{\pi^*}}^{\top}\ensuremath{U}}{\ensuremath{V_{\pi^*}}^{\top}\ensuremath{V}} \leq 0.
\]
Equivalently, we always have $\frobnorm{\ensuremath{U_{\pi^*}}^{\top}\ensuremath{U} - \ensuremath{V_{\pi^*}}^{\top}\ensuremath{V}}^2 \geq 0$, and thus prove \eqref{eq:it}.

\subsection{Proof of Lemma \ref{lem:smoothness}} \label{proof:lem:smoothness}
First, we turn to prove \eqref{eq:smooth_G}. As
\[
\nabla_{\ensuremath{U}} \mathcal{G}(\ensuremath{U}, \ensuremath{V}) = \frac{1}{2}\ensuremath{U}(\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}),  ~\nabla_{\ensuremath{V}} \mathcal{G}(\ensuremath{U}, \ensuremath{V}) = \frac{1}{2}\ensuremath{V}(\ensuremath{V}^{\top}\ensuremath{V} - \ensuremath{U}^{\top}\ensuremath{U}),
\]
we have
\[
\frobnorm{\nabla_{\ensuremath{U}} \mathcal{G}(\ensuremath{U}, \ensuremath{V})}^2 + \frobnorm{\nabla_{\ensuremath{V}} \mathcal{G}(\ensuremath{U}, \ensuremath{V})}^2 \leq \frac{1}{4}\left(\opnorm{U}^2 + \opnorm{V}^2\right)\frobnorm{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}}^2.
\]
As $(\ensuremath{U}, \ensuremath{V}) \in \mathbb{B}_2(\sqrt{\sigma_1^*})$, we thus have $\opnorm{\ensuremath{U}} \leq \opnorm{\ensuremath{U_{\pi^*}}} + \opnorm{\ensuremath{U_{\pi^*}} - \ensuremath{U}} \leq 2\sqrt{\sigma_1^*}$, and similarly $\opnorm{\ensuremath{V}} \leq 2\sqrt{\sigma_1^*}$. We obtain
\[
\frobnorm{\nabla_{\ensuremath{U}} \mathcal{G}(\ensuremath{U}, \ensuremath{V})}^2 + \frobnorm{\nabla_{\ensuremath{V}} \mathcal{G}(\ensuremath{U}, \ensuremath{V})}^2 \leq 2\sigma_1^*\frobnorm{\ensuremath{U}^{\top}\ensuremath{U} - \ensuremath{V}^{\top}\ensuremath{V}}^2.
\]
\vskip .1in
Now we turn to prove \eqref{eq:smooth_L}. We observe that
\[
\nabla_{M} \ensuremath{\mathcal{L}}(\ensuremath{U}, \ensuremath{V}; \ensuremath{S}) = M + \ensuremath{S} - \ensuremath{ M^{*} } - \ensuremath{ S^{*} },
\]
where we let $M := \ensuremath{U}\ensuremath{V}^{\top}$.
We denote the support of $S, \ensuremath{ S^{*} }$ by $\Omega$ and $\Omega^*$ respectively. Based on the sparse estimator \eqref{eq:sparse_estimator} for computing $S$, $\nabla_{M} \ensuremath{\mathcal{L}}(\ensuremath{U}, \ensuremath{V}; \ensuremath{S})$ is only supported on $\Omega^c$. We thus have
\begin{align*}
\frobnorm{\nabla_{M} \ensuremath{\mathcal{L}}(\ensuremath{U}, \ensuremath{V}; \ensuremath{S})} & \leq \frobnorm{\ensuremath{\Pi}_{\Omega^c\setminus \Omega^*}(M - \ensuremath{ M^{*} })} + \frobnorm{\ensuremath{\Pi}_{\Omega^c \cap \Omega^*}(M - \ensuremath{ M^{*} } - \ensuremath{ S^{*} })} \\
& \leq \frobnorm{M - \ensuremath{ M^{*} }} + \frobnorm{\ensuremath{\Pi}_{\Omega^c \cap \Omega^*}(M - \ensuremath{ M^{*} } - \ensuremath{ S^{*} })}.
\end{align*}
It remains to upper bound the second term on the right hand side. Following \eqref{eq:def_b_ij} and \eqref{eq:b_ij}, we have
\begin{align*}
& \frobnorm{\ensuremath{\Pi}_{\Omega^c \cap \Omega^*}(M - \ensuremath{ M^{*} } - \ensuremath{ S^{*} })}^2  \leq \sum_{(i,j)\in \Omega^c \cap \Omega^*} \frac{\twonorm{u_i}^2}{(\gamma - 1)\alpha d_2} + \frac{\twonorm{v_j}^2}{(\gamma - 1)\alpha d_1} \leq \frac{2}{\gamma - 1}\frobnorm{M - \ensuremath{ M^{*} }}^2,
\end{align*}
where the last step is proved in \eqref{eq:tmp3}. By choosing $\gamma = 2$,
we thus conclude that
\[
\frobnorm{\nabla_{M} \ensuremath{\mathcal{L}}(\ensuremath{U}, \ensuremath{V}; \ensuremath{S})} \leq (1+\sqrt{2})\frobnorm{M - \ensuremath{ M^{*} }}.
\]

\subsection{Proof of Lemma \ref{lem:local_descent_L_partial}} \label{proof:lem:local_descent_L_partial}
We denote the support of $\ensuremath{\Pi}_{\Phi}(\ensuremath{ S^{*} })$, $S$ by $\Omega^*_o$ and $\Omega$. We always have $\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 \ref{proof:lem:norm_sampling}, shows that the Frobenius norm of any incoherent matrix whose row (or column) space are equal to $\ensuremath{L}^*$ (or $\ensuremath{R}^*$) is well preserved under partial observations supported on $\Phi$.
\begin{lemma} \label{lem:norm_sampling}
	Suppose $\ensuremath{ M^{*} } \in \ensuremath{\mathbb{R}}^{d_1 \times d_2}$ is a rank $r$ and $\mu$-incoherent matrix that has SVD $\ensuremath{ M^{*} } = \ensuremath{L}^*\ensuremath{\Sigma}^* \ensuremath{R}^{*\top}$. Then there exists an absolute constant $c$ such that for any $\epsilon \in (0,1)$, if $p \geq c\frac{\mu r\log d}{\epsilon^2 (d_1 \wedge d_2)}$, then with probability at least $1 - 2d^{-3}$, we have that for all $A \in \ensuremath{\mathbb{R}}^{d_2 \times r}, B \in \ensuremath{\mathbb{R}}^{d_1\times r}$,
	\[
	(1-\epsilon)\frobnorm{\ensuremath{L}^*A^{\top} + B\ensuremath{R}^{*\top}}^2 \leq p^{-1}\frobnorm{\ensuremath{\Pi}_{\Phi}\left(\ensuremath{L}^*A^{\top} + B\ensuremath{R}^{*\top}\right)}^2 \leq (1+\epsilon)\frobnorm{\ensuremath{L}^*A^{\top} + B\ensuremath{R}^{*\top}}^2.
	\]
\end{lemma}

We need the next result, proved in Section \ref{proof:lem:size_concentration}, to control the number of nonzero entries per row and column in $\Omega^*_o$ and $\Phi$.
\begin{lemma} \label{lem:size_concentration}
	If $p \geq \frac{56}{3}\frac{\log d}{\alpha (d_1 \wedge d_2)}$, then with probability at least $1 - 6d^{-1}$, we have
	\begin{equation*}
	\big| \abs{\row{\Phi}{i}} - pd_2\big| \leq \frac{1}{2}pd_2,~~ \big|\abs{\column{\Phi}{j}} - pd_1\big| \leq \frac{1}{2}pd_1, ~~ \abs{\Omega^*_{o(i,\cdot)}} \leq \frac{3}{2}\alpha p d_2, ~~ \abs{\Omega^*_{o(\cdot, j)}} \leq \frac{3}{2}\alpha p d_1,
	\end{equation*}
	for all $i \in [d_1]$ and $j \in [d_2]$.
\end{lemma}

The next lemma, proved in Section \ref{proof:lem:frobnorm_sampling}, can be used to control the projection of small matrices to $\Phi$.
\begin{lemma} \label{lem:frobnorm_sampling}
	There exists constant $c$ such that for any $\epsilon \in (0,1)$, if $p \geq c\frac{\mu^2 r^2\log d}{\epsilon^2(d_1 \wedge d_2)}$, then with probability at least $1 - \ensuremath{\mathcal{O}}(d^{-1})$, for all matrices $Z \in \ensuremath{\mathbb{R}}^{d_1 \times d_2}$, $\ensuremath{U} \in \ensuremath{\mathbb{R}}^{d_1 \times r}$ and $\ensuremath{V} \in \ensuremath{\mathbb{R}}^{d_2 \times r}$ that satisfy $\twoinfnorm{\ensuremath{U}} \leq \sqrt{\mu r/d_1}$,$\twoinfnorm{\ensuremath{V}} \leq \sqrt{\mu r/d_2}$, we have
	\begin{equation} \label{eq:uv_sampling}
	p^{-1}\frobnorm{\ensuremath{\Pi}_{\Phi}(\ensuremath{U} \ensuremath{V}^{\top})}^2 \leq \frobnorm{\ensuremath{U}}^2\frobnorm{\ensuremath{V}}^2 + \epsilon \frobnorm{\ensuremath{U}}\frobnorm{\ensuremath{V}};
	\end{equation}
	\begin{equation} \label{eq:Zv_sampling}
	p^{-1}\frobnorm{\ensuremath{\Pi}_{\Phi}(Z)V}^2 \leq 2\mu r \frobnorm{\ensuremath{\Pi}_{\Phi}(Z)}^2;
	\end{equation}
	\begin{equation} \label{eq:uZ_sampling}
	p^{-1}\frobnorm{\ensuremath{U}^{\top}\ensuremath{\Pi}_{\Phi}(Z)}^2 \leq 2\mu r \frobnorm{\ensuremath{\Pi}_{\Phi}(Z)}^2.
	\end{equation}
\end{lemma}

In the remainder of this section, we condition on the events in Lemmas \ref{lem:norm_sampling}, \ref{lem:size_concentration} and \ref{lem:frobnorm_sampling}. Now we are ready to prove Lemma \ref{lem:local_descent_L_partial}.

\begin{proof}[Proof of Lemma \ref{lem:local_descent_L_partial}]
	Using shorthand $M := \ensuremath{U}\ensuremath{V}^{\top}$, we have
	\[
	\nabla_{M} \mathcal{\widetilde{L}}(\ensuremath{U}, \ensuremath{V}; S) = p^{-1}\ensuremath{\Pi}_{\Phi}\left(M + S - \ensuremath{ M^{*} } - \ensuremath{ S^{*} }\right).
	\]
	Plugging it back into the left hand side of \eqref{eq:local_descent}, we obtain
	\begin{align}
	&\trinprod{\nabla_{M} \mathcal{\widetilde{L}}(\ensuremath{U}, \ensuremath{V}; S)}{ \ensuremath{U} \ensuremath{V}^{\top} - \ensuremath{U_{\pi^*}}\ensuremath{V_{\pi^*}}^{\top} + \ensuremath{\Delta_{U}} \ensuremath{\Delta_{V}}^{\top} } \notag\\
	& = \frac{1}{p}\trinprod{\ensuremath{\Pi}_{\Phi}\left(M + S - \ensuremath{ M^{*} } - \ensuremath{ S^{*} }\right)}{ M - \ensuremath{ M^{*} } + \ensuremath{\Delta_{U}} \ensuremath{\Delta_{V}}^{\top} }\notag \\
	& \geq \underbrace{\frac{1}{p}\frobnorm{\ensuremath{\Pi}_{\Phi}\left(M - \ensuremath{ M^{*} }\right)}^2}_{T_1} - \underbrace{\frac{1}{p}\abs{   \trinprod{\ensuremath{\Pi}_{\Phi}\left(S - \ensuremath{ S^{*} }\right)}{M - \ensuremath{ M^{*} }}}}_{T_2} - \underbrace{\frac{1}{p}\abs{\trinprod{ \ensuremath{\Pi}_{\Phi}\left(M + S - \ensuremath{ M^{*} } - \ensuremath{ S^{*} }\right)}{\ensuremath{\Delta_{U}} \ensuremath{\Delta_{V}}^{\top}}}}_{T_3}. \label{eq:tmp_local_descent_partial}
	\end{align}

	Next we derive lower bounds of $T_1$, upper bounds of $T_2$ and $T_3$ respectively.

	\paragraph{Lower bound of $T_1$.}  We observe that $M - \ensuremath{ M^{*} } = \ensuremath{U_{\pi^*}}^*\ensuremath{\Delta_{V}}^{\top} + \ensuremath{\Delta_{U}} \ensuremath{V_{\pi^*}}^{\top} + \ensuremath{\Delta_{U}}\ensuremath{\Delta_{V}}^{\top}$. By triangle inequality, we have
	\[
	\frobnorm{\ensuremath{\Pi}_{\Phi}(M - \ensuremath{ M^{*} })} \geq \frobnorm{\ensuremath{\Pi}_{\Phi}(\ensuremath{U_{\pi^*}}\ensuremath{\Delta_{V}}^{\top} + \ensuremath{\Delta_{U}} \ensuremath{V_{\pi^*}}^{\top})} - \frobnorm{\ensuremath{\Pi}_{\Phi}(\ensuremath{\Delta_{U}}\ensuremath{\Delta_{V}}^{\top})}.
	\]
	Note that when $c \geq a - b$ for $a, b \geq 0$, we always have $c^2 \geq \frac{1}{2}a^2 - b^2$. We thus have
	\begin{align*}
	T_1 & \geq \frac{1}{2p}\frobnorm{\ensuremath{\Pi}_{\Phi}(\ensuremath{U_{\pi^*}}\ensuremath{\Delta_{V}}^{\top} + \ensuremath{\Delta_{U}} \ensuremath{V_{\pi^*}}^{\top})}^2 - \frac{1}{p}\frobnorm{\ensuremath{\Pi}_{\Phi}(\ensuremath{\Delta_{U}}\ensuremath{\Delta_{V}}^{\top})}^2 \\
	& \geq \frac{1}{2}(1-\epsilon)\frobnorm{\ensuremath{U_{\pi^*}}\ensuremath{\Delta_{V}}^{\top} + \ensuremath{\Delta_{U}} \ensuremath{V_{\pi^*}}^{\top}}^2 - \frac{1}{p}\frobnorm{\ensuremath{\Pi}_{\Phi}(\ensuremath{\Delta_{U}}\ensuremath{\Delta_{V}}^{\top})}^2 \\
	& \geq \frac{1}{2}(1-\epsilon)\frobnorm{M - \ensuremath{ M^{*} } - \ensuremath{\Delta_{U}}\ensuremath{\Delta_{V}}^{\top}}^2 - \frobnorm{\ensuremath{\Delta_{U}}}^2\frobnorm{\ensuremath{\Delta_{V}}}^2 - 9\epsilon\sigma_1^*\frobnorm{\ensuremath{\Delta_{U}}}\frobnorm{\ensuremath{\Delta_{V}}} \\
	& \geq \frac{1}{4}(1-\epsilon)\frobnorm{M - \ensuremath{ M^{*} }}^2 - \frac{1}{2}(1-\epsilon)\frobnorm{\ensuremath{\Delta_{U}}\ensuremath{\Delta_{V}}}^2 - \frobnorm{\ensuremath{\Delta_{U}}}^2\frobnorm{\ensuremath{\Delta_{V}}}^2 - 9\epsilon\sigma_1^*\frobnorm{\ensuremath{\Delta_{U}}}\frobnorm{\ensuremath{\Delta_{V}}} \\
	& \geq \frac{1}{4}(1-\epsilon)\frobnorm{M - \ensuremath{ M^{*} }}^2 - 2\delta^2 - 5\epsilon\sigma_1^*\delta.
	\end{align*}
	where the second step is implied by Lemma \ref{lem:norm_sampling}, the third step follows from \eqref{eq:uv_sampling} in Lemma \ref{lem:frobnorm_sampling} by noticing that $\twoinfnorm{\ensuremath{\Delta_{U}}} \leq 3\sqrt{\mu r \sigma_1^*/d_1}$ and  $\twoinfnorm{\ensuremath{\Delta_{V}}} \leq 3\sqrt{\mu r \sigma_1^*/d_1}$, which is further implied by \eqref{eq:uvcondition_partial}.

	\paragraph{\bf Upper bound of $T_2$.}
	Since $S - \ensuremath{ S^{*} }$ is supported on $\Omega^*_{0}\cup \Omega$, we have
	\begin{equation} \label{eq:V2_partial}
	pT_2 \leq \abs{\trinprod{ \ensuremath{\Pi}_{\Omega^*_o\setminus \Omega}(\ensuremath{ S^{*} })}{\ensuremath{\Pi}_{\Omega^*_o\setminus \Omega}(M - \ensuremath{ M^{*} })}} + \abs{\trinprod{\ensuremath{\Pi}_{\Omega}(S - \ensuremath{ S^{*} })}{\ensuremath{\Pi}_{\Omega}(M - \ensuremath{ M^{*} })}}.
	\end{equation}
	For any $(i,j) \in \Omega$, we have $(S - \ensuremath{ S^{*} })_{(i,j)} = (\ensuremath{ M^{*} } - M)_{(i,j)}$. Therefore, for the second term on the right hand side, we have
	\begin{equation} \label{eq:tmp4}
	\abs{\trinprod{\ensuremath{\Pi}_{\Omega}(S - \ensuremath{ S^{*} })}{\ensuremath{\Pi}_{\Omega}(M - \ensuremath{ M^{*} })}} \leq \frobnorm{\ensuremath{\Pi}_{\Omega}(M - \ensuremath{ M^{*} })}^2 \leq 18\gamma p \alpha \mu r \sigma_1^* \delta,
	\end{equation}
	where the last inequality follows from Lemma \ref{lem:projection} and the fact that $\abs{\Omega_{(i,\cdot)}} \leq \gamma p \alpha d_2$, $\abs{\Omega_{(\cdot, j)}} \leq \gamma p \alpha d_1$ for all $i \in [d_1]$, $j \in [d_2]$.

	We denote the $i$-th row of $\ensuremath{\Pi}_{\Phi}(M - \ensuremath{ M^{*} })$ by $u_i$, and we denote the $j$-th column of $\ensuremath{\Pi}_{\Phi}(M - \ensuremath{ M^{*} })$ by $v_j$. We let $u_i^{(k)}$ denote the element of $u_i$ that has the $k$-th largest magnitude. We let $v_j^{(k)}$ denote the element of $v_j$ that has the $k$-th largest magnitude.

	For the first term on the right hand side of \eqref{eq:V2_partial}, we first observe that for $(i,j) \in \Omega^*_o\setminus \Omega$, $\abs{(\ensuremath{ M^{*} } + \ensuremath{ S^{*} } - M)_{(i,j)}}$ is either less than the $\gamma p\alpha d_2$-th largest element in the $i$-th row of $\ensuremath{\Pi}_{\Phi}(\ensuremath{ M^{*} } + \ensuremath{ S^{*} } - M)$, or less than $\gamma p \alpha d_1$-th largest element in the $j$-th row of $\ensuremath{\Pi}_{\Phi}(\ensuremath{ M^{*} } + \ensuremath{ S^{*} } - M)$. Based on Lemma \ref{lem:size_concentration}, $\ensuremath{\Pi}_{\Phi}(\ensuremath{ S^{*} })$ has at most $3p\alpha d_2/2$ nonzero entries per row and at most $3p \alpha d_1/2$ nonzero entries per column. Therefore, we have
	\begin{equation} \label{eq:bij_partial}
	\abs{\entry{(\ensuremath{ M^{*} } + \ensuremath{ S^{*} } - M)}{i}{j}} \leq \max\left\{ \abs{u_i^{((\gamma - 1.5)p\alpha d_2)}},~ \abs{v_j^{((\gamma - 1.5)p\alpha d_1)}}\right\}.
	\end{equation}
	In addition, we observe that
	\begin{align}
	& \abs{\trinprod{ \ensuremath{\Pi}_{\Omega^*_o\setminus \Omega}(\ensuremath{ S^{*} })}{ \ensuremath{\Pi}_{\Omega^*_o\setminus \Omega}(M - \ensuremath{ M^{*} })}} \notag \\
	&  \leq \sum_{(i,j) \in \Omega^*_o\setminus \Omega} \abs{(\ensuremath{ M^{*} } + \ensuremath{ S^{*} } - M)_{(i,j)}}\abs{(\ensuremath{ M^{*} } - M)_{(i,j)}} + \abs{(\ensuremath{ M^{*} } - M)_{(i,j)}}^2 \notag\\
	& \leq \left(1 + \frac{\beta}{2}\right)\frobnorm{\ensuremath{\Pi}_{\Omega^*_o}(\ensuremath{ M^{*} } - M)}^2 + \frac{1}{2\beta}\sum_{(i,j) \in \Omega^*_o\setminus \Omega} \abs{(\ensuremath{ M^{*} } + \ensuremath{ S^{*} } - M)_{(i,j)}}^2, \notag \\
	& \leq (27 + 14\beta)p\alpha\mu r\sigma_1^*\delta + \frac{1}{2\beta}\sum_{(i,j) \in \Omega^*_o\setminus \Omega} \abs{(\ensuremath{ M^{*} } + \ensuremath{ S^{*} } - M)_{(i,j)}}^2, \label{eq:tmp5}
	\end{align}
	where the second step holds for any $\beta > 0$ and the last step follows from Lemma \ref{lem:projection} under the size constraints of $\Omega^*_o$ shown in Lemma \ref{lem:size_concentration}. For the second term in \eqref{eq:tmp5}, using \eqref{eq:bij_partial}, we have
	\begin{align} \label{eq:tmp6}
	& \sum_{(i,j) \in \Omega^*_o\setminus \Omega} \abs{(\ensuremath{ M^{*} } + \ensuremath{ S^{*} } - M)_{(i,j)}}^2 \leq \sum_{(i,j) \in \Omega^*_o} \abs{u_i^{((\gamma - 1.5)p\alpha d_2)}}^2 + \abs{v_j^{((\gamma - 1.5)p\alpha d_1)}}^2 \notag \\
	& = \sum_{i \in [d_1]} \sum_{j \in \Omega^*_{o(i,\cdot)} }  \abs{u_i^{((\gamma - 1.5)p\alpha d_2)}}^2 +
	\sum_{j \in [d_2]}\sum_{i \in \Omega^*_{o(\cdot, j)} } \abs{v_j^{((\gamma - 1.5)p\alpha d_1)}}^2 \notag\\
	& \leq \sum_{i \in [d_1]} \frac{1.5}{\gamma - 1.5}\twonorm{u_i}^2 + \sum_{j \in [d_2]} \frac{1.5}{\gamma - 1.5}\twonorm{v_j}^2 \leq  \frac{3}{\gamma - 1.5} \frobnorm{\ensuremath{\Pi}_{\Phi}(M - \ensuremath{ M^{*} })}^2.
	\end{align}
	Moreover, we have
	\begin{align}
	& \frobnorm{\ensuremath{\Pi}_{\Phi}(M - \ensuremath{ M^{*} })}^2 \leq 2\frobnorm{\ensuremath{\Pi}_{\Phi}(\ensuremath{U_{\pi^*}}\ensuremath{\Delta_{V}}^{\top} + \ensuremath{\Delta_{U}} \ensuremath{V_{\pi^*}}^{\top})}^2 + 2\frobnorm{\ensuremath{\Pi}_{\Phi}(\ensuremath{\Delta_{U}}\ensuremath{\Delta_{V}}^{\top})}^2 \notag\\
	& \leq 2(1+\epsilon)p\frobnorm{\ensuremath{U_{\pi^*}}\ensuremath{\Delta_{V}}^{\top} + \ensuremath{\Delta_{U}} \ensuremath{V_{\pi^*}}^{\top}}^2 + 2p\frobnorm{\ensuremath{\Delta_{U}}}^2\frobnorm{\ensuremath{\Delta_{V}}}^2 + 18p\epsilon\sigma_1^* \frobnorm{\ensuremath{\Delta_{U}}}\frobnorm{\ensuremath{\Delta_{V}}} \notag\\
	& \leq 4(1+\epsilon)p\left(\opnorm{\ensuremath{U_{\pi^*}}}^2 \frobnorm{\ensuremath{\Delta_{V}}}^2 + \opnorm{\ensuremath{V_{\pi^*}}}^2 \frobnorm{\ensuremath{\Delta_{U}}}^2\right) + 2p\frobnorm{\ensuremath{\Delta_{U}}}^2\frobnorm{\ensuremath{\Delta_{V}}}^2 + 18p\epsilon\sigma_1^* \frobnorm{\ensuremath{\Delta_{U}}}\frobnorm{\ensuremath{\Delta_{V}}} \notag\\
	& \leq (4 + 13\epsilon)p\sigma_1^*\delta + 2p\delta^2, \label{eq:tmp7}
	\end{align}
	where the second step follows from Lemma \ref{lem:norm_sampling} and inequality \eqref{eq:uv_sampling} in Lemma \ref{lem:frobnorm_sampling}.
	Putting \eqref{eq:V2_partial}-\eqref{eq:tmp7} together, we obtain
	\begin{align*}
	T_2 & \leq (18\gamma + 14\beta + 27)\alpha \mu r \sigma_1^* \delta + \frac{3[(2 + 7\epsilon)\sigma_1^*\delta + \delta^2]}{\beta(\gamma - 1.5)}.
	\end{align*}

	\paragraph{Upper bound of $T_3$.}
	By Cauchy-Schwarz inequality, we have
	\begin{align*}
	pT_3 & \leq \frobnorm{ \ensuremath{\Pi}_{\Phi}(M - \ensuremath{ M^{*} } +  S - \ensuremath{ S^{*} })} \frobnorm{\ensuremath{\Pi}_{\Phi}( \ensuremath{\Delta_{U}}\ensuremath{\Delta_{V}}^{\top})} \\
	& \leq \frobnorm{ \ensuremath{\Pi}_{\Phi}(M - \ensuremath{ M^{*} } +  S - \ensuremath{ S^{*} })}  \sqrt{p\frobnorm{\ensuremath{\Delta_{U}}}^2\frobnorm{\ensuremath{\Delta_{V}}}^2 + 9p\epsilon\sigma_1^* \frobnorm{\ensuremath{\Delta_{U}}}\frobnorm{\ensuremath{\Delta_{V}}}} \\
	& \leq \frobnorm{ \ensuremath{\Pi}_{\Phi}(M - \ensuremath{ M^{*} } +  S - \ensuremath{ S^{*} })}\sqrt{p\delta^2 + 5p\epsilon\sigma_1^*\delta}.
	\end{align*}
	where we use \eqref{eq:uv_sampling} in Lemma \ref{lem:frobnorm_sampling} in the second step.

	We observe that $\ensuremath{\Pi}_{\Phi}(M - \ensuremath{ M^{*} } + S - \ensuremath{ S^{*} })$ is supported on $\Phi\setminus\Omega$. Therefore, we have
	\begin{align*}
	\frobnorm{\ensuremath{\Pi}_{\Phi}(M - \ensuremath{ M^{*} } +  S - \ensuremath{ S^{*} })} & \leq \frobnorm{\ensuremath{\Pi}_{\Phi \cap \Omega^c \cap \Phi^{*c}}(M - \ensuremath{ M^{*} })} + \frobnorm{ \ensuremath{\Pi}_{\Phi \cap \Omega^c \cap \Phi^{*}}(M - \ensuremath{ M^{*} } - \ensuremath{ S^{*} })} \\
	& \leq \frobnorm{\ensuremath{\Pi}_{\Phi}(M - \ensuremath{ M^{*} })}+\frobnorm{ \ensuremath{\Pi}_{\Omega^c \cap \Phi^{*}}(M - \ensuremath{ M^{*} } - \ensuremath{ S^{*} })} \\
	& \leq \frobnorm{\ensuremath{\Pi}_{\Phi}(M - \ensuremath{ M^{*} })} + \sqrt{\frac{3}{\gamma -1.5}}\frobnorm{\ensuremath{\Pi}_{\Phi}(M - \ensuremath{ M^{*} })} \\
	& \leq \left(1 + \sqrt{\frac{3}{\gamma -1.5}}\right)\sqrt{(4 + 13\epsilon)p\sigma_1^*\delta + 2p\delta^2},
	\end{align*}
	where the third step follows from \eqref{eq:tmp6}, and the last step is from \eqref{eq:tmp7}. Under assumptions $\gamma = 3$, $\epsilon \leq 1/4$ and $\delta \leq \sigma_1^*$, we have
	\[
	T_3 \leq 3\sqrt{9\sigma_1^*\delta + 2\delta^2}\sqrt{\delta^2 + 5\epsilon\sigma_1^*\delta} \leq 10\sqrt{\sigma_1^*\delta^3} + 23\sqrt{\epsilon}\sigma_1^*\delta.
	\]

	\paragraph{Combining pieces.} Under the aforementioned assumptions, putting all pieces together leads to
	\begin{align*}
	& \trinprod{\nabla_{M} \mathcal{\widetilde{L}}(\ensuremath{U}, \ensuremath{V}; S)}{ \ensuremath{U} \ensuremath{V}^{\top} - \ensuremath{U_{\pi^*}}\ensuremath{V_{\pi^*}}^{\top} + \ensuremath{\Delta_{U}} \ensuremath{\Delta_{V}}^{\top} } \\
	& \geq \frac{3}{16}\frobnorm{M - \ensuremath{ M^{*} }}^2 - (14\beta + 81)\alpha \mu r \sigma_1^*\delta -\left(26\sqrt{\epsilon} + \frac{18}{\beta}\right)\sigma_1^*\delta - 10\sqrt{\sigma_1^*\delta^3} - 2\delta^2.
	\end{align*}
\end{proof}

\subsection{Proof of Lemma \ref{lem:smoothness_partial}} \label{proof:lem:smoothness_partial}
Let $M := \ensuremath{U}\ensuremath{V}^{\top}$. We find that
\begin{align*}
& \nabla_{\ensuremath{U}}\widetilde{\ensuremath{\mathcal{L}}}(\ensuremath{U},\ensuremath{V};\ensuremath{S}) = p^{-1}\ensuremath{\Pi}_{\Phi}\left(M + S -  \ensuremath{ M^{*} } - \ensuremath{ S^{*} }\right)\ensuremath{V}, \\
& \nabla_{\ensuremath{V}}\widetilde{\ensuremath{\mathcal{L}}}(\ensuremath{U},\ensuremath{V};\ensuremath{S}) = p^{-1}\ensuremath{\Pi}_{\Phi}\left(M + S -  \ensuremath{ M^{*} } - \ensuremath{ S^{*} }\right)^{\top}\ensuremath{U}.
\end{align*}
Conditioning on the event in Lemma \ref{lem:frobnorm_sampling}, since $(\ensuremath{U}, \ensuremath{V}) \in \widebar{\ensuremath{\mathcal{U}}} \times \widebar{\ensuremath{\mathcal{V}}}$, inequalities \eqref{eq:Zv_sampling} and \eqref{eq:uZ_sampling} imply that
\[
\frobnorm{\nabla_{\ensuremath{U}}\widetilde{\ensuremath{\mathcal{L}}}(\ensuremath{U},\ensuremath{V};\ensuremath{S})}^2 + \frobnorm{\nabla_{\ensuremath{V}}\widetilde{\ensuremath{\mathcal{L}}}(\ensuremath{U},\ensuremath{V};\ensuremath{S})}^2 \leq \frac{12}{p}\mu r\sigma_1^*\frobnorm{\ensuremath{\Pi}_{\Phi}\left(M + S -  \ensuremath{ M^{*} } - \ensuremath{ S^{*} }\right)}^2.
\]
It remains to bound the term  $\frobnorm{\ensuremath{\Pi}_{\Phi}\left(M + S -  \ensuremath{ M^{*} } - \ensuremath{ S^{*} }\right)}^2$. Let $\Omega^*_o$ and $\Omega$ be the support of $\ensuremath{\Pi}_{\Phi}(\ensuremath{ S^{*} })$ and $S$ respectively. We observe that
\begin{align*}
\frobnorm{\ensuremath{\Pi}_{\Phi}\left(M + S -  \ensuremath{ M^{*} } - \ensuremath{ S^{*} }\right)}^2 & = \frobnorm{\ensuremath{\Pi}_{\Omega^*_o \setminus \Omega}\left(M -  \ensuremath{ M^{*} } - \ensuremath{ S^{*} }\right)}^2 + \frobnorm{\ensuremath{\Pi}_{\Phi^{*c} \cap \Omega^c \cap \Phi}\left(M -  \ensuremath{ M^{*} }\right)}^2 \\
& \leq \frobnorm{\ensuremath{\Pi}_{\Omega^*_o \setminus \Omega}\left(M -  \ensuremath{ M^{*} } - \ensuremath{ S^{*} }\right)}^2 + \frobnorm{\ensuremath{\Pi}_{\Phi}\left(M -  \ensuremath{ M^{*} }\right)}^2.
\end{align*}
In the proof of Lemma \ref{lem:local_descent_L_partial}, it is shown in \eqref{eq:tmp6} that
\[
\frobnorm{\ensuremath{\Pi}_{\Omega^*_o \setminus \Omega}\left(M -  \ensuremath{ M^{*} } - \ensuremath{ S^{*} }\right)}^2 \leq \frac{3}{\gamma - 1.5}\frobnorm{\ensuremath{\Pi}_{\Phi}(M - \ensuremath{ M^{*} })}^2.
\]
Moreover, following \eqref{eq:tmp7}, we have that
\begin{align*}
\frobnorm{\ensuremath{\Pi}_{\Phi}(M - \ensuremath{ M^{*} })}^2 & \leq 2(1+\epsilon)p\frobnorm{\ensuremath{U_{\pi^*}}\ensuremath{\Delta_{V}}^{\top} + \ensuremath{\Delta_{U}} \ensuremath{V_{\pi^*}}^{\top}}^2 + 2p\frobnorm{\ensuremath{\Delta_{U}}}^2\frobnorm{\ensuremath{\Delta_{V}}}^2 + 18p\epsilon\sigma_1^* \frobnorm{\ensuremath{\Delta_{U}}}\frobnorm{\ensuremath{\Delta_{V}}} \\
& \leq 4(1+\epsilon)p\frobnorm{M - \ensuremath{ M^{*} }}^2 + (6+4\epsilon)p\frobnorm{\ensuremath{\Delta_{U}}}^2\frobnorm{\ensuremath{\Delta_{V}}}^2 + 18p\epsilon\sigma_1^* \frobnorm{\ensuremath{\Delta_{U}}}\frobnorm{\ensuremath{\Delta_{V}}} \\
& \leq 4(1+\epsilon)p\frobnorm{M - \ensuremath{ M^{*} }}^2 + (6+4\epsilon)p\delta^2 + 9p\epsilon\sigma_1^*\delta.
\end{align*}
We thus finish proving our conclusion by combining all pieces and noticing that $\gamma = 3$ and $\epsilon \leq 1/4$.

\section{Proofs for Technical Lemmas} \label{sec:proof_lemmas}

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

\subsection{Proof of Lemma \ref{lem:op_to_inf}} \label{proof:lem:op_to_inf}
We observe that
\[
\opnorm{A} = \sup_{x \in \mathbb{S}^{d_1 - 1}} \sup_{y \in \mathbb{S}^{d_2 - 1}} x^{\top}Ay.
\]
We denote the support of $A$ by $\Omega$. For any $x \in \ensuremath{\mathbb{R}}^{d_1}$, $y \in \ensuremath{\mathbb{R}}^{d_2}$ and $\beta > 0$, we have
\begin{align*}
x^{\top}Ay & = \sum_{(i,j) \in \Omega} x_i A_{(i,j)}y_j \leq \sum_{(i,j) \in \Omega}  \frac{1}{2}\infnorm{A}(\beta^{-1}x_i^2 + \beta y_j^2) \\
& = \frac{1}{2}\infnorm{A}\left(\sum_{i} \sum_{j \in \row{\Omega}{i}} \beta^{-1}x_i^2 + \sum_{j} \sum_{i \in \column{\Omega}{j}} \beta y_j^2 \right) \\
& \leq \frac{1}{2}\infnorm{A} \left(\alpha d_2 \beta^{-1}\twonorm{x}^2 + \alpha  d_1 \beta\twonorm{y}^2\right).
\end{align*}
It is thus implied that $\opnorm{A} \leq \frac{1}{2}\alpha(\beta^{-1}d_2 + \beta d_1)\infnorm{A}$. Choosing $\beta = \sqrt{d_2/d_1}$ completes the proof.

\subsection{Proof of Lemma \ref{lem:norm_sampling}} \label{proof:lem:norm_sampling}
We define a subspace $\mathcal{K} \subseteq \ensuremath{\mathbb{R}}^{d_1 \times d_2}$ as
\[
\mathcal{K} := \left\{X ~\big|~ X = \ensuremath{L}^*A^{\top} + B\ensuremath{R}^{*\top} ~\text{for some}~ A \in \ensuremath{\mathbb{R}}^{d_2 \times r}, B \in \ensuremath{\mathbb{R}}^{d_1 \times r}~\right\}.
\]
Let $\ensuremath{\Pi}_{\mathcal{K}}$ be Euclidean projection onto $\mathcal{K}$. Then according to Theorem 4.1 in \cite{candes2009exact}, under our assumptions, for all matrices $X \in \ensuremath{\mathbb{R}}^{d_1 \times d_2}$, inequality
\begin{equation} \label{eq:exact_mc}
p^{-1}\frobnorm{\left(\ensuremath{\Pi}_{\mathcal{K}}\ensuremath{\Pi}_{\Phi}\ensuremath{\Pi}_{\mathcal{K}} - p\ensuremath{\Pi}_{\mathcal{K}}\right)X} \leq \epsilon\frobnorm{X}
\end{equation}
holds with probability at least $1 - 2d^{-3}$.

In our setting, by restricting $X = \ensuremath{L}^*A^{\top} + B\ensuremath{R}^{*\top}$, we have $\ensuremath{\Pi}_{\mathcal{K}}X = X$. Therefore, \eqref{eq:exact_mc} implies that
\[
\frobnorm{\ensuremath{\Pi}_{\mathcal{K}}\ensuremath{\Pi}_{\Phi}X - pX} \leq p\epsilon\frobnorm{X}.
\]
For $\frobnorm{\ensuremath{\Pi}_{\Phi}X}^2$, we have
\begin{align*}
\frobnorm{\ensuremath{\Pi}_{\Phi}X}^2 & = \trinprod{\ensuremath{\Pi}_{\Phi}X}{\ensuremath{\Pi}_{\Phi}X} = \trinprod{\ensuremath{\Pi}_{\Phi}X}{X} \\
& = \trinprod{\ensuremath{\Pi}_{\mathcal{K}}\ensuremath{\Pi}_{\Phi}X}{X} \leq \frobnorm{\ensuremath{\Pi}_{\mathcal{K}}\ensuremath{\Pi}_{\Phi}X}\frobnorm{X} \leq p(1+\epsilon)\frobnorm{X}^2.
\end{align*}
On the other hand, we have
\begin{align*}
\frobnorm{\ensuremath{\Pi}_{\Phi}X}^2 & = \trinprod{\ensuremath{\Pi}_{\mathcal{K}}\ensuremath{\Pi}_{\Phi}X}{X} = \trinprod{\ensuremath{\Pi}_{\mathcal{K}}\ensuremath{\Pi}_{\Phi}X - pX + pX}{X} \\
& = p\frobnorm{X}^2 - \trinprod{X}{-\ensuremath{\Pi}_{\mathcal{K}}\ensuremath{\Pi}_{\Phi}X + pX} \\
& \geq p\frobnorm{X}^2 - \frobnorm{X}\frobnorm{\ensuremath{\Pi}_{\mathcal{K}}\ensuremath{\Pi}_{\Phi}X - pX} \geq p(1-\epsilon)\frobnorm{X}^2.
\end{align*}
Combining the above two inequalities, we complete the proof.

\subsection{Proof of Lemma \ref{lem:size_concentration}} \label{proof:lem:size_concentration}
We observe that $\abs{\row{\Phi}{i}}$ is a summation of $d_2$ i.i.d. binary random variables with mean $p$ and variance $p(1-p)$. By Bernstein's inequality, for any $i \in [d_1]$,
\[
\Pr\left[ \big| \abs{\row{\Phi}{i}} - pd_2\big| \geq \frac{1}{2}pd_2 \right] \leq 2\exp\left(-\frac{-\frac{1}{2}(pd_2/2)^2}{d_2p(1-p) + \frac{1}{3}(pd_2/2)}\right) \leq 2\exp\left(-\frac{3}{28}pd_2\right).
\]
By probabilistic union bound, we have
\[
\Pr\left[ \sup_{i \in [d_1]} \big|\abs{\row{\Phi}{i}} - pd_2\big| \geq \frac{1}{2}pd_2\right] \leq 2d_1\exp\left(-\frac{3}{28}pd_2\right) \leq 2d^{-1},
\]
where the last inequality holds by assuming $p \geq \frac{56}{3}\frac{\log d}{d_2}$.

The term $\abs{\Omega^*_{o(i,\cdot)}}$ is a summation of at most $\alpha d_2$ i.i.d. binary random variables with mean $p$ and variance $p(1-p)$. Again, applying Bernstein's inequality leads to
\[
\Pr\left[ \abs{\Omega^*_{o(i,\cdot)}} - \mathbb{E}\left[\abs{\Omega^*_{o(i,\cdot)}}\right]  \geq \frac{1}{2}p\alpha d_2\right] \leq \exp\left(-\frac{3}{28}p\alpha d_2\right).
\]
Accordingly, by the assumption $p \geq \frac{56}{3}\frac{\log d}{\alpha d_2}$, we obtain
\[
\Pr\left[ \sup_{i \in [d_1]}\abs{\Omega^*_{o(i,\cdot)}} - pk  \geq \frac{1}{2}pk\right]  \leq d_1\exp\left(-\frac{3}{28}p\alpha d_2\right) \leq d^{-1}.
\]
The proofs for $\abs{\column{\Phi}{j}}$ and $\abs{\Omega^*_{o(\cdot,j)}}$ follow the same idea.
\subsection{Proof of Lemma \ref{lem:frobnorm_sampling}} \label{proof:lem:frobnorm_sampling}
According to Lemma 3.2 in \cite{candes2011robust}, under condition $p \geq c_1\frac{\mu \log d}{d_1 \wedge d_2}$, for any fixed matrix $A \in \ensuremath{\mathbb{R}}^{d_1 \times d_2}$, we have
\[
\opnorm{A - p^{-1}\ensuremath{\Pi}_{\Phi}A} \leq c_2\sqrt{\frac{d\log d}{p}}\infnorm{A},
\]
holds with probability at least $1 - \ensuremath{\mathcal{O}}(d^{-3})$. Letting $A$ be all-ones matrix, then we have that for all $u \in \ensuremath{\mathbb{R}}^{d_1}, v \in \ensuremath{\mathbb{R}}^{d_2}$,
\[
\sum_{(i,j)\in \Phi} u_i v_j \leq p\onenorm{u}\onenorm{v} + c_2\sqrt{pd\log d}\twonorm{u}\twonorm{v}.
\]
We find that
\begin{align*}
\frobnorm{\ensuremath{\Pi}_{\Phi}(\ensuremath{U} \ensuremath{V}^{\top})}^2 & \leq \sum_{(i,j)\in \Phi}\twonorm{\row{\ensuremath{U}}{i}}^2\twonorm{\row{\ensuremath{V}}{j}}^2 \\
& \leq p \frobnorm{\ensuremath{U}}^2\frobnorm{\ensuremath{V}}^2 + c_2\sqrt{pd\log d} \sqrt{\sum_{i \in [d_1]} \twonorm{\row{\ensuremath{U}}{i}}^4}\sqrt{\sum_{j \in [d_2]} \twonorm{\row{\ensuremath{V}}{j}}^4} \\
& \leq p \frobnorm{\ensuremath{U}}^2\frobnorm{\ensuremath{V}}^2 + c_2\sqrt{pd \log d}\frobnorm{\ensuremath{U}}\frobnorm{\ensuremath{V}}\twoinfnorm{\ensuremath{U}}\twoinfnorm{\ensuremath{V}} \\
& \leq p \frobnorm{\ensuremath{U}}^2\frobnorm{\ensuremath{V}}^2 + c_2\sqrt{\frac{p \mu^2r^2d\log d}{d_1 d_2}}\frobnorm{\ensuremath{U}}\frobnorm{\ensuremath{V}}.
\end{align*}
By the assumption $p \gtrsim \frac{\mu^2r^2\log d}{\epsilon^2(d_1 \wedge d_2)}$, we finish proving \eqref{eq:uv_sampling}.

According to the proof of Lemma \ref{lem:size_concentration}, if $p \geq c\frac{\log d}{d_1\wedge d_2}$, with probability at least $1 - \ensuremath{\mathcal{O}}(d^{-1})$, we have $\abs{\row{\Phi}{i}} \leq \frac{3}{2}pd_2$ and $\abs{\column{\Phi}{j}} \leq \frac{3}{2}pd_1$ for all $i \in [d_1]$ and $j \in [d_2]$. Conditioning on this event, we have
\begin{align*}
\frobnorm{\ensuremath{\Pi}_{\Phi}(Z)V}^2 & = \sum_{i \in [d_1]} \sum_{k \in [r]} \inprod{\row{(\ensuremath{\Pi}_{\Phi}(Z))}{i}}{\column{H}{k}}^2 \\
& \leq \sum_{i \in [d_1]} \sum_{k \in [r]} \twonorm{\row{(\ensuremath{\Pi}_{\Phi}(Z))}{i}}^2 \sum_{j \in \row{\Omega}{i}} \entry{\ensuremath{V}}{j}{k}^2 \\
& = \frobnorm{\ensuremath{\Pi}_{\Phi}Z}^2 \sum_{j \in \row{\Omega}{i}} \twonorm{\row{\ensuremath{V}}{i}}^2 \\
& \leq  \frobnorm{\ensuremath{\Pi}_{\Phi}Z}^2 \frac{3}{2}pd_2 \cdot \twoinfnorm{\ensuremath{V}}^2 \leq 2\mu r p \frobnorm{\ensuremath{\Pi}_{\Phi}Z}^2.
\end{align*}
We thus finish proving \eqref{eq:Zv_sampling}. Inequality \eqref{eq:uZ_sampling} can be proved in the same way.

\subsection{Proof of Lemma \ref{lem:symm}} \label{proof:lem:symm}
Recall that we let $F := [\ensuremath{U}; \ensuremath{V}]$ and $F_{\pi^*} := [\ensuremath{U^*}; \ensuremath{V^*}]\ensuremath{Q}$ for some matrix $\ensuremath{Q} \in \RotateSet{r}$, which minimizes the following function
\begin{equation} \label{eq:opt_Q}
\frobnorm{F - [\ensuremath{U^*}; \ensuremath{V^*}]Q}^2.
\end{equation}
Let $F^* := [\ensuremath{U^*}; \ensuremath{V^*}]$. Expanding the above term, we find that $\ensuremath{Q}$ is the maximizer of $\trinprod{F}{F^*Q} = \text{Tr}(F^{\top}F^{*}Q)$. Suppose $F^{\top}F^{*}$ has SVD with form $Q_1\Lambda Q_2^{\top}$ for $Q_1, Q_2 \in \RotateSet{r}$. When the minimum diagonal term of $\Lambda$ is positive, we conclude that the minimizer of \eqref{eq:opt_Q} is unique and $Q = Q_2Q_1^{\top}$. To prove this argument, we note that
\[
\text{Tr}(F^{\top}F^{*}Q) = \sum_{i \in [r]} \entry{\Lambda}{i}{i}\inprod{p_i}{q_i},
\]
where $p_i$ is the $i$-th column of $Q_1$ and $q_i$ is the $i$-th column of $Q^{\top}Q_2$. Hence, $\text{Tr}(F^{\top}F^{*}Q) \leq \sum_{i \in [r]} \entry{\Lambda}{i}{i}$ and the equality holds if and only if $p_i = q_i$ for all $i \in [r]$ since every $\entry{\Lambda}{i}{i} > 0$. We have $Q_1 = Q^{\top}Q_2$ and thus finish proving the argument.

Under our assumption $\opnorm{F - F_{\pi^*}} < \sqrt{2\sigma_r^*}$, for any nonzero vector $u \in \ensuremath{\mathbb{R}}^r$, we have
\[
\twonorm{F^{\top}F_{\pi^*}u} \geq \twonorm{F_{\pi^*}^{\top}F_{\pi^*}u} - \twonorm{(F_{\pi^*} - F)^{\top}F_{\pi^*}u} \geq (\sqrt{2\sigma_r^*} - \opnorm{F_{\pi^*} - F})\frobnorm{F_{\pi^*}u} > 0.
\]
In the second step, we use the fact that the singular values of $F_{\pi^*}$ are equal to the diagonal terms of $\sqrt{2}\ensuremath{\Sigma}^{*1/2}$. Hence, $F^{\top}F_{\pi^*}$ has full rank. Furthermore, it implies that $F^{\top}F^*$ has full rank and only contains positive singular values.

Proceeding with the proved argument, we have
\[
F^{\top}F_{\pi^*} = Q_1\Lambda Q_2^{\top}Q_2Q_1^{\top} = Q_1\Lambda Q_1^{\top},
\]
which implies that $F^{\top}F_{\pi^*}$ is symmetric. Accordingly, we have $(F - F_{\pi^*})^{\top}F_{\pi^*}$ is also symmetric.

\section*{Acknowledgment}

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

\printbibliography

\newpage

Information Theory (cs.IT); Machine Learning (cs.LG); Statistics Theory (math.ST); Machine Learning (stat.ML)