Query-Efficient Correlation Clustering

2020·Arxiv

ABSTRACT

ABSTRACT

Correlation clustering is arguably the most natural formulation of clustering. Given n objects and a pairwise similarity measure, the goal is to cluster the objects so that, to the best possible extent, similar objects are put in the same cluster and dissimilar objects are put in different clusters.

A main drawback of correlation clustering is that it requires as input the pairwise similarities. This is often infeasible to compute or even just to store. In this paper we study query-efficient algorithms for correlation clustering. Specifically, we devise a correlation clustering algorithm that, given a budget of Q queries, attains a solution whose expected number of disagreements is at most 3is the optimal cost for the instance. Its running time is O(Q), and can be easily made non-adaptive (meaning it can specify all its queries at the outset and make them in parallel) with the same guarantees. Up to constant factors, our algorithm yields a provably optimal trade-off between the number of queries Q and the worst-case error attained, even for adaptive algorithms.

Finally, we perform an experimental study of our proposed method on both synthetic and real data, showing the scalability and the accuracy of our algorithm.

CCS CONCEPTS

Theory of computation → Graph algorithms analysis; Facility location and clustering; Active learning;

KEYWORDS

correlation clustering, active learning, query complexity, algorithm design

ACM Reference Format:

David García–Soriano, Konstantin Kutzkov, Francesco Bonchi, and Charalampos Tsourakakis. 2020. Query-Efficient Correlation Clustering. In Pro- Taiwan. ACM, New York, NY, USA, 11 pages. https://doi.org/10.1145/3366423. 3380220

1 INTRODUCTION

Correlation clustering [3] (or cluster editing) is a prominent clustering framework where we are given a set V = [n] and a symmetric pairwise similarity function sim : , where is the set of unordered pairs of elements of V . The goal is to cluster the items in such a way that, to the best possible extent, similar objects are put in the same cluster and dissimilar objects are put in different clusters. Assuming that cluster identifiers are represented by natural numbers, a clustering is a function , and each cluster is a maximal set of vertices sharing the same label. Correlation clustering aims at minimizing the following cost:

The intuition underlying the above problem definition is that if two objects x and y are dissimilar and are assigned to the same cluster we should pay a cost of 1, i.e., the amount of their dissimilarity. Similarly, if x,y are similar and they are assigned to different clusters we should pay also cost 1, i.e., the amount of their similarity sim(x,y). The correlation clustering framework naturally extends to non-binary, symmetric function, i.e., sim : . In this paper we focus on the binary case; the general non-binary case can be efficiently reduced to this case at a loss of only a constant factor in the approximation [3, Thm. 23]. The binary setting can be viewed very conveniently through graph-theoretic lenses: the n items correspond to the vertices of a similarity graph G, which is a complete undirected graph with edges labeled “+” or “-”. An edge e causes a disagreement (of cost 1) between the similarity graph and a clustering when it is a “+” edge connecting vertices in different clusters, or a “–” edge connecting vertices within the same cluster. If we were given a cluster graph [22], i.e., a graph whose set of positive edges is the union of vertex-disjoint cliques, we would be able to produce a perfect (i.e., cost 0) clustering simply by computing the connected components of the positive graph. However, similarities will generally be inconsistent with one another, so incurring a certain cost is unavoidable. Correlation clustering aims at minimizing such cost. The problem may be viewed as the task of finding the equivalence relation that most closely resembles a given symmetric relation. The correlation clustering problem is NP-hard [3, 22].

Correlation clustering is particularly appealing for the task of clustering structured objects, where the similarity function is domainspecific. A typical application is clustering web-pages based on similarity scores: for each pair of pages we have a score between 0 and 1, and we would like to cluster the pages so that pages with a high similarity score are in the same cluster, and pages with a low similarity score are in different clusters. The technique is applicable to a multitude of problems in different domains, including duplicate detection and similarity joins [13, 17], spam detection [6, 20], coreference resolution [19], biology [4, 8], image segmentation [18], social networks [7], and clustering aggregation [16]. A key feature of correlation clustering is that it does not require the number of clusters as part of the input; instead it automatically finds the optimal number, performing model selection.

