Achieving Exact Cluster Recovery Threshold via Semidefinite Programming
Bruce Hajek, Yihong Wu, Jiaming Xu
Introduction
In this paper, we focus on the following particular cases:
Binary symmetric stochastic block model (assuming is even):
where and are fixed constants, and study the problem of exactly recovering the clusters (up to a permutation of cluster indices) from the observation of the graph .
Exact cluster recovery under the binary symmetric stochastic block model is studied in and a sharp recovery threshold is found.
Under the binary symmetric stochastic block model (1), if ,If , is also sufficient for exact recovery as shown by . But for , since , the ER random graph contains isolated vertices with probability bounded away from zero and exact recovery is impossible. the clusters can be exactly recovered up to a permutation of cluster indices with probability converging to one; if , no algorithm can exactly recover the clusters with probability converging to one.
The optimal reconstruction threshold in Theorem 1 is achieved by the maximum likelihood (ML) estimator, which entails finding the minimum bisection of the graph, a problem known to be NP-hard in the worst case [18, Theorem 1.3]. Nevertheless, it has been shown that the optimal recovery threshold can be attained in polynomial time using a two-step procedure : First, apply the partial recovery algorithms in to correctly cluster all but vertices; Second, flip the cluster memberships of those vertices who do not agree with the majority of their neighbors. It remains open to find a simple direct approach to achieve the exact recovery threshold in polynomial time. It was proved in that a semidefinite programming (SDP) relaxation of the ML estimator succeeds if . Backed by compelling simulation results, it was further conjectured in that the SDP relaxation can achieve the optimal recovery threshold. In this paper, we resolve this conjecture in the positive.
In addition, we prove that the SDP relaxation achieves the optimal recovery threshold for the planted dense subgraph model (2) where the cluster size scales linearly in . This conclusion is in sharp contrast to the following computational barrier established in : If grows and decay sublinearly in , attaining the statistical optimal recovery threshold is at least as hard as solving the planted clique problem (See Section 3 for detailed discussions).
Since the initial posting of this paper to arXiv, a number of interesting papers have been posted, some extending or improving our results. Another resolution of the conjecture in was given in independently. A sharp characterization of the threshold for exact recovery for a general class of stochastic block models is derived in , which includes the two cases considered in this paper as special cases. Extensions of this paper appear in , showing SDP provides exact recovery up to the information theoretic threshold for a fixed number of equal sized clusters or two unequal sized clusters. More recently, the preprint shows similar optimality results of SDP for number of equal-sized clusters and establishes optimality results of SDP for a fixed number of clusters with unequal sizes.
Let denote the adjacency matrix of the graph , denote the identity matrix, and denote the all-one matrix. We write if is positive semidefinite and if all the entries of are non-negative. Let denote the set of all symmetric matrices. For , let denote its second smallest eigenvalue. For any matrix , let denote its spectral norm. For any positive integer , let . For any set , let denote its cardinality and denote its complement. We use standard big notations, e.g., for any sequences and , or if there is an absolute constant such that . Let denote the Bernoulli distribution with mean and denote the binomial distribution with trials and success probability . All logarithms are natural and we use the convention .
Stochastic block model
The cluster structure under the binary symmetric stochastic block model can be represented by a vector such that if vertex is in the first cluster and otherwise. Let correspond to the true clusters. Then the ML estimator of for the case can be simply stated as
which maximizes the number of in-cluster edges minus the number of out-cluster edges. This is equivalent to solving the NP-hard minimum graph bisection problem. Instead, let us consider its convex relaxation similar to the SDP relaxation studied in . Let . Then is equivalent to and if and only if . Therefore, (3) can be recast as
Notice that the matrix is a rank-one positive semidefinite matrix. If we relax this condition by dropping the rank-one restriction, we obtain the following convex relaxation of (4), which is a semidefinite program:
We remark that (5) does not rely on any knowledge of the model parameters except that ; for the case , we replace in (5) by . The SDP formulation introduced in is slightly different from ours: it did not impose the constraint and the objective function is the inner product between a weighted adjacency matrix and
Let and . The following result establishes the optimality of the SDP procedure:
It is worthy to note the relationship between our SDP relaxation and other related formulations that appeared previously in the literature. In fact, (5) coincides with the SDP studied in [14, p. 659] for MIN BISECTION, where a sufficient condition is obtained in [14, Lemma 19] that is not optimal. The SDP relaxation considered in [17, Equation (15)] for MAX BISECTION is equivalent to (5) with max replaced by min (used when ) and by . The proof of Theorem 2 shows that this more relaxed version also works. Finally, we note that the SDP used in can be viewed as a penalized version of (5) with added to the objective function.
Theorem 2 naturally extends to the semirandom model considered in , where after a graph is instantiated from the SBM, a monotone adversary can add in-cluster edges and delete cross-cluster edges arbitrarily. Although this process may appear to make the cluster structure more visible, such an adversarial model is known to foil many procedures based on degrees, local search or graph spectrum . By design, the SDP (5) enjoys robustness against such a monotone adversary, which has already been observed in and proved in [9, Lemma 1]. More specifically, let denote the adjacency matrix of the altered graph. Whenever is the unique maximizer of (5), i.e., for any other feasible , then is also the unique maximizer of (5) with replaced by . To see this, note that, by Cauchy-Schwartz inequality, and imply that for all . Consequently, . Then , establishing the unique optimality of .
Planted dense subgraph model
In this section we turn to the planted dense subgraph model in the asymptotic regime (2), where there exists a single cluster of size . To specify the optimal reconstruction threshold, define the following function: For , let
where if and . We show that if , exact recovery is achievable in polynomial-time via SDP with probability tending to one; if , any estimator fails to recover the cluster with probability tending to one regardless of the computational costs. The sharp threshold is plotted in Fig. 1 for various values of .
We first introduce the maximum likelihood estimator and its convex relaxation. For ease of notation, in this section we use a vector , as opposed to used in Section 2 for the SBM, as the indicator function of the cluster, such that if vertex is in the cluster and otherwise. Let be the indicator of the true cluster. Assuming , i.e., the vertices in the cluster are more densely connected, the ML estimation of is simply
which maximizes the number of in-cluster edges. Due to the integrality constraints, it is computationally difficult to solve (11), which prompts us to consider its convex relaxation. Note that (11) can be equivalentlyHere (11) and (12) are equivalent in the following sense: for any feasible for (11), is feasible for (12); for any feasible for (12), either or is feasible for (11). formulated as
where the matrix is positive semidefinite and rank-one. Removing the rank-one restriction leads to the following convex relaxation of (12), which is a semidefinite program.
We note that, apart from the assumption that , the only model parameter needed by the estimator (13) is the cluster size ; for the case , we replace in (13) by .
Let correspond to the true cluster and define {\mathcal{Z}}_{n}=\big{\{}\xi\xi^{\top}:\xi\in\{0,1\}^{n},\xi^{\top}\mathbf{1}=K\big{\}}. The recovery threshold for the SDP (13) is given as follows.
Under the planted dense subgraph model (2), if
Next we prove a converse for Theorem 3 which shows that the recovery threshold achieved by the SDP relaxation is in fact optimal.
Under the planted dense subgraph model, our investigation of the exact cluster recovery problem thus far in this paper has been focused on the regime where the cluster size grows linearly with and , where the statistically optimal threshold can be attained by SDP in polynomial time. However, this need not be the case if grows sublinearly in . In fact, the exact cluster recovery problem has been studied in in the following asymptotic regime:
where and are fixed constants. The statistical and computational complexities of the cluster recovery problem depend crucially on the value of and (see [22, Figure 2] for an illustration):
: the planted cluster can be perfectly recovered in polynomial-time with high probability via the SDP relaxation (13).In fact, an even looser SDP relaxation than (13) has been shown to exactly recover the planted cluster with high probability for . See [10, Theorem 2.3].
: the planted cluster can be detected in linear time with high probability by thresholding the total number of edges, but it is conjectured to be computationally intractable to exactly recover the planted cluster.
: the planted cluster can be exactly recovered with high probability via ML estimation; however, no randomized polynomial-time solver exists conditioned on the planted clique hardness hypothesis.Here the planted clique hardness hypothesis refers to the statement that for any fixed constants and , there exist no randomized polynomial-time tests to distinguish an Erdős-Rényi random graph and a planted clique model which is obtained by adding edges to vertices chosen uniformly from to form a clique. For various hardness results of problems reducible from the planted clique problem, see and the references within.
: regardless of the computational costs, no algorithm can exactly recover the planted cluster with vanishing probability of error.
Consequently, assuming the planted clique hardness hypothesis, in the asymptotic regime of (16) when (and, quite possibly, the entire range ), there exists a significant gap between the information limit (recovery threshold of the optimal procedure) and the computational limit (recovery threshold for polynomial-time algorithms). In contrast, in the asymptotic regime of (2), the computational constraint imposes no penalty on the statistical performance, in that the optimal threshold can be attained by SDP relaxation in view of Theorem 3.
Proofs
In this section, we give the proofs of our main theorems. Our analysis of the SDP relies on two key ingredients: the spectrum of Erdős-Rényi random graphs and the tail bounds for the binomial distributions, which we first present.
Let denote the Erdős-Rényi random graph model with the edge probability for all . Results similar to Theorem 5 have been obtained in for the special case of for some sufficiently large . In fact, Theorem 5 can be proved by strengthening the combinatorial arguments in [15, Section 2.2]. Here we provide an alternative proof using results from random matrices and concentration of measures and a seconder-order stochastic comparison argument from .
where follow from the Jensen’s inequality; follows because has the same distribution as , where denotes the element-wise product; follow from the triangle inequality; follows from the fact that has the same distribution as . Then, we apply the result of Seginer which characterized the expected spectral norm of i.i.d. random matrices within universal constant factors. Let , which are independent . Since is symmetric, [33, Theorem 1.1] and Jensen’s inequality yield
for some universal constant . In view of the following Chernoff bound for the binomial distribution [27, Theorem 4.4]:
for all , setting and applying the union bound, we have
where the last inequality follows from . Assembling (17) – (20), we obtain
2 Tail of the Binomial Distribution
Let be such that and for some and . Then
We use the following non-asymptotic bound on the binomial tail probability [7, Lemma 4.7.2]: For ,
where and is the binary divergence function. Then (23) follows from (24) by noting that .
To prove (22), we use the following bound on binomial coefficients [7, Lemma 4.7.1]:
3 Proofs for the stochastic block model
The following lemma provides a deterministic sufficient condition for the success of SDP (5) in the case .
Then is the unique solution to (5).
where holds because ; holds because by (27). Hence, is an optimal solution. It remains to establish its uniqueness. To this end, suppose is an optimal solution. Then,
where holds because ; holds because and for all . In view of (27), since , with , must be a multiple of . Because for all , . ∎
The theorem is proved first for . Let with
and choose any . It suffices to show that satisfies the conditions in Lemma 3 with probability .
By definition, for all , i.e., . Since , (27) holds, that is, . It remains to verify that and with probability at least which amounts to showing that
Applying the union bound implies that holds with probability at least . It follows from the assumption and (30) that the desired (29) holds, completing the proof in the case .
For simplicity so far we have focused on the case where with being fixed constants. If we are strictly above the recover threshold, namely, , Theorem 2 shows that the probability of error is polynomially small in , which cannot be improved by MLE (though a better exponent is conceivable). Now, if we allow and to vary with , the following result for the optimal estimator (MLE) has been obtained in that gives the second-order refinement of Theorem 1: If , then clusters can be exactly recovered up to a permutation of cluster indices with probability converging to if and only if
for some universal constant , which is stronger than the optimal condition (31). It is unclear whether (32) is necessary, nor do we know if SDP requires to be positive to succeed.
4 Proofs for the planted densest subgraph model
Then is the unique solution to (13).
where follows because , , and ; holds due to ; holds because and . Hence, is an optimal solution. It remains to establish the uniqueness. To this end, suppose is another optimal solution. Then,
where holds because and ; holds because , , and since and for all . Since and with , needs to be a multiple of . Then since . ∎
Define for . Let be given by
It suffices to show that satisfies the conditions in Lemma 4 with probability at least
By definition, we have and for all . Moreover, for all ,
where the last equality holds because if ; for all ,
where the last equality follows from our choice of . Hence, and consequently .
We next show that , with probability at least It follows from Theorem 5 that with probability at least for some positive constant depending only on . Furthermore, let . Then if and otherwise. We divide the analysis into two separate cases. First consider the case , then for all . Since in this case, holds automatically. For any , applying Lemma 2 with yields
By definition, in this case. Applying the union bound implies that with probability at least ,
Since and by the assumption (14), it follows that with probability at least and .
It remains to verify with with probability at least , i.e.,
It follows that for any and ,
where holds because for all and
To lower bound the worst-case probability of error, consider the Bayesian setting where the planted cluster is uniformly chosen among all -subsets of with . If , then the cluster is unidentifiable from the graph.
Next, we prove the theorem first for the case . If , then perfect recovery is possible if and only if the subgraph formed by the vertices in cluster, which is , contains no isolated vertex.To be more precise, if there is an isolated vertex in the cluster , then the likelihood has at least maximizers, which, in turn, implies that the probability of exact recovery for any estimator is at most . This occurs with high probability if .
By symmetry, we can condition on being the first vertices. Let denote the set of first vertices. Then
completing the proof in the case .
completing the proof for the case . ∎
Appendix A The sharpness of the condition for Theorem 5
Consider the case where is the adjacency matrix of . We show that if and , then
where . Using the inequality for with the convention that ,
Since for , it follows that
Plugging into (38), we get that
where the last inequality follows due to since and since , By the independence of , we have
where the last inequality follows in view of (40).