Graph Sampling for Matrix Completion Using Recurrent Gershgorin Disc Shift
Fen Wang, Yongchao Wang, Gene Cheung, Cheng Yang

TL;DR
This paper introduces a novel graph signal processing-based sampling strategy for matrix completion, optimizing sample selection to improve stability and accuracy while maintaining computational efficiency.
Contribution
It proposes a greedy sampling method leveraging Gershgorin circle theorem and eigenvector analysis to select informative samples for matrix completion.
Findings
Outperforms existing sampling schemes in reducing completion error.
Efficiently scales to large matrices through block-diagonal decomposition.
Enhances stability of the linear system via eigenvalue maximization.
Abstract
Matrix completion algorithms fill missing entries in a large matrix given a subset of observed samples. However, how to best pre-select informative matrix entries given a sampling budget is largely unaddressed. In this paper, we propose a fast sample selection strategy for matrix completion from a graph signal processing perspective. Specifically, we first regularize the matrix reconstruction objective using a dual graph signal smoothness prior, resulting in a system of linear equations for solution. We then select appropriate samples to maximize the smallest eigenvalue of the coefficient matrix, thus maximizing the stability of the linear system. To efficiently solve this combinatorial problem, we derive a greedy sampling strategy, leveraging on Gershgorin circle theorem, that iteratively selects one sample (equivalent to shifting one Gershgorin disc) at a time…
| dataset | Exp. | users | items | features | entries | density | entry levels |
| Synthetic Netflix [3] | Fig. 3 | 200 | 100 | - | 20,000 | 100% | 1,2,…,5 |
| ML100K [52] | Fig. 3 | 100 | 200 | - | 12,566 | 62.83% | 1,2,…,5 |
| Tab. II/III | 943 | 1682 | ✓ | 100,000 | 6.3% | 1,2,…,5 | |
| ML10M [52] | Fig. 3 | 100 | 200 | - | 18,119 | 90.6% | 0.5,1,…,5 |
| Tab. IV | 1000 | 500 | - | 345,904 | 69.18% | 0.5,1,…,5 | |
| Douban [26] | Tab. IV | 3000 | 3000 | - | 136,891 | 1.52% | 1,2,…,5 |
| Flixster [26] | Tab. IV | 3000 | 3000 | - | 26,173 | 0.29% | 0.5,1,…,5 |
| YahooMusic [26] | Tab. IV | 3000 | 3000 | - | 5,335 | 0.06% | 1,2,…,100 |
| Book-Crossing [53] | Tab. IV | 1000 | 1000 | - | 3,166 | 0.32% | 1,2,…,10 |
| Jester [54] | Tab. IV | 1000 | 100 | - | 73,320 | 73.32% | (0,1) |
| ML1M [52] | Tab. IV | 6040 | 3706 | ✓ | 1,000,209 | 4.47% | 1,2,…,5 |
| FilmTrust [55] | Tab. IV | 1000 | 1000 | - | 31,880 | 3.19% | 0.5,1,…,4 |
| MC | LSS | eigs | |||||
| G1 | GRALS | 0.962 | 0.931 | 0.927 | 0.935 | 0.934 | 0.931 |
| GC-MC | 0.910 | 0.896 | 0.889 | 0.895 | 0.897 | 0.891 | |
| NMC | 0.891 | 0.907 | 0.880 | 0.888 | 0.889 | 0.886 | |
| Time (s) | - | 1.975 | 1.104 | 0.503 | 0.375 | 0.320 | |
| G2 | GRALS | 0.958 | 0.889 | 0.871 | 0.870 | 0.882 | 0.882 |
| GC-MC | 0.909 | 0.860 | 0.839 | 0.840 | 0.847 | 0.851 | |
| NMC | 0.907 | 0.858 | 0.840 | 0.845 | 0.843 | 0.852 | |
| Time (s) | - | 1.278 | 1.216 | 0.573 | 0.441 | 0.388 |
| dataset | random | LSS | ||||
| Flixster | 1.029 | 1.207 | 0.932 | 1.057 | 1.046 | 1.045 |
| Douban | 0.744 | 0.750 | 0.715 | 0.720 | 0.736 | 0.730 |
| YahooMusic | 96.987 | 125.0 | 59.172 | 44.546 | 52.391 | 47.082 |
| ML1M | 0.905 | 0.930 | 0.829 | 0.833 | 0.835 | 0.838 |
| Book-Crossing | 3.987 | 5.095 | 3.578 | 3.704 | 3.804 | 4.185 |
| ML10M | 0.706 | 0.777 | 0.655 | 0.656 | 0.656 | 0.656 |
| Jester | 0.214 | 0.217 | 0.160 | 0.162 | 0.162 | 0.165 |
| FilmTrust | 0.820 | 0.941 | 0.668 | 0.735 | 0.711 | 0.742 |
Peer Reviews
No public reviews on file for this paper yet. If you reviewed it on a platform where reviews are public (OpenReview, ICLR, NeurIPS, ICML), you can paste yours below so the community can read it here.
Videos
No videos yet. Explain this paper in a talk, walkthrough, or lecture? Add one.
Graph Sampling for Matrix Completion Using Recurrent Gershgorin Disc Shift
Fen Wang, Yongchao Wang, Member, IEEE, Gene Cheung, Senior Member, IEEE and Cheng Yang, Member, IEEE F. Wang conducted this work during her visit to York University under the scholarship from China Scholarship Council. *(Corresponding author: Yongchao Wang and Gene Cheung.)*F. Wang and Y. Wang are with State Key Laboratory of ISN, Xidian University, Xi’an 710071, Shaanxi, China (e-mail: [email protected]; [email protected]).G. Cheung and C. Yang are with the department of EECS, York University, 4700 Keele Street, Toronto, M3J 1P3, Canada (e-mail:[email protected]; [email protected]).
Abstract
Matrix completion algorithms fill missing entries in a large matrix given a subset of observed samples. However, how to best pre-select informative matrix entries given a sampling budget is largely unaddressed. In this paper, we propose a fast sample selection strategy for matrix completion from a graph signal processing perspective. Specifically, we first regularize the matrix reconstruction objective using a dual graph signal smoothness prior, resulting in a system of linear equations for solution. We then select appropriate samples to maximize the smallest eigenvalue of the coefficient matrix, thus maximizing the stability of the linear system. To efficiently solve this combinatorial problem, we derive a greedy sampling strategy, leveraging on Gershgorin circle theorem, that iteratively selects one sample (equivalent to shifting one Gershgorin disc) at a time corresponding to the largest magnitude entry in the first eigenvector of a modified graph Laplacian matrix. Our algorithm benefits computationally from warm start as the first eigenvectors of incremented Laplacian matrices are computed recurrently for more samples. To achieve computation scalability when sampling large matrices, we further rewrite the coefficient matrix as a sum of two separate components, each of which exhibits block-diagonal structure that we exploit for alternating block-wise sampling. Extensive experiments on both synthetic and real-world datasets show that our graph sampling algorithm substantially outperforms existing sampling schemes for matrix completion and reduces the completion error, when combined with a range of modern matrix completion algorithms.
Index Terms:
Graph sampling, matrix completion, Gershgorin circle theorem, graph Laplacian regularization
I Introduction
Big data means not only that the volume of acquired data is large, but the dimensionality of the dataset is also considerable. Matrix completion (MC) [1] is an example of this “curse of dimensionality” problem, where two large dimensional item sets (e.g., viewers and movies in the famed Netflix challenge) are correlated within and across sets. Specifically, given a small subset of pairwise observations (e.g., viewers’ ratings on movies), an MC algorithm reconstructs missing entries in the target matrix signal. Many MC algorithms have been devised using different priors to regularize the under-determined inverse problem, such as low rank of the target matrix [2] and graph signal smoothness priors [3]. See [4] for an introductory exposition.
While MC has been investigated intensively, how to pre-select matrix entries to collect informative samples given a sampling budget is largely unaddressed. This sampling problem is of practical concern for applications where sampling is expensive and/or time-consuming [5] (e.g., requesting viewers to fill out movie surveys is cumbersome and costly). Conventionally, entries in matrix were collected based on their informative uncertainty computed using different methods [6, 7, 8], which are typically computation-intensive. There exist fast sampling strategies for coherent matrices that assigned each entry a probability for non-uniform random sampling [9]. However, performance of random selection schemes is in general inferior compared to their deterministic counterparts.
Recently, rating matrix in a recommendation system was investigated from a graph signal processing (GSP) perspective [10], where the target matrix signal was assumed to be bandlimited / smooth with respect to both the row (movies) and column (viewers) graphs (called factor graphs), [11, 3]. Under this assumption, [12, 13] identified a structured set for MC by sampling entries that are intersections of greedily selected rows and columns. However, the imposed structure severely limits the possible sampling patterns and thus is too restrictive to achieve a general sampling budget. More general sampling methods for single graphs are not applicable for MC due to their high complexities on the product graph (one large graph containing all matrix entries as nodes) [14].
In contrast, in this paper we propose a fast unstructured graph sampling method for MC. We first regularize the sampling objective with a dual graph smoothness prior—a generalization of the well-known Tikhonov regularizer [15] to the graph signal domain—which was shown effective in completing missing matrix entries previously [3, 16]. This formulation leads to a system of linear equations for solution, which can be computed efficiently using known numerical linear algebra algorithms such as conjugate gradient (CG) [17]. To maximize the stability of the linear system, we select samples to maximize the smallest eigenvalue of the coefficient matrix 111Maximizing of a matrix is also known as the E-optimality criterion in optimal design of experiments [18, 19, 20]., which we show to also mean minimizing the upper bound of the reconstructed matrix signal’s squared error.
We propose to optimize the formulated objective greedily: select one node at a time such that the current sample set results in the largest . However, in each greedy step, computing for all candidates and choosing the largest one would still be expensive. Instead, leveraging on an insightful corollary of the Gershgorin circle theorem [21], we greedily select the sample corresponding to the largest magnitude entry in the first eigenvector of an augmented coefficient matrix, which also minimizes a related objective. Our algorithm benefits from warm start as the first eigenvectors of incrementally updated Laplacian matrices are computed recurrently during sampling using the well-known locally optimal block preconditioned conjugated gradient (LOBPCG) method [22].
To achieve computation scalability when sampling large matrices, we further partition the coefficient matrix into two matrices, each exhibiting attractive block-diagonal structure after permutation. We then propose an iterative sampling strategy that efficiently collects a pre-determined number of samples block-wise on smaller blocks alternately. Extensive experiments on synthetic and real-world datasets show that our proposed graph sampling methods achieve much smaller RMSE than competing sampling schemes for MC [12, 23, 24], when combined with a variety of state-of-the-art MC methods [25, 26, 27].
The outline of the paper is as follows. We first overview related works in graph sampling and active matrix completion in Section II. We then derive our graph sampling objective for MC using the dual graph smoothness prior in Section III. In Section IV, we describe our sampling strategy via the Gershgorin circle theorem, and then we propose an iterative block-wise sampling scheme for large matrices in Section V. Finally, extensive experiments and conclusion are presented in Section VII and VIII, respectively.
II Related works
We first discuss related works in graph sampling. Then we review some literature in conventional active matrix completion domain.
II-A *Subset Sampling of Graph Signals *
Subset sampling of graph signals is a fundamental problem in GSP [10]: how to select a node subset in a graph for sampling such that the remaining samples in the signal can be reconstructed with high accuracy. Most existing works [28, 29, 18, 14, 30, 31, 24, 12] extended the notion of critical sampling (also known as Nyquist sampling) in regular data kernels to bandlimited / smooth signals on graphs, where graph frequencies are defined as eigenvalues of a graph variation operator like the graph Laplacian or adjacency matrix. Sampling methods in GSP can be broadly divided into two categories: i) deterministic schemes [28, 29, 18, 14, 30, 32], and ii) random schemes [31, 24].
Most deterministic schemes [18, 33, 32] assumed that the graph signal is bandlimited: its spectral coefficients are concentrated on a set of extreme eigenvectors. [14] proposed a lightweight sampling method using the notion of spectral proxies, which collected samples based on the first eigenvector of a submatrix in each greedy step. One recent work [30] avoided eigenvector computation via Neumann series expansion, but required a large number of matrix series multiplications for accurate approximation. Recently, [34] proposed a sampling method based on Gershgorin disc alignment without any explicit eigen-decompositions. [23] also proposed an eigen-decomposition-free sampling method based on localization operator’s coverage surface, but had no notion of global errors in its optimization objective. However, those sampling methods cannot be directly applied to real-world MC problem because of their high complexities on the corresponding product graph.
In parallel, [24] proposed a non-uniform random graph sampling scheme to select nodes based on the notion of graph coherence, such that each node was sampled with a designed probability. However, the performance of random sampling is generally inferior compared to its deterministic competitors.
To the best of our knowledge, we are the first to propose an unstructured and deterministic graph sampling strategy specifically for MC in the literature, with complexity roughly linear to the size of the factor graphs.
II-B Active Matrix Completion
In the active learning literature, strategically selecting matrix entries is also called active matrix completion [35]. Active matrix completion approaches can be categorized into two types: i) statistical approach [6, 35], and ii) GSP approach [12, 13].
Among works pursuing a statistical approach, [35] formulated an active learning objective for MC, which was tackled using collaborative filtering. In [6], three different active querying strategies were proposed based on the reconstruction uncertainty of each entry; this querying idea was further investigated in [36, 37, 8]. However, those methods are generally computation-expensive since they must evaluate all candidates based on expected reconstructed error. Separately, adaptive sensing was proposed in [7] to select a subset of informative columns for MC with bounded complexity. Using the coherent property of matrix signals, [9] proposed a fast leveraged score-based sampling (LSS) to assign each matrix entry a sampling probability for non-uniform random sampling. Nevertheless, the performance of random sampling is not comparable to the deterministic strategies.
Recently, from a GSP viewpoint, the target matrix was interpreted as a bandlimited signal on the two factor graphs [11]. Based on such bandlimited model, [12, 13] proposed an efficient structured graph sampling strategy for MC, and then extended it to multidimensional tensor graph signals, whose effectiveness has been validated in recommendation system and point cloud sampling. However, structured sampling—selected samples must correspond to matrix entries that are intersections of chosen rows and columns of the matrix—is too restrictive to achieve arbitrary sampling budgets. In this paper, we propose a fast unstructured graph sampling strategy for MC, with comparable complexity to the structured counterpart [12, 13].
III Problem Formulation
We derive an objective function for matrix sampling using a dual graph signal smoothness prior [3, 16]. We first define the graph-based MC problem in Section III-A, and then formulate the graph spectral matrix sampling problem in Section III-B.
III-A Dual Graph Smoothness based Matrix Completion
Denote the original matrix signal and additive noise by and respectively, where . Given a sampling set , its corresponding sampling operator can be defined as
[TABLE]
With the above notations, the sampled noise-corrupted observation is , where denotes the element-wise matrix multiplication operator. We assume that elements in noise are zero-mean, independent and identically distributed (i.i.d.) noise with the same variance.
MC methods attempt to reconstruct the original matrix from the partial noisy observations , under an assumed prior for matrix , like low-rank [2]:
[TABLE]
where is set sufficiently small to enforce similar reconstruction of the observed samples in signal .
Recently, [3] introduced a dual graph smoothness prior to promote low rank matrix reconstruction. Specifically, columns of are assumed to be smooth with respect to an undirected weighted row graph with vertices and edges . Weight matrix specifies pairwise similarities among vertices in . The combinatorial graph Laplacian matrix of row graph is , where the degree matrix is a diagonal matrix with entries . Taking the -th column of , denoted by , as an example, the total graph variation of on graph is defined as [38]:
[TABLE]
Thus a smaller variation value would mean similar sample reconstructions between strongly connected nodes.
Similarly, the rows of are assumed smooth with respect to a column graph with vertices , edges and weight matrix . Corresponding graph Laplacian matrix for the column graph is . Using the movie recommendation systems as an example, the row graph is a similarity graph among movies, and the column graph is a social relationship graph among viewers. Row and column graphs can be constructed from observed data using different methods [3, 39, 40]; we describe our adopted graph construction schemes in Section VI.
We now formulate the MC problem with dual graph Laplacian regularization (DGLR) [41] as follows:
[TABLE]
where and are parameters trading off the first fidelity term with the two signal smoothness priors.
It has been shown through extensive experiments that the dual graph signal smoothness prior enables good MC performance [3]. More generally, a graph smoothness prior is a generalization of the well-known Tikhonov regularizer—popular regularization for ill-posed problems—to graph data kernels [10]. In the fast growing field of GSP [38], the graph smoothness prior has already been shown effective empirically for a wide range of inverse problems (e.g., image denoising / deblurring [42, 41], point cloud denoising [43, 44]), and recently is successfully applied to MC also [11, 3, 16, 13]. Next, we derive our sampling algorithm based on this well-accepted prior in the GSP community.
To solve the unconstrained QP problem (4), we take the derivative of with respect to , set it to [math] and solve for , resulting in a system of linear equations for unknown :
[TABLE]
where , means a vector form of a matrix by stacking its columns, and creates a diagonal matrix with input vector as its diagonal elements. See Appendix A for a detailed derivation.
Since the coefficient matrix is in general symmetric, sparse and positive definite (PD)222See Appendix B for a detailed description of ., (5) can be solved efficiently using a plethora of mature numerical linear algebra methods such as conjugate gradient (CG) [45]. This is one notable appeal of formulating the MC problem using the dual graph signal smoothness prior in (4), where computing its solution requires only solving a system of linear equations.
III-B Graph Sampling for Matrix Completion based on DGLR Formulation
The stability of the linear system in (5) is determined by the condition number of coefficient matrix , which is the ratio of the largest eigenvalue of to its smallest eigenvalue . Given that is upper-bounded for a degree-constrained graph (see Appendix C for a proof), to maximize stability, we seek to maximize through sampling, i.e.,
[TABLE]
Maximizing of a coefficient matrix is also known as the E-optimality criterion in optimal design [18, 19, 20], and is a common objective for many well-known linear system optimizations, e.g., active learning [46], sensor placement [47] and polynomial regression [48]. In our sampling scenario, we show further that maximizing (6) also means minimizing the MSE upper bound, as stated formally in the lemma below.
Lemma 1**.**
Given dual graph Laplacians and , assuming ground truth signal is corrupted by independent additive noise , MSE of the reconstructed signal with respect to the original signal is upper-bounded by
[TABLE]
where .
Proof.
In vector form, . Thus, the solution to the system of linear equations (5) is
[TABLE]
where .
Thus the squared error of estimator with respect to is
[TABLE]
where .
From inequality (9), we see that sampling set only influences the MSE upper bound by manipulating . Moreover, for symmetric and positive definite matrix , we know
[TABLE]
We complete this proof by substituting (10) into equation (9). ∎
In the next section, we will present a fast graph sampling strategy to solve optimization problem (6).
IV Fast Sampling on Product Graph via Gershgorin Circle Theorem
For brevity, we interpret as the Laplacian of a scaled product graph333Cartesian product between two matrices and is defined by: .. Thus optimization (6) becomes the maximization of for matrix . By definition, is a diagonal matrix:
[TABLE]
where .
Thus, coefficient matrix can be rewritten as:
[TABLE]
where , and is an indicator vector with and for .
Finding an optimal (or ) to maximize is combinatorial in nature. Towards a low-complexity sampling strategy, we take a greedy approach, where we iteratively add a locally optimal sample to a selected sample set until the sample budget is exhausted. Hence, assuming we have collected samples in , at the -th iteration, we solve the following local optimization problem:
[TABLE]
where }, with , and with .
To find an optimal solution in (13) for each new sample, one can compute of the incremented Laplacian444We use “increment” here to mean increasing one diagonal element of a matrix by 1 (equivalently shifting the center of one Gershogrin disc right by 1), while other matrix entries remain unchanged. corresponding to all candidate nodes and identify the largest one, which is computation-intensive. Instead, we circumvent multiple computations of the smallest eigenvalue for candidates using a strategy based on the Gershgorin circle theorem (GCT).
IV-A Gershgorin Disc Shift based Graph Sampling
We first review GCT and its corollary [21], which will lead to a lightweight sampling method later.
Theorem 1**.**
Given an matrix with entries , define the -th Gershgorin disc , corresponding to the -th row of , with center and radius . Each eigenvalue of lies within at least one Gershgorin disc, i.e.,
[TABLE]
Corollary 1**.**
If the largest magnitude component of an eigenvector is at index , then its corresponding eigenvalue must be within the -th Gershgorin disc .
This corollary implies that of matrix must reside in the -th Gershgorin disc, where and is the first eigenvector of corresponding to 555In this paper, eigenvectors are all normalized, i.e., .. By (13), shifts the center of the -th Gershgorin disc of to the right by 1. * Our strategy is then to right-shift the Gershgorin disc corresponding to the largest magnitude entry in which contains , thus promoting a larger in ; i.e., select sample where *
[TABLE]
Remark: To choose one sample, our strategy requires computation of only the first eigenvector of a sparse matrix once, without multiple evaluations for all candidates.
Note that in (15) we select the index with the largest magnitude only among entries in the unsampled set instead of the entire vector, as specified in Corollary 1. However, one can guarantee that the largest magnitude index in , in fact, only resides in , and thus (15) is consistent with Corollary 1. We state this formally in the following Proposition:
Proposition 1**.**
* computed from (15) is also the index with the largest magnitude in , i.e., .*
Proof.
Based on the definition below equation (13), we can deduce that
[TABLE]
Hence, in matrix , left-ends of Gershgorin discs corresponding to indices are at 1, and the other discs’ left-ends are at 0. Suppose now that and . According to Corollary 1, the smallest eigenvalue must be within -th Gershgorin disc, i.e., .
We also know the first eigenvector of matrix is a constant vector with eigenvaue 0. This yields:
[TABLE]
where the last inequality holds since .
This is contradictory to previous result that . Thus and . ∎
Thus, we conclude that entries within set cannot have the largest energy in first eigenvector of .
We can alternatively justify our strategy by showing that the index chosen by our strategy optimizes a related objective to (13). We state this formally in the following lemma.
Lemma 2**.**
An optimal solution to the problem
[TABLE]
is , where is the first eigenvector of corresponding to smallest eigenvalue .
Proof.
Since matrix is augmented by a small matrix , the resulting matrix is , where . Using the Rayleigh quotient theorem [49], we can write of matrix as
[TABLE]
where the minimizer is the first eigenvector of when , i.e., . Therefore,
[TABLE]
where is omitted since .
Given collected , does not depend on . Hence,
[TABLE]
∎
Thus, by computing using (15), we are optimally solving problem (18), which is a proxy approximating original (13).
IV-B Fast Repeated Eigenvector Computation with Warm Start
Our proposed formulation (15) requires computing the first eigenvector of in each greedy step. In this paper, we will adopt the state-of-the-art LOBPCG method [22] to compute the first eigenvector, which has been proved very efficient for large sparse matrices [14]. With an initial input , in each iteration, LOBPCG works as follows:
Multiply with (guess of the first eigenvector in -th iteration) with complexity , where is the number of non-zero entries in ; 2. 2.
Perform a Rayleigh-Ritz step [22] to compute the combination coefficients, solving an eigenvalue problem with complexity . is the number of computed eigenvectors and in our problem, ; 3. 3.
Update based on Rayleigh-Ritz coefficients, go to step (1) until convergence.
Our proposed algorithm can benefit computationally from warm start when deploying LOBPCG: we use the estimated first eigenvector of in the last iteration as the initial guess for . The small change between and (the Forbenius norm difference is only 1) ensures a good initial guess, reducing the number of iterations for LOBPCG to converge. Simulation results in Section VII show that warm start does reduce sampling time noticeably.
We write the pseudo code of our sampling strategy in Algorithm 1, called Gershgorin circle shift (GCS)-based sampling.
IV-C Complexity Analysis
The complexity of our proposed GCS method is dominated by three components: i) times greedy search, ii) first eigenvector computation, and iii) finding the largest element’s location in each greedy search.
Identifying the largest energy index in a vector with length has complexity . The complexity of using LOBPCG to compute the first eigenvector of is , where is the number of iterations till convergence in LOBPCG. Note that since , and the diagonal terms of are all non-zero for a connected graph. Because , i.e., matrix is consisted of matrix and matrix , the number nonzero entries in is at most , where and are the numbers of edges in graph and respectively.
Therefore, denoting by , in each greedy step, the complexity of LOBPCG is ; combined with times greedy search and signal sorting, our GCS method has the complexity . If the row graph and column graph are both sparse such that and , then the complexity of GCS can be abbreviated as . Though the spectral proxy based sampling method in [14] also computes the first eigenvector of a submatrix of via LOBPCG, it does not benefit from warm start, and the complexity of computing will be higher than by at least by a factor .
IV-D Explanation from Graph Spectral Energy Perspective
We now interpret the GCS sampling from an energy spreading perspective. First, we define the absolute value of the first eigenvector of incremented Laplacian as graph spectral energy. Our proposed GCS method is to select node with the largest energy at each greedy step. Once node is sampled, intuitively, the energy of this node and nodes near should decrease such that in the next step, the proposed strategy will not sample those nodes. This is formally stated in the next lemma:
Lemma 3**.**
Denote by , and , . Then
[TABLE]
In our problem, .
Proof.
From the Rayleigh quotient theorem [49],
[TABLE]
and,
[TABLE]
From equation (23), we know , which implies . From equation (24), we can derive that , which is exactly the right part of the lemma. ∎
This Lemma states that sampling node will reduce the spectral energy at node . Moreover, sampling node with the largest actually maximizes the upper-bound of . We know that the value of is penalized from , so selecting the node with largest will also promote a reasonably large , thus provide a large lower-bound of .
Equation (3) tells us that strongly connected nodes would have similar signal, thus the energy of nodes near is also decreased when node is sampled. Therefore, nodes close to will not be sampled by the proposed GCS method with highly probability. This agrees with our intuition: node carries information of its local neighborhood; after sampling it, there is no need to sample connected nodes in its neighborhood.
We conduct toy experiments on a community graph with 100 nodes using the proposed GCS sampling, whose results are shown in Fig.1. As depicted in this experiment, the first four samples lie in four different communities. From the graph energy perspective, sampling one node in one community will decrease the energy of nodes within this community, thus leading to sampling the next node from other communities.
V Iterative Graph Spectral Sampling for Matrix Completion
Though we identify samples using LOBPCG to compute first eigenvectors repeatedly with warm start, sampling on a very large product graph (matrix ) with nodes is still expensive for large real-world MC datasets. We thus propose an efficient block-wise sampling method for MC problem operating on two corresponding row and column graphs, while retaining the same sampling idea in GCS. Towards a simpler presentation, we omit in in the sequel.
Since coefficient matrix is a combination of row and column from and , it does not exhibit any structure that one can exploit for optimization. We thus split into two separate matrices and as follows:
[TABLE]
where is a split parameter.
Since and are both Hermitian, by Weyl’s inequality [49]
[TABLE]
which indicates that each selected sample affects respective ’s of and and the lower bound of .
From a Gershgorin circle perspective, each selected sample shifts one disc in and by and , respectively. Next, we will exploit the block diagonal property of matrices and to develop an efficient sampling framework.
V-A Block Diagonal Structure and Inner Connections
has block-diagonal structure, i.e.,
[TABLE]
where is the -th diagonal block of .
When entry of the target matrix is sampled, since and . Equivalently, if , we know that the information from -th row (movie) and -th column (customer) is collected.
In contrast, matrix is:
[TABLE]
It is known that matrix and are permutation similar, i.e., there exists a permutation matrix such that [50]:
[TABLE]
Thus the permuted sampling matrix for is also a block-diagonal matrix.
[TABLE]
where is the -th diagonal block of .
Combined with the block diagonal property of , we will write the permuted form of as follows:
[TABLE]
where for brevity.
From the property of similarity transform, we know that since for any permutation matrices. From (45), we see that when , the -th disc in -th block in matrix is shifted.
We illustrate the relationship between matrix and via a simple example. Assuming , we have the following two matrices:
[TABLE]
[TABLE]
where are the elements in matrix .
We first assign the Gershgorin discs in matrices and with indices, as illustrated in Fig. 2. Assuming , the disc in matrix is shifted by . After permutation, this means that the disc in matrix is shifted, i.e., . Thus, is equivalent to .
Therefore, sampling data at will promote the smallest eigenvalue of -th (-th) diagonal block of () by making (). Given this connection, we next propose an iterative sampling strategy. Block matrices in and will be called ‘clusters’ and ‘groups’ respectively.
V-B Iterative Sampling between Clusters and Groups
We here propose to alternately collect samples based on one cluster in only or one group in only. Specifically, we start sampling the matrix signal from the first column () based on the Laplacian matrix of the first cluster in . If its first eigenvetor has the largest energy at the -th index, we will sample data at and then proceed sampling based on corresponding incremented Laplacian matrix of the -th group in . We continue to choose samples alternating between clusters and groups until the sampling budget is exhausted.
Since our GCS sampling strategy using LOBPCG benefits from warm start, the computation complexity of this iterative scheme can be further reduced if we choose more than one sample from the same cluster (or group). We thus introduce a warm start parameter to trade off sampling performance and computation complexity; its sensitivity will be examined in Section VII. Detailed iterative sampling pseudo-code is shown in Algorithm 2, called iterative Gershgorin circle shift (IGCS)-based sampling. As we analyzed in Section IV, our proposed GCS has complexity , while the complexity of IGCS is just . is the convergence iteration number of LOBPCG in IGCS, and . Therefore, the complexity is reduced by at least a factor using this iterative sampling framework, i.e., the complexity is roughly linear to the size of factor graph.
VI Graph Construction
Before using graph sampling for MC via the proposed IGCS, one has to first acquire the row graph and the column graph . There exist many methods to construct finite row / column graphs from data, so that the observed signal(s) are smooth (low-pass) with respect to the constructed graphs [3, 11]. For completeness, we overview methods we chose to construct row and column graphs using which we select samples. We stress that our work focuses on sampling; the discussion here merely demonstrates that our graph sampling schemes can be practically realized in combination with existing graph learning methods.
G1: Feature-based graph
As done in graph-based MC methods [16, 26], when the user / item profiles (e.g., age, gender and occupation of users and genre of the items) are available, we construct a weighted 10-nearest neighbor graph using GSPBox [51] based on feature vectors of each node.
G2: Content-based graph from observed information
When features of data points are not available, we construct row and column graphs only from partial matrix entries, extending method used in [3]. Specifically, the observed matrix is for a given random initial set . Then, for each pair of users , their partial ratings are in the - and -th rows of matrix , denoted by and . We then compute the inter-node distance as
[TABLE]
where , and is the set of items rated by user .
If , we set . We then compute the edge weight between users and as
[TABLE]
where and is the threshold of user for sparsifying ; is a factor to control function shape for weight computation.
Likewise, the item graph is constructed similarly using column ratings in observed matrix . In our experiments, for real-world datasets, we first assume partial ground-truth data is known, and then use them to construct the factor graphs based on the above method. With the constructed graphs, we proceed the following sampling based on different schemes and then compute the completion error on the unobserved entries.
VII Experimentation
In this section, we present experimental results of our proposed sampling methods and other competing schemes, combined with several state-of-the-art MC strategies. We list profiles of the simulated datasets in Table. I.
VII-A Experimental Setup
In all experiments, we set for GCS and IGCS, and set for IGCS. We implement five sampling methods for comparison, whose specific settings are as follows:
- •
Graph weight coherence (GWC-random) [24]: the ‘estimated’ setting was used for computing the probability for random sampling without replacement. The bandwidth information was set to be 1000.
- •
Localized operator coverage (LOC) [23]: the bandwidth prior was set to be 1000.
- •
Product Graph-based sampling (PG) [12]: the dual graph bandwidth was set to be or for conducting structured sampling.
- •
LSS [9]: we set rank to be 5 for implementing this method.
The last competing method is uniform random sampling. In the following, we list the details for all simulated MC methods:
- •
IMC [56] and SVT [57] were simulated with the same settings as reproducible codes.
- •
GMC [3]: we set , and . For completion method in equation (4), we set and keep other parameters the same.
- •
GRALS [25]: we set the rank to be 5 for GRALS, and its Laplacian matrix input were computed from and , as used in [12].
- •
GCMC [26]: the training epochs were set to be 1000.
- •
NMC [27]: the number of training epochs was 10000.
VII-B Performance on Small Datasets
We first conduct experiments on small-size Synthetic Netflix datasets for performance comparison, where the matrix is completed by dual smoothness based method (4). For our proposed GCS, LOBPCG is employed with warm start to compute the first eigenvector of matrix , while the IGCS uses the MATLAB’s inbuilt function (Krylov-Schur method) for eigen-decomposition since the factor graphs are small. Two classical graph sampling methods GWC-random [24] and LOC [23] are implemented on the product graph directly. For structured method PG, with bandwidth input or , we artificially increase the parameter to get rectangular output, where and are its selected row and column indices, respectively. After sampling, we record its sample size and corresponding reconstruction error. The root mean square error (RMSE) is computed on unobserved entries in terms of ground-truth value for evaluation, as done in [12, 29, 13].
Specific experimental results on noiseless and noisy synthetic Netflix dataset (noisy one is depicted in Fig. 3 (a)) in terms of sample size are shown in Fig. 3 (b) and (c). We observe that our proposed GCS outperforms all competitors especially when the sampling budget is small and the matrix signal is noisy. Though with bandwidth , PG has comparable RMSE value, its performance deteriorates drastically by just changing the bandwidth to . This means that PG is very sensitive to bandwidth settings. Further, PG cannot achieve arbitrary sample size due to its rigid sampling structure.
Our proposed iterative sampling method IGCS also achieves good performance when sample size is relatively large with complexity . Recall that the complexity of GCS is and LOC is , where is the number of non-zero entries in matrix . We know that , where is the largest degree in product graph . Hence IGCS has much lower complexity. Fig. 3 (d) further illustrates GCS’s superiority for different noise levels, where we remove some inferior competitors for better visualization.
We also test sampling methods on the dense submatrix () from two real-world datasets ML100K and ML10M. As done in [3], by assuming that the information outside this submatrix is given as prior, we construct content-based graph G2 via strategy described in Section VI. Since the ground truth submatrix is sparse, the sampling method must be constrained to sample on the sparse entries. PG, being a structured sampling strategy, cannot satisfy this requirement. Further, LOC requires computation of a Chebyshev polynomial graph filter before sampling, which always results in out-of-memory error using our constructed graph. Thus, we show only executable sampling schemes for comparison. The resulting RMSE is shown in Fig. 3 (e) and (f), which illustrate IGCS’s superiority over GWC, GCS and random sampling for those constructed small real-world datasets.
VII-C Performance on Real-world Large Datasets
To actively sample entries on large real-world datasets, both GCS and GWC are not applicable, since the size of the product graph can contain millions of nodes. For the following real-world large datasets, we only test uniform random sampling and LSS [9] for performance comparison to our proposed IGCS.
Simulations on Movielens 100K
We first deploy our proposed IGCS on a large real-world dataset ML100K [52] to collect samples. Since the typically used ratio between training and testing in ML100K is 80% to 20%, in our experiments, we first randomly select 60K samples from 100K datasets as the initial available data, and then proceed to sample 20K from the rest 40K data pool based on our proposed IGCS or random sampling. The final un-selected 20K samples are used for computing RMSE. For this experiment, we use MATLAB’s inbuilt function for eigen-decomposition in IGCS.
We create the feature-based graph G1 from ML100k’s features and content-based graph G2 from 60K initial samples using the second method in Section VI. Average RMSE on different MC methods and graphs are listed in Table. II, where the best performance number for each MC method is marked in boldface. In the widely used feature-based graph (G1), our proposed IGCS (right side) achieves better performance than random sampling (left side) for almost all popular MC methods. Further, when we select entries on the content-based graph (G2), IGCS substantially outperforms random sampling. Note that when G2 is used instead of G1, RMSE for random sampling is almost the same for every graph-based MC method. Hence, we can conclude that the performance improvement using G2 is due to the more informative samples chosen using our proposed IGCS.
Warm start parameter’s effect on sampling time
In this experiment, we deploy IGCS on ML100K using different graphs and in combination with different MC methods. The resulting RMSE values and sampling times are shown in Table. III, along with LSS for comparison. “eigs” means the eigen-decompostion in IGCS is computed using the Krylov-Schur method, while LOBPCG is used for different . Note that LSS is essentially a random sampling with specified selection probability for each entry. Thus we didn’t record its running time. All experiments are performed on a laptop with Intel Core i7-8750H and 16GB of RAM on Windows 10 for counting time. In Table. III, the best performance numbers for each method are marked in boldface. Table. III shows that IGCS is always superior to LSS using different state-of-the-art MC methods under different graphs. It also shows that with increasing , execution time of IGCS with LOBPCG decreases substantially, while the performance become slightly worse. Note that when , there is no warm start in LOBPCG. Simulation results show that LOBPCG is more efficient than Krylov-Schur method for computing the first eigenvector and achieves better performance for MC.
Simulations on other popular real-world datasets
We next evaluate IGCS on various well-known real-world datasets, combined with GRALS MC method. Random sampling and LSS are simulated for comparison. Since features for constructing G1 are not available for most datasets, we use G2 as the underlying graph for sampling. For datasets Flixter, YahooMusic, Douban, random 90/10 training/test splits are used for simulations. Specifically, we first choose 80% entries in the given 90% training set as the initial samples to construct G2 and then use our IGCS method (or competing schemes) to sample entries to form a new 90% training set. RMSE is computed on final un-selected 10% entries. For other datasets, we first randomly generate 90/10 training/test split and then use the above-mentioned procedure to collect samples and compute RMSE 666 For datasets ML1M and ML10M, the percentage of initial samples for constructing G2 is changed from 80% into 90%. .
Experimental RMSEs of different sampling methods are shown in Table. IV. Table. IV shows that IGCS outperforms random sampling and LSS in all datasets, which have various data size, density and rating level. Moreover, when the warm start parameter in IGCS becomes larger, RMSE of IGCS deteriorates only slightly, but still significantly outperforms the competitors for almost all datasets.
VIII Conclusion
Pre-selection of entries for matrix completion is an important but under-addressed problem. In this paper, we propose a graph sampling strategy for matrix completion based on recurrent Gershgorin disc shift. Specifically, assuming that the target matrix signal is smooth with respect to dual graphs, we can complete the matrix via partial observations by solving a system of linear equations. To maximize the stability of the linear system, we select samples to maximize the smallest eigenvalue of the coefficient matrix, which is equivalent to minimize the upper-bound of the reconstructed error. We tackle the formulated sampling objective with a greedy scheme to select one sample at a time (equivalent to shifting one Gershorin disc). To achieve fast sampling, inspired by one corollary of the Gershgorin circle theorem, we select the node corresponding to the largest energy in the first eigenvector of the incremented Laplacian matrix. We employ LOBPCG to compute the first eigenvector of an incremented Laplacian matrix, which benefits from warm start as the first eigenvectors are computed repeatedly. To efficently sample large real-world datasets, we further devise a block-wise graph sampling scheme, where the samples are collected alternately between blocks in two separate block-diagonal matrices. Extensive experiments have validated the superiority of our proposed graph sampling method for matrix completion, compared with other graph sampling and active matrix completion methods, in different datasets, under different graphs, and in combination with different popular matrix completion methods.
Acknowledgement
The authors would like to thank the great help from Wes Eardley for investigating available datasets for simulations.
Appendix A Derivations of linear equation
We first compute the derivative of with respect to the optimization variable :
[TABLE]
whose vector form by using the property is:
[TABLE]
Note that and , so . To obtain an optimal solution, we set , which in vector form leads to
[TABLE]
Appendix B The property of matrix
Note that since and , so of is at least 0 based on Weyl’s inequality on eigenvalues that [49]. Moreover, the vectorized sampling operator is positive semi-definite (PSD). If matrix is invertible with enough samples in matrix , we know that is sparese, symmetric and positive definite (PD), and the optimal solution to problem (73) in closed form is :
[TABLE]
Appendix C The upper-bound of
Reusing the notations in Section IV, let , and . Specifically,
[TABLE]
For with and , the -th row of matrix (denoted by ) is a combination of the -th row of (denoted by ) and the -th row of (denoted by ). It is easy to see that and . Since and by the definitions of combinatorial Laplacian matrix and , we know that . Based on the Gershgorin circle theorem presented in Section IV-A, we know that the eigenvalues of matrix are all bounded in [0,], where 0 is the lower bound of all left ends of Gershgorin discs, and is the upper bound of all right ends of those discs. Note that and for connected graph without self-loop. Assuming and (degree constrained graphs), we will have
[TABLE]
Therefore, will be upper-bounded by . Because is a diagonal matrix with diagonal entries 0 or 1, will be upper-bounded by if the two factor graphs are degree-bounded by and respectively.
The reference list from the paper itself. Each links out to its DOI / PubMed record.
- 1[1] G. Liu, Q. Liu, and X. Yuan, “A new theory for matrix completion,” in Advances in Neural Information Processing Systems , 2017, pp. 785–794.
- 2[2] E. J. Candès and B. Recht, “Exact matrix completion via convex optimization,” Foundations of Computational mathematics , vol. 9, no. 6, p. 717, 2009.
- 3[3] V. Kalofolias, X. Bresson, M. Bronstein, and P. Vandergheynst, “Matrix completion on graphs,” ar Xiv preprint ar Xiv:1408.1717 , 2014.
- 4[4] E. Candes and Y. Plan, “Matrix completion with noise,” in Proceedings of the IEEE , vol. 98, no.6, April 2010, pp. 925–936.
- 5[5] N. Rubens, M. Elahi, M. Sugiyama, and D. Kaplan, “Active learning in recommender systems,” in Recommender systems handbook . Springer, 2015, pp. 809–846.
- 6[6] S. Chakraborty, J. Zhou, V. Balasubramanian, S. Panchanathan, I. Davidson, and J. Ye, “Active matrix completion,” in 2013 IEEE 13th International Conference on Data Mining . IEEE, 2013, pp. 81–90.
- 7[7] A. Krishnamurthy and A. Singh, “Low-rank matrix and tensor completion via adaptive sampling,” in Advances in Neural Information Processing Systems , 2013, pp. 836–844.
- 8[8] S.-J. Huang, M. Xu, M.-K. Xie, M. Sugiyama, G. Niu, and S. Chen, “Active feature acquisition with supervised matrix completion,” in Proceedings of the 24th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining . ACM, 2018, pp. 1571–1579.