Despite its appeal, the main practical drawback of correlation clustering is the fact that, given n items to be clustered, larity computations are needed to prepare the similarity graph that serves as input for the algorithm. In addition to the obvious algorithmic cost involved with queries, in certain applications there is an additional type of cost that may render correlation clustering algorithms impractical. Consider the following motivating real-world scenarios. In biological sciences, in order to produce a network of interactions between a set of biological entities (e.g., proteins), a highly trained professional has to devote time and costly resources (e.g., equipment) to perform tests between all pairs of entities. In entity resolution, a task central to data integration and data cleaning [23], a crowdsourcing-based approach performs queries to workers of the form “does the record x represent the same entity as the record y?”. Such queries to workers involve a monetary cost, so it is desirable to reduce their number. In both scenarios developing clustering tools that use fewer than queries is of major interest. This is the main motivation behind our work. At a high level we answer the following question:

Contributions. The main contributions of this work are summarized as follows:

• We design a computationally efficient randomized algorithm QECC that, given a budget of Q queries, attains a solution whose expected number of disagreements is at most 3 , where OPT is the optimal cost of the cor- relation clustering instance (Theorem 3.1). We can achieve this via a non-adaptive algorithm (Theorem 3.4).

• We show (Theorem 4.1) that up to constant factors, our algorithm is optimal, even for adaptive algorithms: any algorithm making Q queries must make at least

• We give a simple, intuitive heuristic modification of our algorithm QECC-heur which helps reduce the error of the algorithm in practice (specifically the recall of positive edges), thus partially bridging the constant-factor gap between our lower and upper bounds.

• We present an experimental study of our two algorithms, compare their performance with a baseline based on affinity propagation, and study their sensitivity to parameters such as graph size, number of clusters, imbalance, and noise.

2 RELATED WORK

We review briefly the related work that lies closest to our paper.

Correlation clustering. The correlation clustering problem is NP-hard [3, 22] and, in its minimizing disagreements formulation used above, is also APX-hard [11], so we do not expect a polynomial-time approximation scheme. Nevertheless, there are constant-factor approximation algorithms [1, 3, 11]. Ailon et al. [1] present QwickCluster, a simple, elegant 3-approximation algorithm. They improve the approximation ratio to 2.5 by utilizing an LP relaxation of the problem; the best approximation factor known to date is 2.06, due to Chawla et al. [12]. The interested reader may refer to the extensive survey due to Bonchi, García-Soriano and Liberty [6].

Query-efficient algorithms for correlation clustering. Queryefficient correlation clustering has received less attention. There exist two categories of algorithms, non-adaptive and adaptive. The former choose their queries before-hand, while the latter can select the next query based on the response to previous queries.

In an earlier preprint [5] we initiated the study of query-efficient correlation clustering. Our work there focused on a stronger local model which requires answering cluster-id queries quickly, i.e., outputting a cluster label for each given vertex by querying at most q edges per vertex. Such a q-query local algorithm allows a global clustering of the graph with qn queries; hence upper bounds for local clustering imply upper bounds for global clustering, which is the model we consider in this paper. The algorithm from [5] is non-adaptive, and the upper bounds we present here (Theorems 3.1 and 3.4) may be recovered by setting in [5, Thm. 3.3]. A matching lower bound was also proved in [5, Thm. 6.1], but the proof therein applied only to non-adaptive algorithms. In this paper we present a self-contained analysis of the algorithm from [5] in the global setting (Theorems 3.1 and 3.4) and strengthen the lower bound so that it applies also to adaptive algorithms (Theorem 4.1). Additionally, we perform an experimental study of the algorithm.

Some of the results from [5] have been rediscovered several years later (in a weaker form) by Bressan, Cesa-Bianchi, Paudice, and Vitale [9]. They study the problem of query-efficient correlation clustering (Problem 1) in the adaptive setting, and provide a query-efficient algorithm, named ACC. The performance guarantee they obtain in [9, Thm. 1] is asymptotically the same that had already been proven in [5], but it has worse constant factors and is attained via an adaptive algorithm. They also modify the lower bound proof from [5] to make it adaptive [9, Thm. 9], and present some new results concerning the cluster-recovery properties of the algorithm.

In terms of techniques, the only difference between our algorithm QECC and the ACC algorithm from [9] is that the latter adds a check that discards pivots when no neighbor is found after inspecting a random sample of size . This additional check is unnecessary from a theoretical viewpoint (see Theorem 3.1) and it has the disadvantage that it necessarily results in an adaptive algorithm. Moreover, the analysis of [9] is significantly more complex than ours, because they need to adapt the proof of the approximation guarantees of the QwickCluster algorithm from [1] to take into account the additional check, whereas we simply take the approximation guarantee as given and argue that stopping QwickCluster after k pivots have been selected only incurs an expected additional cost of (Lemma 3.3).

