Distributed Low-rank Subspace Segmentation
Ameet Talwalkar, Lester Mackey, Yadong Mu, Shih-Fu Chang, Michael I. Jordan
Introduction
Visual data, though innately high dimensional, often reside in or lie close to a union of low-dimensional subspaces. These subspaces might reflect physical constraints on the objects comprising images and video (e.g., faces under varying illumination or trajectories of rigid objects ) or naturally occurring variations in production (e.g., digits hand-written by different individuals ). Subspace segmentation techniques model these classes of data by recovering bases for the multiple underlying subspaces . Applications include image clustering , segmentation of images, video, and motion , and affinity graph construction for semi-supervised learning .
One promising, convex formulation of the subspace segmentation problem is the low-rank representation (LRR) program of Liu et al. :
Much of the computational burden in solving LRR stems from the nuclear norm penalty, which is known to encourage low-rank solutions, so one might hope to leverage the large body of past work on parallel and distributed matrix factorization to improve the scalability of LRR. Unfortunately, these techniques are tailored to optimization problems with losses and constraints that decouple across the entries of the input matrix. This decoupling requirement is violated in the LRR problem due to the constraint of Eq. (1), and this non-decomposable constraint introduces new algorithmic and analytic challenges that do not arise in decomposable matrix factorization problems.
To address these challenges, we develop, analyze, and evaluate a provably accurate divide-and-conquer approach to large-scale subspace segmentation that specifically accounts for the non-decomposable structure of the LRR problem. Our contributions are three-fold:
Algorithm: We introduce a parallel, divide-and-conquer approximation algorithm for LRR that is suitable for large-scale subspace segmentation problems. Scalability is achieved by dividing the original LRR problem into computationally tractable and communication-free subproblems, solving the subproblems in parallel, and combining the results using a technique from randomized matrix approximation. Our algorithm, which we call DFC-LRR, is based on the principles of the Divide-Factor-Combine (DFC) framework for decomposable matrix factorization but can cope with the non-decomposable constraints of LRR.
Analysis: We characterize the segmentation behavior of our new algorithm, showing that DFC-LRR maintains the segmentation guarantees of the original LRR algorithm with high probability, even while enjoying substantial speed-ups over its namesake. Our new analysis features a significant broadening of the original LRR theory to treat the richer class of LRR-type subproblems that arise in DFC-LRR. Moreover, since our ultimate goal is subspace segmentation and not matrix recovery, our theory guarantees correctness under a more substantial reduction of problem complexity than the work of (see Sec. 3.2 for more details).
Applications: We first present results on face clustering and synthetic subspace segmentation to demonstrate that DFC-LRR achieves accuracy comparable to LRR in a fraction of the time. We then propose and validate a novel application of the LRR methodology to large-scale graph-based semi-supervised learning. While LRR has been used to construct affinity graphs for semi-supervised learning in the past , prior attempts have failed to scale to the sizes of real-world datasets. Leveraging the favorable computational properties of DFC-LRR, we propose a scalable strategy for constructing such subspace affinity graphs. We apply our methodology to a variety of computer vision tasks – multimedia event detection, concept detection, and image tagging – demonstrating an order of magnitude improvement in speed and accuracy that exceeds the state of the art.
The remainder of the paper is organized as follows. In Section 2 we first review the low-rank representation approach to subspace segmentation and then introduce our novel DFC-LRR algorithm. Next, we present our theoretical analysis of DFC-LRR in Section 3. Section 4 highlights the accuracy and efficiency of DFC-LRR on a variety of computer vision tasks. We present subspace segmentation results on simulated and real-world data in Section 4.1. In Section 4.2 we present our novel application of DFC-LRR to graph-based semi-supervised learning problems, and we conclude in Section 5.
Divide-and-Conquer Segmentation
In this section, we review the LRR approach to subspace segmentation and present our novel algorithm, DFC-LRR.
The LRR approach of seeks to recover the row space of by solving the convex optimization problem presented in Eq. (1). Importantly, the LRR solution comes with a guarantee of correctness: the column space of is exactly equal to the row space of whenever certain technical conditions are met (see Sec. 3 for more details).
Moreover, as we will show in this work, LRR is also well-suited to the construction of affinity graphs for semi-supervised learning. In this setting, the goal is to define an affinity graph in which nodes correspond to data points and edge weights exist between nodes drawn from the same subspace. LRR can thus be used to recover the block-sparse structure of the graph’s affinity matrix, and these affinities can be used for semi-supervised label propagation.
2 Divide-Factor-Combine LRR (DFC-LRR)
We now present our scalable divide-and-conquer algorithm, called DFC-LRR, for LRR-based subspace segmentation. DFC-LRR extends the principles of the DFC framework of to a new non-decomposable problem. The DFC-LRR algorithm is summarized in Algorithm 1, and we next describe each step in further detail. D step - Divide input matrix into submatrices: DFC-LRR randomly partitions the columns of into -column submatrices, . For simplicity, we assume that divides evenly. F step - Factor submatrices in parallel: DFC-LRR solves subproblems in parallel. The th LRR subproblem is of the form
where the input matrix is used as a dictionary but only a subset of columns is used as the observations.An alternative formulation involves replacing both instances of with in Eq. (1). The resulting low-rank estimate would have dimensions , and the C step of DFC-LRR would compute a low-rank approximation on the block-diagonal matrix diag(). A typical LRR algorithm can be easily modified to solve Eq. (2) and will return a low-rank estimate in factored form. C step - Combine submatrix estimates: DFC-LRR generates a final approximation to the low-rank LRR solution by projecting onto the column space of . This column projection technique is commonly used to produce randomized low-rank matrix factorizations and was also employed by the DFC-Proj algorithm of . Runtime: As noted in , many state-of-the-art solvers for nuclear-norm regularized problems like Eq. (1) have per-iteration time complexity due to the rank- truncated SVD required on each iteration. DFC-LRR reduces this per-iteration complexity significantly and requires just O time for the th subproblem. Performing the subsequent column projection step is relatively cheap computationally, since an LRR solver can return its solution in factored form. Indeed, if we define , then the column projection step of DFC-LRR requires only O time.
Theoretical Analysis
Despite the significant reduction in computational complexity, DFC-LRR provably maintains the strong theoretical guarantees of the LRR algorithm. To make this statement precise, we first review the technical conditions for accurate row space recovery required by LRR.
The LRR analysis of Liu et al. relies on two key quantities, the rank of the clean data matrix and the coherence of the singular vectors . We combine these properties into a single definition:
Intuitively, when the coherence is small, information is well-distributed across the rows of a matrix, and the row space is easier to recover from outlier corruption. Using these properties, Liu et al. established the following recovery guarantee for LRR.
In other words, LRR can exactly recover the row space of even when a constant fraction of the columns has been corrupted by outliers. As the rank and coherence shrink, grows allowing greater outlier tolerance.
2 High Probability Subspace Segmentation
Our main theoretical result shows that, with high probability and under the same conditions that guarantee the accuracy of LRR, DFC-LRR also exactly recovers the row space of . Recall that in our independent subspace setting accurate row space recovery is tantamount to correct segmentation of the columns of . The proof of our result, which generalizes the LRR analysis of to a broader class of optimization problems and adapts the DFC analysis of , can be found in the appendix.
Fix any failure probability . Under the conditions of Thm, 2, let be a solution returned by DFC-LRR. Then there exists a constant (depending on and ) for which the column space of exactly equals the row space of whenever for each DFC-LRR subproblem, , and for
Thm. 3 establishes that, like LRR, DFC-LRR can tolerate a constant fraction of its data points being corrupted and still recover the correct subspace segmentation of the clean data points with high probability. When the number of datapoints is large, solving LRR directly may be prohibitive, but DFC-LRR need only solve a collection of small, tractable subproblems. Indeed, Thm. 3 guarantees high probability recovery for DFC-LRR even when the subproblem size is logarithmic in . The corresponding reduction in computational complexity allows DFC-LRR to scale to large problems with little sacrifice in accuracy.
Notably, this column sampling complexity is better than that established by in the matrix factorization setting: we require O columns sampled, while requires in the worst case columns for matrix completion and for robust matrix factorization.
Experiments
We now explore the empirical performance of DFC-LRR on a variety of simulated and real-world datasets, first for the traditional task of robust subspace segmentation and next for the more complex task of graph-based semi-supervised learning. Our experiments are designed to show the effectiveness of DFC-LRR both when the theory of Section 3 holds and when it is violated. Our synthetic datasets satisfy the theoretical assumptions of low rank, incoherence, and a small fraction of corrupted columns, while our real-world datasets violate these criteria.
For all of our experiments we use the inexact Augmented Lagrange Multiplier (ALM) algorithm of as our base LRR algorithm. For the subspace segmentation experiments, we set the regularization parameter to the values suggested in previous works , while in our semi-supervised learning experiments we set it to as suggested in prior work.http://perception.csl.illinois.edu/matrix-rank In all experiments we report parallel running times for DFC-LRR, i.e., the time of the longest running subproblem plus the time required to combine submatrix estimates via column projection. All experiments were implemented in Matlab. The simulation studies were run on an x- architecture using a single Ghz core and GB of main memory, while the real data experiments were performed on an x- architecture equipped with a 2.67GHz 12-core CPU and 64GB of main memory.
We first aim to verify that DFC-LRR produces accuracy comparable to LRR in significantly less time, both in synthetic and real-world settings. We focus on the standard robust subspace segmentation task of identifying the subspace associated with each input datapoint.
In our first experiments we fix , , , and , set the regularizer to , and vary the fraction of outliers. We measure with what frequency LRR and DFC-LRR are able to recover of the row space of and identify the outlier columns in , using the same criterion as defined in .Success is determined by whether the oracle constraints of Eq. (8) in the Appendix are satisfied within a tolerance of . Figure 1(a) shows average performance over trials. We see that DFC-LRR performs quite well, as the gaps in the phase transitions between LRR and DFC-LRR are small when sampling of the columns (i.e., ) and are virtually non-existent when sampling of the columns (i.e., ).
Figure 1(b) shows corresponding timing results for the accuracy results presented in Figure 1(a). These timing results show substantial speedups in DFC-LRR relative to LRR with a modest tradeoff in accuracy as denoted in Figure 1(a). Note that we only report timing results for values of for which DFC-LRR was successful in all trials, i.e., for which the success rate equaled in Figure 1(a). Moreover, Figure 1(c) shows timing results using the same parameter values, except with a fixed fraction of outliers () and a variable number of samples in each subspace, i.e., ranges from to . These timing results also show speedups with minimal loss of accuracy, as in all of these timing experiments, LRR and DFC-LRR were successful in all trials using the same criterion defined in and used in our phase transition experiments of Figure 1(a).
1.2 Face Clustering
We next demonstrate the comparable quality and increased performance of DFC-LRR relative to LRR on real data, namely, a subset of Extended Yale Database B,http://vision.ucsd.edu/~leekc/ExtYaleDatabase a standard face benchmarking dataset. Following the experimental setup in , frontal face images of human subjects are chosen, each of which is resized to be pixels and forms a 2016-dimensional feature vector. As noted in previous work , a low-dimensional subspace can be effectively used to model face images from one person, and hence face clustering is a natural application of subspace segmentation. Moreover, as illustrated in Figure 2, a significant portion of the faces in this dataset are “corrupted” by shadows, and hence this collection of images is an ideal benchmark for robust subspace segmentation.
As in , we use the feature vector representation of these images to create a dictionary matrix, , and run both LRR and DFC-LRR with the parameter set to . Next, we use the resulting low-rank coefficient matrix to compute an affinity matrix , where contains the top left singular vectors of . The affinity matrix is used to cluster the data into clusters (corresponding to the human subjects) via spectral embedding (to obtain a D feature representation) followed by -means. Following , the comparison of different clustering methods relies on segmentation accuracy. Each of the clusters is assigned a label based on majority vote of the ground truth labels of the points assigned to the cluster. We evaluate clustering performance of both LRR and DFC-LRR by computing segmentation accuracy as in , i.e., each cluster is assigned a label based on majority vote of the ground truth labels of the points assigned to the cluster. The segmentation accuracy is then computed by averaging the percentage of correctly classified data over all classes.
Figures 3(a) and 3(b) show the computation time and the segmentation accuracy, respectively, for LRR and for DFC-LRR with varying numbers of subproblems (i.e., values of ). On this relatively-small data set ( faces), LRR requires over minutes to converge. DFC-LRR demonstrates a roughly linear computational speedup as a function of , comparable accuracies to LRR for smaller values of and a quite gradual decrease in accuracy for larger .
2 Graph-based Semi-Supervised Learning
Graph representations, in which samples are vertices and weighted edges express affinity relationships between samples, are crucial in various computer vision tasks. Classical graph construction methods separately calculate the outgoing edges for each sample. This local strategy makes the graph vulnerable to contaminated data or outliers. Recent work in computer vision has illustrated the utility of global graph construction strategies using graph Laplacian or matrix low-rank based regularizers. L1 regularization has also been effectively used to encourage sparse graph construction . Building upon the success of global construction methods and noting the connection between subspace segmentation and graph construction as described in Section 2.1, we present a novel application of the low-rank representation methodology, relying on our DFC-LRR algorithm to scalably yield a sparse, low-rank graph (SLR-graph). We present a variety of results on large-scale semi-supervised learning visual classification tasks and provide a detailed comparison with leading baseline algorithms.
We adopt the following three large-scale benchmarks:
Columbia Consumer Video (CCV) Content Detectionhttp://www.ee.columbia.edu/ln/dvmm/CCV/: Compiled to stimulate research on recognizing highly-diverse visual content in unconstrained videos, this dataset consists of YouTube videos over semantic categories (e.g., baseball, beach, music performance). Three popular audio/visual features (5000-D SIFT, 5000-D STIP, and 4000-D MFCC) are extracted.
MED12 Multimedia Event Detection: The MED12 video corpus consists of K multimedia videos, with an average duration of minutes, and is used for detecting 20 specific semantic events. For each event, to videos are provided as positive examples, and the remainder of the videos are “null” videos that do not correspond to any event. In this work, we keep all positive examples and sample K null videos, resulting in a dataset of videos. We extract six features from each video, first at sampled frames and then accumulated to obtain video-level representations. The features are either visual (1000-D sparse-SIFT, 1000-D dense-SIFT, 1500-D color-SIFT, 5000-D STIP), audio (2000-D MFCC), or semantic features (2659-D CLASSEME ).
NUS-WIDE-Lite Image Tagging: NUS-WIDE is among the largest available image tagging benchmarks, consisting of over K crawled images from Flickr that are associated with over K user-provided tags. Ground-truth images are manually provided for selected concept tags. We generate a lite version by sampling K images. For each image, 128-D wavelet texture, 225-D block-wise LAB-based color moments and 500-D bag of visual words are extracted, normalized and finally concatenated to form a single feature representation for the image.
2.2 Graph Construction Algorithms
The three graph construction schemes we evaluate are described below. Note that we exclude other baselines (e.g., NNLRS , LLE graph , L1-graph ) due to either scalability concerns or because prior work has already demonstrated inferior performance relative to the SPG algorithm defined below .
NN-graph: We construct a nearest neighbor graph by connecting (via undirected edges) each vertex to its nearest neighbors in terms of distance in the specified feature space. Exponential weights are associated with edges, i.e., w_{ij}=\operatorname{exp}\mathopen{}\mathclose{{}\left(-d_{ij}^{2}/\sigma^{2}}\right), where is the distance between and and is an empirically-tuned parameter .
SPG: Cheng et al. proposed a noise-resistant L1-graph which encourages sparse vertex connectedness, motivated by the work of sparse representation . Subsequent work, entitled sparse probability graph (SPG) enforced positive graph weights. Following the approach of , we implemented a variant of SPG by solving the following optimization problem for each sample:
where is a feature representation of a sample and is the basis matrix for constructed from its nearest neighbors. We use an open-source toolhttp://sparselab.stanford.edu to solve this non-negative Lasso problem.
2.3 Experimental Design
For each benchmarking dataset, we first construct graphs by treating sample images/videos as vertices and using the three algorithms outlined in Section 4.2.2 to create (sparse) weighted edges between vertices. For fair comparison, we use the same parameter settings, namely and for both SPG and SLR-graph. Moreover, we set for NN-graph after tuning over the range through .
We then use a given graph structure to perform semi-supervised label propagation using an efficient label propagation algorithm that enjoys a closed-form solution and often achieves the state-of-the-art performance. We perform a separate label propagation for each category in our benchmark, i.e., we run a series of binary classification label propagation experiments for CCV/MED12 and experiments for NUS-WIDE-Lite. For each category, we randomly select half of the samples as training points (and use their ground truth labels for label propagation) and use the remaining half as a test set. We repeat this process times for each category with different random splits. Finally, we compute Mean Average Precision (mAP) based on the results on the test sets across all runs of label propagation.
2.4 Experimental Results
We first performed experiments using the CCV benchmark, the smallest of our datasets, to explore the tradeoff between computation and accuracy when using DFC-LRR as part of our proposed SLR-graph. Figure 4(a) presents the time required to run SLR-graph with LRR versus DFC-LRR with three different numbers of subproblems (), while Figure 4(b) presents the corresponding accuracy results. The figures show that DFC-LRR performs comparably to LRR for smaller values of , and performance gradually degrades for larger . Moreover, DFC-LRR is up to two orders of magnitude faster and achieves superlinear speedups relative to LRR.We restricted the maximum number of internal LRR iterations to to ensure that LRR ran to completion in less than two days. Given the scalability issues of LRR on this modest-sized dataset, along with the comparable accuracy of DFC-LRR, we ran SLR-graph exclusively with DFC-LRR () for our two larger datasets.
Table 1 summarizes the results of our semi-supervised learning experiments using the three graph construction techniques defined in Section 4.2.2. The results show that our proposed SLR-graph approach leads to significant performance gains in terms of mAP across all benchmarking datasets for the vast majority of features. These results demonstrate the benefit of enforcing both low-rankedness and sparsity during graph construction. Moreover, conventional low-rank oriented algorithms, e.g., would be computationally infeasible on our benchmarking datasets, thus highlighting the utility of employing DFC’s divide-and-conquer approach to generate a scalable algorithm.
Conclusion
Our primary goal in this work was to introduce a provably accurate algorithm suitable for large-scale low-rank subspace segmentation. While some contemporaneous work also aims at scalable subspace segmentation, this method offers no guarantee of correctness. In contrast, DFC-LRR provably preserves the theoretical recovery guarantees of the LRR program. Moreover, our divide-and-conquer approach achieves empirical accuracy comparable to state-of-the-art methods while obtaining linear to superlinear computational gains, both on standard subspace segmentation tasks and on novel applications to semi-supervised learning. DFC-LRR also lays the groundwork for scaling up LRR derivatives known to offer improved performance, e.g., LatLRR in the setting of standard subspace segmentation and NNLRS in the graph-based semi-supervised learning setting. The same techniques may prove useful in developing scalable approximations to other convex formulations for subspace segmentation, e.g., .
References
Appendix A Proof of Theorem 3
Our proof of Thm. 3 rests upon three key results: a new deterministic recovery guarantee for LRR-type problems that generalizes the guarantee of , a probabilistic estimation guarantee for column projection established in , and a probabilistic guarantee of showing that a uniformly chosen submatrix of a -coherent matrix is nearly -coherent. These results are presented in Secs. A.1, A.2, and A.3 respectively. The proof of Thm. 3 follows in Sec. A.4.
In what follows, the unadorned norm represents the spectral norm of a matrix. We will also make use of a technical condition, introduced by Liu et al. to ensure that a corrupted data matrix is well-behaved when used as a dictionary:
A matrix is -RWD if
A larger value of corresponds to improved recovery properties.
Thm. 1 of analyzes LRR recovery under the constraint when the observation matrix and the dictionary are both equal to the input matrix . Our next theorem provides a comparable analysis when the observation matrix is a column submatrix of the dictionary.
and let be a solution to the problem
with . If , then the column space of equals the row space of .
The proof of Thm. 5 can be found in Sec. B.
A.2 Analysis of Column Projection
The following lemma, due to , shows that, with high probability, column projection exactly recovers a -coherent matrix by sampling a number of columns proportional to .
exactly with probability at least .
A.3 Conservation of Incoherence
The following lemma of shows that, with high probability, captures the full rank of and has coherence not much larger than .
A.4 Proof of DFC-LRR Guarantee
Recall that, under Alg. 1, the input matrix has been partitioned into column submatrices . Let and be the corresponding partitions of and , let be the size of the column support of for each index , and let be a solution to the th DFC-LRR subproblem.
For each index , we further define as the event that is -coherent, as the event that , and as the event that the column space of the matrix is equal to the row space of . Under our choice of , Thm. 5 implies that holds when and are both realized. Hence, when and hold for all indices , the column space of precisely equals the row space of , and the median rank of equals .
shows that, given and for all indices , equals with probability at least . To establish with probability at least , it therefore remains to show that
by our assumption that .
each submatrix is -coherent with probability at least . A second application of Hoeffding’s inequality for the hypergeometric further implies that
since . Hence, {\mathbf{P}}\mathopen{}\mathclose{{}\left({A_{i}^{c}}}\right)\leq\delta/(2t).
Appendix B Proof of Theorem 5
Our proof of Thm. 5 will parallel Thm. 1 of . We begin by introducing two oracle constraints that would guarantee the desired outcome if satisfied.
Under the assumptions of Thm. 5, suppose that for some matrices . If additionally satisfy the oracle constraints
then the column space of equals the row space of .
B.2 Conditions for Optimality
To this end, we derive sufficient conditions for solving Eq. 4 and moreover show that if any solution to Eq. 4 satisfies the oracle constraints of Eq. 8, then all solutions do.
Under the assumptions of Thm. 5, suppose that for some matrices . If there exists a matrix satisfying
then is a solution to Eq. 4. If, in addition, , and satisfy the oracle constraints of Eq. 8, then all solutions to Eq. 4 satisfy the oracle constraints of Eq. 8.
Proof The proof of this theorem is identical to that of [18, Thm. 3] which establishes the same result when the observation is replaced by . ∎ It remains to construct a feasible pair satisfying the oracle constraints and and a dual certificate satisfying the conditions of Thm. 9.
B.3 Constructing a Dual Certificate
To this end, we consider the oracle problem:
Let be the binary matrix that selects the columns of from . Then is feasible for this problem, and hence an optimal solution must exist. By explicitly constructing a dual certificate , we will show that also solves the LRR subproblem of Eq. 4.
We will need a variety of lemmas paralleling those developed in . Let
Let . Then
Proof The proof is identical to that of Lem. 9 of . ∎
.
Proof The proof is identical to that of Lem. 10 of , save for the size of , which is now bounded by . ∎ Note that under the assumption , we have .
If , then .
Lem. 12 of is unchanged in our setting. The next lemma parallels Lem. 13 of .
and therefore that is of full row rank. The remainder of the proof is identical to that of Lem. 13 of , save for the coherence factor of in place of . ∎
where the first relation follows from Lem. 11. Our final theorem parallels Thm. 4 of .
then satisfies the conditions in Thm. 9.
Proof The proof of property requires a small modification. Thm. 4 of establishes that . To conclude that , we note that and that the column space of contains the column space of by assumption. Hence, and therefore .
The proofs of properties and are unchanged except for the dimensionality factor which changes from to . ∎
Finally, Lem. 14 of guarantees that the preconditions of Thm. 15 are met under our assumptions on and .