3 ALGORITHM AND ANALYSIS

Before presenting our algorithm, we describe in greater detail the elegant algorithm due to Ailon et al. [1] for correlation clustering, as it lies close to our proposed method.

QwickCluster algorithm. The QwickCluster algorithm selects a random pivot v, creates a cluster with v and its positive neighborhood, removes the cluster, and iterates on the induced remaining subgraph. Essentially it finds a maximal independent set in the positive graph in random order. The elements in this set serve as cluster centers (pivots) in the order in which they were found. In the pseudocode below, denotes the set of vertices to which there is a positive edge in G from v.

When the graph is clusterable, QwickCluster makes no mistakes. In [1], the authors show that the expected cost of the clustering found by QwickCluster is at most 3 OPT, where OPT denotes the optimal cost.

QECC. Our algorithm QECC (Query-Efficient Correlation Clustering) runs QwickCluster until the query budget Q is complete, and then outputs singleton clusters for the remaining unclustered vertices. The following subsection is devoted to the proof of our

main result, stated next.

Theorem 3.1. Let G be a graph with n vertices. For any Q > 0, Algorithm QECC finds a clustering of G with expected cost at most 3 making at most Q edge queries. It runs in time O(Q) assuming unit-cost queries.

3.1 Analysis of QECC

For simplicity, in the rest of this section we will identify a complete “+,-” labeled graph G with its graph of queries correspond to querying a pair of vertices for the existence of an edge. The set of (positive) neighbors ofv in a graph G = (V, E) will be denoted ; a similar notation is used for the set of positive neighbors of a set . The cost of the optimum clustering for G is denoted OPT. When is a clustering, costdenotes the cost (number of disagreements) of this clustering, defined by (1) with sim

In order to analyze QECC, we need to understand how early stopping of QwickCluster affects the accuracy of the clustering found. For any non-empty graph denote the subgraph ofG resulting from removing all edges incident to (keeping all vertices). Define a random sequence of graphs by and , where are chosen independently and uniformly at random from that if at step i a vertex is chosen for a second time.

The following lemma is key:

Lemma 3.2. Let have average degree ˜d. When going from , the number of edges decreases in expectation by at least

Proof. Let denote the degree of in . Consider an edge . It is deleted if the chosen pivot is an element of (which contains u and v). Let be the 0-1 random variable associated with this event, which occurs with probability

Let be the number of edges deleted (we as- sume an ordering ofV to avoid double-counting edges). By linearity of expectation,

Now we compute

where in the last line, denotes uniform sampling and we used the Cauchy-Schwarz inequality. Hence

Lemma 3.3. LetG be a graph withn vertices and let be the first r pivots chosen by running QwickCluster on G. Then the

expected number of positive edges of G not incident with an element of is less than

Proof. Recall that at each iteration QwickCluster picks a random pivot from R. This selection is equivalent to picking a random pivot v from the original set of vertices V and discarding it if repeating until some is found, in which case a new pivot is added. Consider the following modification of QwickCluster, denoted SluggishCluster, which picks a pivot v at random from V but always increases the counter r of pivots found, even if (ignoring the cluster creation step if ). We can couple both algorithms into a common probability space where each point contains a sequence of randomly selected vertices and each algorithm picks the next one in sequence. For any , whenever the first r pivots of SluggishCluster are , then the first pivots of QwickCluster are the sequence obtained from S by removing previously appearing elements, where . Hence and . Thus the number of edges not incident with the first r pivots and their neighbors in SluggishCluster stochastically dominates the number of edges not incident with the first r pivots and their neighbors in SluggishCluster, since both numbers are decreasing with r.

Therefore it is enough to prove the claim for SluggishCluster. Let and define by We claim that for all 1 the following inequalities hold:

Indeed, is a random function of only, and the average degree of so, by Lemma 3.2,

proving (2). Now (3) now follows from Jensen’s inequality: since

and the function is concave in [0, 1], we have

so (4) follows from (3) by induction on i:

We are now ready to prove Theorem 3.1:

Proof of Theorem 3.1. Let OPT denote the cost of the optimal clustering of be a random variable denoting the clustering obtained by stopping QwickCluster after r pivots are found (or running it to completion if it finds r pivots or less), and putting all unclustered vertices into singleton clusters. Note that whenever makes a mistake on a negative edge, so does; on the other hand, every mistake on a positive edge by is either a mistake by ) or the edge is not incident to any of the vertices clustered in the first i rounds. By Lemma 3.3, there are at most latter in expectation. Hence

Algorithm QECC runs for k rounds, where because each pivot uses 1 queries. Then

On the other hand, we have because of the expected 3-approximation guarantee of QwickCluster from [1]. Thus , proving our approximation guarantee.

Finally, the time spent inside each iteration of the main loop is dominated by the time spent making queries to vertices in R, since this number also bounds the size of the cluster found. Therefore the running time of QECC is

3.2 A non-adaptive algorithm.

Our algorithm QECC is adaptive in the way we have chosen to present it: the queries made when picking a second pivot depend on the result of the queries made for the first pivot. However, this is not necessary: we can instead query for the neighborhood of a random sample S of size . If we use the elements of S to find pivots, the same analysis shows that the output of this variant meets the exact same error bound of 3 OPT . For completeness, we include pseudocode for the adaptive variant of QECC below (Algorithm 3).

In practice the adaptive variant we have presented in Algorithm 2 will run closer to the query budget, choosing more pivots and reducing the error somewhat below the theoretical bound, because it does not “waste” queries between a newly found pivot and the neighbors of previous pivots. Nevertheless, in settings where the similarity computations can be performed in parallel, it may become advantageous to use Non-adaptive QECC. Another benefit of the non-adaptive variant is that it gives a one-pass streaming algorithm for correlation clustering that uses only O(Q) space and processes edges in arbitrary order.

Theorem 3.4. For any Q > 0, Algorithm Non-adaptive QECC finds a clustering of G with expected cost at most 3at most Q non-adaptive edge queries. It runs in time O(Q) assuming unit-cost queries.

Proof. The number of queries it makes is Note that . The proof of the error bound proceeds exactly as in the proof of Theorem 3.1 (because ). The running time of the querying phase of Non-adaptive QECC is O(Q) and, assuming a hash table is used to store query answers, the expected running time of the second phase is bounded by

Another interesting consequence of this result (coupled with our lower bound, Theorem 4.1), is that adaptivity does not help for correlation clustering (beyond possibly a constant factor), in stark contrast to other problems where an exponential separation is known between the query complexity of adaptive and non-adaptive algorithms (e.g., [10, 14]).

4 LOWER BOUND

In this section we show that QECC is essentially optimal: for any given budget of queries, no algorithm (adaptive or not) can find a solution better than that of QECC by more than a constant factor.

Theorem 4.1. For anyalgorithm finding a clustering with expected cost at most must make at least adaptive edge similarity queries.

Note that this also implies that any purely multiplicative approximation guarantee needs queries (e.g. by taking T = 10n).

Proof. Let ; then 1. By Yao’s minimax principle [25], it suffices to produce a distribution G over graphs with the following properties:

• the expected cost of the optimal clustering of is • for any deterministic algorithm making fewer than queries, the expected cost (over G) of the clustering produced exceeds 2

Let . We can assume that integral (here we use the fact that and

Consider the following distribution G of graphs: partition the vertices of A into exactly k equal-sized clusters . The set of positive edges will be the union of the cliques defined by , plus edges joining each vertex to all the elements of for a randomly chosen is chosen independently of

Define the natural clustering of a graph by the classes ). We view N also as a graph formed by a disjoint union of the k cliques determined by . This clustering will have a few disagreements because of the negative edges between different vertices with . For any pair of distinct elements , this happens with probability 1/k. The cost of the optimal clustering of G is bounded by that of the natural clustering N, hence

We have to show that any algorithm making < L/2 queries to graphs drawn from G produces a clustering with expected cost larger than 2. Since all graphs in G induce the same subgraphs on A and B separately, we can assume without loss of generality that the algorithm queries only edges between A and B. Note that the neighborhoods in G of every pair of vertices from the same are the same: and for all , ; moreover, u and v are joined by a positive edge. Therefore, if but the algorithm assigns u and v to different clusters, either moving u to v’s cluster or v to u’s cluster will not decrease the cost. All in all, we can assume that the algorithm outputs k clusters with for all i, plus (possibly) some clusters ) involving only elements of B.

For , let denote the cluster that the algorithm assigns v to. For every , let denote the event that the algorithm queries (u,v) for some and, whenever does not hold, let us add a “fictitious” query to the algorithm between v and some arbitrary element of . This ensures that whenever , the last query of the algorithm verifies its guess and returns 1 if the correct cluster has been found. This adds at most queries in total. Let be the (random) sequence of queries issued by the algorithm and let be the indices of those queries involving a fixed vertex that is independent of the response to all queries not involving v and, conditioned on the result of all queries up to time is uniformly distributed among the set , whose size is upper-bounded by 1. Therefore

which becomes an equality if the algorithm does not query the same cluster twice. It follows by induction that

Let be the event that the algorithm makes more than k/2 queries involving v. The event is equivalent to , i.e., the event , because of our addition of one fictitious query for v. We have

In other words, either the algorithm makes many queries forv, or it hits the correct cluster with few queries. (Without fictitious queries, we would have to add a third term for the probability that the algorithm picks by chance the correct .) We will use the first term Prto control the expected query complexity. The second term, Pr, is bounded by 12 by (5) because 2 whenever

holds. Hence

so

Each vertex with , causes disagreements with all of and , introducing at least 2new disagreements.

If we denote by X the cost of the clustering found and by Z the number of queries made, we have

But then we can lower bound the expected number of queries by

of which at most L/2 are the fictitious queries we added. This completes the proof.

5 A PRACTICAL IMPROVEMENT

As we will see in Section 6, algorithm QECC, while provably optimal up to constant factors, sometimes returns solutions with poor recall of positive edges when the query budget is low. Intuitively, the reason is that, while picking a random pivot works in expectation, sometimes a low-degree pivot is chosen and all are spent querying its neighbors, which may not be worth the effort for a small cluster when the query budget is tight. To entice the algorithm to choose higher-degree vertices (which would also improve the recall), we propose to bias it so that pivots are chosen with probability proportional to their positive degree in the subgraph induced by R. The conclusion of Lemma 3.3 remains unaltered in this case, but whether this change preserves the approximation guarantees from [1] on which we rely is unclear. In practice, this heuristic modification consistently improves the recall on all the tests we performed, as well as the total number of disagreements in most cases.

We cannot afford to compute the degree of each vertex with a small number of queries, but the following scheme is easily seen to choose each vertex with probability degree of u in the subgraph G[R] induced by R, and E > 0 is the total number of edges in G[R]:

(1) Pick random pairs of vertices to query until an edge (2) Select the first endpoint u of this edge as a pivot. When E = 0, this procedure will simply run out of queries to make. Pseudocode for QECC-heur is shown below.

6 EXPERIMENTS

In this section we present the results of our experimental evaluations of QECC and QECC-heur, on both synthetic and real-world graphs. We view an input graph as defining the set of positive edges; missing edges are interpreted as negative edges.

6.1 Experimental setup

Clustering quality measures. We evaluate the clustering S produced by QECC and QECC-heur in terms of total cost (number of disagreements), precision of positive edges (ratio between the number of positive edges between pairs of nodes clustered together in S and the total number of pairs of vertices clustered together in S), and recall of positive edges (ratio between the number of positive edges between pairs of nodes clustered together in S and the total number of positive edges in G). Although our algorithms have been designed to minimize total cost, we deem it important to consider precision and recall values to detect extreme situations in which, for example, a graph is clustered into n singletons clusters which, if the graph is very sparse, may have small cost, but very low recall. All but one of the graphs G we use are accompanied with a ground-truth clustering (by design in the case of synthetic graphs), which we compare against.

Baseline. As QECC is the first query-efficient algorithm for correlation clustering, any baseline must be based on another clustering method. We turn to affinity propagation methods, in which a matrix of similarities (affinities) are given as input, and then messages about the “availability” and “responsibility” of vertices as possible cluster centers are transmitted along the edges of a graph, until a

Table 1: Dataset characteristics: name, type, size and ground truth error measures.

high-quality set of cluster pivots is found; see [15]. We design the following query-efficient procedure as a baseline:

(1) Pick k random vertices without replacement and query their complete neighborhood. Here k is chosen as high as possible within the query budget Q, i.e.,

(2) Set the affinity of any pair of vertices queried to 1 if there exists an edge.

(3) Set all remaining affinities to zero.

(4) Run the affinity propagation algorithm from [15] on the resulting adjacency matrix.

We also compare the quality measures for QECC and QECC-heur for a range of query budgets Q with those from the expected 3-approximation algorithm QwickCluster from [1]. While better approximation factors are possible (2.5 from [1], 2.06 from [12]), these algorithms require writing a linear program with constraints and all pairs of vertices need to be queried. By contrast, QwickCluster typically performs much fewer queries, making it more suitable for comparison. Synthetic graphs. We construct a family of synthetic graphs S = , parameterized by the number of vertices n, number of clusters in the ground truth k, imbalance factor , and noise rate . The ground truthconsists of one clique of size and 1 cliques of size , all disjoint. To construct the input graph , we flip the sign of every edge between same-cluster nodes in with probability , and we flip the sign of every edge between distinct-cluster nodes with probability . (This ensures that the number total number of positive and negative edges flipped is roughly the same.) Real-world graphs. For our experiments with real-world graphs, we choose three with very different characteristics:

• The cora dataset1, where each node is a scientific publication represented by a string determined by its title, authors, venue, and date. Following [21], nodes are joined by a positive edge when the Jaro string similarity between them exceeds or equals 0.5.

• The Citeseer dataset2, a record of publication citations for Alchemy. We put an edge between two publications if one of them cites the other [24].

• The Mushrooms dataset3, including descriptions of mushrooms classified as either edible or poisonous, corresponding to the two ground-truth clusters. Each mushroom is described by a set of features. To construct the graph, we

remove the edible/poisonous feature and place an edge between two mushrooms if they differ on at most half the remaining features. This construction has been inspired by [16], who show that high-quality clusterings can often be obtained by aggregating clusterings based on single features.

Methodology. All the algorithms we test are randomized, hence we run each of them 50 times and compute the empirical average and standard deviations of the total cost, precision and recall values. We compute the average number A of queries made by QwickCluster and then run our algorithm with an allowance of queries ranging from 2n to A at regular intervals.

We use synthetic graphs to study how cost and recall vary in terms of (1) number of nodes n; (2) number of clusters k; (3) imbalance parameter ; (4) noise parameter . For each plot, we fix all remaining parameters and vary one of them.

As the runtime for QECC scales linearly with the number of queries Q, which is an input parameter, we chose not to report detailed runtimes. We note that a simple Python implementation of our methods runs in under two seconds in all cases on an Intel i7 CPU at 3.7Ghz, and runs faster than the affinity propagation baseline we used (as implemented in Scikit-learn).

6.2 Experimental results

Table 1 summarizes the datasets we tested. Figure 1 shows the measured clustering cost against the number of queries Q performed by QECC and QECC-heur in the synthetic graph S(2000, 20, 0.15, 2) and the real-world Cora, Citeseer and Mushrooms datasets.

Comparison with the baseline. It is clearly visible that both QECC-heur and QECC perform noticeably better than the baseline for all query budgets Q. As expected, all accuracy measures are improved with higher query budgets. The number of non-singleton clusters found by QECC-heur and QECC increases with higher values of Q, but decreases when using the affinity-propagation-based baseline. We do not show this value for the baseline on Mushrooms because it is of the order of hundreds; in this case the ground truth number of clusters is just two, and QwickCluster, QECC and QECC-heur need very few queries (compared to n) to find the clusters quickly.

QwickCluster vs QECC and QECC-heur. At the limit, where Q equals the average number A of queries made by QwickCluster, both QECC and QECC-heur perform as well as QwickCluster. In our synthetic dataset, the empirical average cost of QwickCluster is roughly 2.3 times the cost of the ground truth, suggesting that it is nearly a worst-case instance for our algorithm since QwickCluster has an expected 3-approximation guarantee. Remarkably, in the real-world dataset cora, QECC-heur can find a solution just as

Figure 1: Accuracy measures of our two algorithms, the baseline, and ground truth on the datasets of Table 1.

good as the ground truth with just 40,000 queries. Notice that this is half what QwickCluster needs and much smaller than the 1.7 million queries that full-information methods for correlation clustering such as [12] require.

Effect of graph characteristics. As expected, total cost and recall improve with the number of queries on all datasets (Figure 1); precision, however, remains mainly constant throughout a wide range of query budgets. To evaluate the impact of the graph and noise parameters on the performance of our algorithms, we perform additional tests on synthetic datasets where we fix all parameters to those of S(2000, 20, 0.15, 2) except for the one under study. Figure 2 shows the effect of graph size n (1st row), number of clusters k, (2nd row), imbalance parameter (3rd row) and noise parameter (4th row) on total cost and recall, in synthetic datasets. Here we used Q = 15000 for all three query-bounded methods: QECC, QECC-heur and the baseline. Naturally, QwickCluster gives the best results as it has no query limit. All other methods tested follow the same trends, most of which are intuitive:

• Cost increases with n, and recall decreases, indicating that more queries are necessary in larger graphs to achieve the same quality. Precision, however stays constant.

• Cost decreases with k (because the graph has fewer positive edges). Recall stays constant for the ground truth and the unbounded-query method QwickCluster as it is essentially determined by the noise level, but it decreases with k for the query-bounded methods. Again, precision remains constant except for the baseline, where it decreases with k.

• Recall increases with imbalance because the largest cluster, which is the easiest to find, accounts for a larger fraction of the total number of edges m. Precision also increases. On

Figure 2: Effect of graph sizen (1rst row), number of clustersk, (2nd row), imbalance parameter (3rd row) and noise parameter (4rth row) total cost and recall, for a fixed number Q = 15000 of queries, except for QwickCluster.

the other hand, m itself increases with imbalance, possibly explaining the increase in total cost.

• Finally, cost increases linearly with the level of noise recall and precision decrease as grows higher.

Effect of adaptivity. Finally, we compare the adaptive QECC with Non-adaptive QECC as described at the end of Section 3. Figure 3 compares the performance of both on the synthetic dataset and on Cora. While both have the same theoretical guarantees, it can be observed that the non-adaptive variant of QECC comes at moderate increase in cost and, decrease in recall and precision.

Figure 3: Comparison of QECC and its non-adaptive variant.

7 CONCLUSIONS

This paper presents the first query-efficient correlation clustering algorithm with provable guarantees. The trade-off between the running time of our algorithms and the quality of the solution found is nearly optimal. We also presented a more practical algorithm that consistently achieves higher recall values than our theoretical algorithm. Both of our algorithms are amenable to simple implementations.

A natural question for further research would be to obtain query-efficient algorithms based on the better LP-based approximation algorithms [12], improving the constant factors in our guarantee. Another intriguing question is whether one can devise other graphquerying models that allow for improved theoretical results while being reasonable from a practical viewpoint. The reason an additive term is needed in the error bounds is that, when the graph is very sparse, many queries are needed to distinguish it from an empty graph (i.e., finding a positive edge). We note that if we allow neighborhood oracles (i.e., given v, we can obtain a linked list of the positive neighbours of v in time linear in its length), then we can derive a constant-factor approximation algorithm with neighborhood queries, which can be significantly smaller than the number of edges. Indeed, Ailon and Liberty [2] argue that with a neighborhood oracle, QwickCluster runs in time O(n + OPT); if this is . On the other hand, if we can stop the algorithm after rounds, and by Lemma 3.3, we incur an additional cost of only . This shows that more powerful oracles allow for smaller query complexities. Our heuristic QECC-heur also suggests that granting the ability to query a random positive edge may help. These questions are particularly relevant to clustering graphs with many small clusters.

ACKNOWLEDGMENTS

Part of this work was done while CT was visiting ISI Foundation. DGS, FB, and CT acknowledge support from Intesa Sanpaolo Innovation Center. The funders had no role in study design, data collection and analysis, decision to publish, or preparation of the manuscript.

REFERENCES

[1] Nir Ailon, Moses Charikar, and Alantha Newman. 2008. Aggregating inconsistent information: Ranking and clustering. J. ACM 55, 5 (2008), 1–27.

[2] Nir Ailon and Edo Liberty. 2009. Correlation Clustering Revisited: The “True“ Cost of Error Minimization Problems. In Proc. of 36th ICALP. 24–36.

[3] Nikhil Bansal, Avrim Blum, and Shuchi Chawla. 2004. Correlation Clustering. Machine Learning 56, 1-3 (2004), 89–113.

[4] Amir Ben-Dor, Ron Shamir, and Zohar Yakhini. 1999. Clustering Gene Expression Patterns. Journal of Computational Biology 6, 3/4 (1999), 281–297.

[5] Francesco Bonchi, David García-Soriano, and Konstantin Kutzkov. 2013. Local Correlation Clustering. Technical Report. arXiv preprint arXiv:1312.5105.

[6] Francesco Bonchi, David García-Soriano, and Edo Liberty. 2014. Correlation clustering: from theory to practice. In KDD. 1972. http://videolectures.net/ kdd2014_bonchi_garcia_soriano_liberty_clustering/

[7] Francesco Bonchi, Aristides Gionis, Francesco Gullo, Charalampos Tsourakakis, and Antti Ukkonen. 2015. Chromatic correlation clustering. ACM Transactions on Knowledge Discovery from Data (TKDD) 9, 4 (2015), 34.

[8] Francesco Bonchi, Aristides Gionis, and Antti Ukkonen. 2013. Overlapping correlation clustering. Knowl. Inf. Syst. 35, 1 (2013), 1–32.

[9] Marco Bressan, Nicolò Cesa-Bianchi, Andrea Paudice, and Fabio Vitale. 2019. Correlation Clustering with Adaptive Similarity Queries. Technical Report. arXiv preprint arXiv:1905.11902.

[10] Joshua Brody, Kevin Matulef, and Chenggang Wu. 2011. Lower bounds for testing computability by small width OBDDs. In International Conference on Theory and Applications of Models of Computation. Springer, 320–331.

[11] Moses Charikar, Venkatesan Guruswami, and Anthony Wirth. 2005. Clustering with qualitative information. J. Comput. System Sci. 71, 3 (2005), 360–383.

[12] Shuchi Chawla, Konstantin Makarychev, Tselil Schramm, and Grigory Yaroslavtsev. 2015. Near optimal LP rounding algorithm for correlationclustering on complete and complete k-partite graphs. In Proceedings of the forty-seventh annual ACM symposium on Theory of computing. ACM, 219–228.

[13] Erik D. Demaine, Dotan Emanuel, Amos Fiat, and Nicole Immorlica. 2006. Correlation clustering in general weighted graphs. Theoretical Computer Science 361, 2-3 (2006), 172–187.

[14] Eldar Fischer. 2004. On the strength of comparisons in property testing. Information and Computation 189, 1 (2004), 107–116.

[15] Brendan J Frey and Delbert Dueck. 2007. Clustering by passing messages between data points. Science 315, 5814 (2007), 972–976.

[16] Aristides Gionis, Heikki Mannila, and Panayiotis Tsaparas. 2007. Clustering aggregation. ACM Transactions on Knowledge Discovery from Data 1, 1, Article 4 (March 2007).

[17] Oktie Hassanzadeh, Fei Chiang, Renée J. Miller, and Hyun Chul Lee. 2009. Framework for Evaluating Clustering Algorithms in Duplicate Detection. PVLDB 2, 1 (2009), 1282–1293.

[18] Sungwoong Kim, Sebastian Nowozin, Pushmeet Kohli, and Chang Dong Yoo. 2011. Higher-Order Correlation Clustering for Image Segmentation. In NIPS. 1530–1538.

[19] Andrew McCallum and Ben Wellner. 2005. Conditional models of identity uncertainty with application to noun coreference. In Advances in neural information processing systems. 905–912.

[20] Anirudh Ramachandran, Nick Feamster, and Santosh Vempala. 2007. Filtering spam with behavioral blacklisting. In Proceedings of the 14th ACM conference on Computer and communications security. ACM, 342–351.

[21] Barna Saha and Sanjay Subramanian. 2019. Correlation Clustering with SameCluster Queries Bounded by Optimal Cost. Technical Report. arXiv preprint arXiv:1908.04976.

[22] Ron Shamir, Roded Sharan, and Dekel Tsur. 2004. Cluster graph modification problems. Discrete Applied Mathematics 144, 1-2 (2004), 173–182.

[23] Jiannan Wang, Tim Kraska, Michael J Franklin, and Jianhua Feng. 2012. Crowder: Crowdsourcing entity resolution. Proceedings of the VLDB Endowment 5, 11 (2012), 1483–1494.

[24] Zhilin Yang, William W Cohen, and Ruslan Salakhutdinov. 2016. Revisiting semi-supervised learning with graph embeddings. In Proceedings of the 33rd International Conference on International Conference on Machine Learning, Vol. 48. 40–48.

[25] Andrew Chi-Chin Yao. 1977. Probabilistic computations: Toward a unified measure of complexity. In 18th Annual Symposium on Foundations of Computer Science (FOCS 1977). IEEE, 222–227.