Abstract 1 Introduction 2 Preliminaries 3 The first upper bound 4 General candidate centers 5 Improved upper bound: Removing the 𝐥𝐨𝐠𝐥𝐨𝐠𝒏 term 6 Lower bounds References

Dimension Reduction for Clustering: The Curious Case of Discrete Centers

Shaofeng H.-C. Jiang ORCID Peking University, Beijing, China Robert Krauthgamer ORCID The Harry Weinrebe Professorial Chair of Computer Science, Weizmann Institute of Science, Rehovot, Israel Shay Sapir ORCID Weizmann Institute of Science, Rehovot, Israel Sandeep Silwal ORCID University of Wisconsin-Madison, WI, USA Di Yue ORCID Peking University, Beijing, China
Abstract

The Johnson-Lindenstrauss transform is a fundamental method for dimension reduction in Euclidean spaces, that can map any dataset of n points into dimension O(logn) with low distortion of their distances. This dimension bound is tight in general, but one can bypass it for specific problems. Indeed, tremendous progress has been made for clustering problems, especially in the continuous setting where centers can be picked from the ambient space d. Most notably, for k-median and k-means, the dimension bound was improved to O(logk) [Makarychev, Makarychev and Razenshteyn, STOC 2019].

We explore dimension reduction for clustering in the discrete setting, where centers can only be picked from the dataset, and present two results that are both parameterized by the doubling dimension of the dataset, denoted as ddim. The first result shows that dimension Oε(ddim+logk+loglogn) suffices, and is moreover tight, to guarantee that the cost is preserved within factor 1±ε for every set of centers. Our second result eliminates the loglogn term in the dimension through a relaxation of the guarantee (namely, preserving the cost only for all approximately-optimal sets of centers), which maintains its usefulness for downstream applications.

Overall, we achieve strong dimension reduction in the discrete setting, and find that it differs from the continuous setting not only in the dimension bound, which depends on the doubling dimension, but also in the guarantees beyond preserving the optimal value, such as which clusterings are preserved.

Keywords and phrases:
dimension reduction, clustering, k-median, k-means, doubling dimension
Funding:
Shaofeng H.-C. Jiang: Work partially supported by a national key R&D program of China No. 2021YFA1000900.
Robert Krauthgamer: Work partially supported by the Israel Science Foundation grant #1336/23.
Copyright and License:
[Uncaptioned image] © Shaofeng H.-C. Jiang, Robert Krauthgamer, Shay Sapir, Sandeep Silwal, and Di Yue; licensed under Creative Commons License CC-BY 4.0
2012 ACM Subject Classification:
Theory of computation Random projections and metric embeddings
Related Version:
Full Version: https://arxiv.org/abs/2509.07444 [23]
Acknowledgements:
The authors would like to thank Lingxiao Huang and Yury Makarychev for insightful discussions during the initial stage of this work.
Editor:
Shubhangi Saraf

1 Introduction

Oblivious dimension reduction, in the spirit of the Johnson and Lindenstrauss (JL) Lemma [24], is a fundamental technique for many Euclidean optimization problems over large, high-dimensional datasets. It has a strong guarantee: there is a random linear map π:dt, for a suitable target dimension t=O(ε2logn), such that for every n-point dataset Pd, with high probability, π preserves all pairwise distances in P within factor 1±ε:

x,yP,π(x)π(y)(1±ε)xy,

where throughout is the Euclidean norm. This guarantee is extremely powerful, particularly for algorithms: to solve a Euclidean problem on input P, one can apply the map π, solve the same problem on π(P), which is often more efficient since π(P) lies in low dimension, and “lift” the solution back to the original dimension (as discussed further in Section 1.2).

However, many problems require computational resources that grow exponentially with the dimension (the curse of dimensionality), and hence even dimension t=O(ε2logn) might be too large. Unfortunately, this dimension bound is tight in general, i.e., for preserving all pairwise distances [29], but interestingly one may bypass it for specific optimization problems, by showing that the optimal value/solution is preserved even when the dimension is reduced beyond the JL Lemma, say to dimension t=O(ε2), which is completely independent of n. This raises an important question:

For which problems does dimension o(ε2logn) suffice for oblivious dimension reduction?

Prior work has revealed an affirmative answer for several key problems, as we discuss below. This paper studies this question for fundamental clustering problems, captured by (k,z)-clustering, which includes the famous k-means and k-median problems as its special cases. In (k,z)-clustering, the input is a dataset Pd, and the goal is to find a set of centers C of size |C|k that minimizes

costz(P,C):=pPdistz(p,C), where distz(p,C):=mincCpcz.

We can distinguish two variants, differing in their space of potential centers. In the continuous variant, C is a subset of d (the centers lie in the ambient space), and in the discrete variant, also called sometimes k-medoids, C is a subset of P (or maybe of a larger set given as input). A key feature of the discrete version, is that π:Pπ(P) is invertible, hence each potential center in π(P) corresponds to a unique potential center in P (in contrast, a potential center in the ambient space t has many preimages in d). Thus, in the discrete version, a set of centers computed for the dataset π(P) can be mapped back to the higher dimension and serve as centers for the dataset P. See Section 1.3 for a discussion on practical applications of the discrete variant.

The continuous variant is a success story of the “beyond JL” program. A series of papers [3, 6, 2, 32] has culminated showing that target dimension t=O(ε2logkε), which is independent of n, suffices to preserve all the solutions within factor 1±ε. Curiously, Charikar and Waingarten [4] observed that the discrete variant behaves very differently: certain instances require t=Ω(logn), even for k=1 (when using the standard Gaussian-based map π). Counterintuitively, restricting the centers to be data points makes dimension reduction significantly harder!

To bypass this limitation, we consider the doubling dimension, which was identified in previous work as a natural parameter that is very effective in achieving “beyond JL” bounds [20, 33, 22, 18, 14]. Formally, the doubling dimension of P, denoted ddim(P), is the smallest positive number such that every ball in the finite metric P can be covered by 2ddim(P) balls of half the radius. For several problems, including nearest neighbor [20], facility location [33, 18], and maximum matching [14], target dimension t=O(ε2log1εddim(P)) suffices. Note that restricting the doubling dimension does not immediately imply a better dimension reduction of the JL flavor, as there are datasets Pd with ddim(P)=O(1) where no linear map can approximately preserve all pairwise distances (see e.g., [20, Remark 4.1]).

1.1 Main results

We present the first dimension reduction results for discrete (k,z)-clustering, along with matching lower bounds. Our first result (Theorem 1.1) provides a strong approximation guarantee, but requires a loglogn term in the target dimension, which we show is necessary. Our main result (Theorem 1.2) avoids this loglogn term, through a relaxation of the guarantee that maintains its algorithmic usefulness, e.g., it still implies that the optimal value is preserved up to factor 1±ε.

In all our results, the random linear map π is given by a matrix Gt×d of iid Gaussians N(0,1t), which we refer to as a Gaussian JL map. This is nowadays a standard JL map [19, 13], and our results may extend to other JL maps, similarly to prior work in this context. We denote the optimal value of discrete (k,z)-clustering by

optz(P)=minCP,|C|=kcostz(P,C),

however for sake of exposition, we omit z and focus on z=1 or z=2, which are discrete k-median and k-means. We use the notation O~(f) to hide factors that are logarithmic in f, although below it only hides a log1ε factor.

Theorem 1.1 (Informal version of Theorem 3.1).

For suitable t=O~(ε2(ddim(P)+logk+loglogn)), with probability at least 2/3,

  1. 1.

    opt(G(P))(1+ε)opt(P), and

  2. 2.

    for all CP,|C|k, we have cost(G(P),G(C))(1ε)cost(P,C).

This theorem has immediate algorithmic applications. First, it implies that the optimal value is preserved, i.e., opt(G(P))(1±ε)opt(P). Second, for every CP and β>1, if the set of centers G(C) is a β-approximate solution for the instance G(P), then C is a (1+O(ε))β-approximate solution for the instance P. Therefore, the theorem fit into the general paradigm of using oblivious linear maps – apply the mapping, solve the problem in low dimension, and lift the centers back to the higher dimension.

It is interesting to compare our result with the continuous variant of (k,z)-clustering. On the one hand, to preserve the optimal value in the continuous variant, we know from [32] that target dimension O(ε2logkε) suffices, independently of ddim(P). On the other hand, Theorem 1.1 further provides a “for all centers” guarantee, which is not attainable in the continuous version (by any linear map), by simply considering centers in the kernel of the linear map (see Theorem 6.1). We examine and discuss these guarantees more carefully in Section 1.2.

Matching lower bounds.

The results in Theorem 1.1 are nearly tight for Gaussian JL maps, and likely for all oblivious linear maps. It is known that achieving opt(G(P))(1±ε)opt(P) requires target dimension t=Ω(logk), even for a dataset P of doubling dimension O(1) [33], and another known lower bound is that t=Ω(ddim(P)), even for k=O(1) [4]. It is easy to tighten these bounds with respect to the dependence on ε. We complete the picture, and show in Theorem 6.2 the multiplicative approximation of Theorem 1.1 requires dimension t=Ω(ε2loglogn), even for k=1 and a dataset P of doubling dimension O(1).

To get some intuition about the discrete variant, we briefly recall the hard instance of [4], taking z=1 for simplicity. Consider k=2, and let P be the first n standard basis vectors, thus ddim(P)=logn. The pairwise distances all equal 2, hence opt(P)=2(n2). The standard basis vectors form a well-known hard instance for the JL Lemma, hence, when using target dimension t=o(ε2logn), with high probability, there exists j1[n2] such that Gej1<110ε. Similarly, let j2>n2 be such an index for the last n2 standard basis vectors. Let Gej1,Gej2 be the two centers for G(P), and assign the first n2 basis vectors to Gej2 and the last n2 vectors to Gej1. Now a simple argument using the independence between the two halves shows that opt(G(P))(1ε)2n(1ε/2)opt(P) with probability 2/3.

A relaxed guarantee.

Our main result avoids the loglogn term in Theorem 1.1 by slightly relaxing the guarantee, while keeping it useful for downstream applications.

Theorem 1.2 (Informal version of Theorem 5.1).

For suitable t=O~(ε2(ddim(P)+logk)), with probability at least 2/3,

  1. 1.

    opt(G(P))(1+ε)opt(P), and

  2. 2.

    for all CP,|C|k, we have cost(G(P),G(C))min{(1ε)cost(P,C),100opt(P)}.

This theorem implies that the optimal value is preserved, i.e., opt(G(P))(1±ε)opt(P). Let us further examine which solutions are preserved under this guarantee: For all CP and 1<β<1001+ε, if the set of centers G(C) is a β-approximate solution for the instance G(P), then C is a (1+O(ε))β-approximate solution for the instance P. Recall that for Theorem 1.1, we had a similar claim, but without the restriction β<1001+ε. The constant 100 here is arbitrary, and can be changed to any α>2, at the cost of increasing the target dimension by an additive O(ε2loglogα) term.

1.2 Various notions for preserving solutions

We study several definitions for dimension reduction for k-clustering. All these definitions require (perhaps implicitly) that opt(G(P))(1+ε)opt(P), i.e., that the optimal value has bounded expansion. This direction is often easy because it suffices to analyze one optimal solution for P. In the other direction, one may naively require that opt(G(P))(1ε)opt(P), however this is rather weak, as it does not guarantee that solutions are preserved. Moreover, even requiring that an optimal solution for G(P) is a near-optimal solution for P is quite limited, because a near-optimal solution for G(P), say one found by a (1+ε)-approximation algorithm, may be lifted to a poor solution for P. In fact, such a phenomenon was observed for minimum spanning tree (MST) when using target dimension t=o(logn): an optimal MST of G(P) is a (1+ε)-approximate MST of P, however a (1+ε)-approximate MST of G(P) may have large cost for P [33]. Ideally, we want the cost of every solution to have bounded contraction, as it allows to lift any solution for G(P) to a solution for P, and we thus consider several different notions for the set of solutions, as follows. For simplicity, we present these for z=1 in the discrete setting, but they extend naturally to all z1 and to the continuous setting.

  1. 1.

    Partitions. A solution is a partition 𝒫=(P1,,Pk) of P. Its cost is defined as cost(𝒫):=i=1kmincPipPipc.

  2. 2.

    Centers. A solution is a set of centers C=(c1,,ck)P. Its cost is defined as cost(P,C):=pPdist(p,C).

  3. 3.

    Centers and partitions. A solution is a partition 𝒫=(P1,,Pk) of P and a set of centers C=(c1,,ck)P. Its cost is defined as cost(𝒫,C):=i=1kpPipci.

These definitions are fairly natural, and were used in prior work on dimension reduction, e.g., partition-based solutions were used in [32] for k-means and k-median, and center-based solutions were used in [22] for k-center. It was observed in [4] that not all “for all” guarantees are the same; in particular, “for all centers” and “for all partitions” are incomparable. However,“for all centers and partitions” is clearly stronger than both.

Next, we define contraction for solutions, capturing the two notions in Theorems 1.1 and 1.2. The notion in Theorem 1.1 is simply of multiplicative contraction: A solution S has (1ε)-contraction if cost(G(S))(1ε)cost(S). The notion in Theorem 1.2 is new, at least in the context of dimension reduction, and goes as follows.

Definition 1.3 (Relaxed Contraction).

A solution S has α-relaxed (1ε)-contraction (for α>1, ε>0) if cost(G(S))min{αopt(P),(1ε)cost(S)}.

Using these definitions, we can restate Theorem 1.1 as having (1ε)-contraction for all centers, and restate Theorem 1.2 as achieving 100-relaxed (1ε)-contraction for all centers. In fact, we can strengthen Theorem 1.1 to assert (1ε)-contraction for all centers and partitions.

Theorem 1.4 (Strengthened Theorem 1.1, informal).

For suitable t=O~(ε2(ddim(P)+logk+loglogn)), with probability at least 2/3, for all partitions 𝒫=(P1,,Pk) of P and sets of centers C=(c1,,ck)P,

cost(G(𝒫),G(C))(1ε)cost(𝒫,C).

This strengthening is not attainable for Theorem 1.2, as dimension Ω(ε2loglogn) is needed to get a “for all centers and partitions” guarantee, even for relaxed contraction (see Theorem 6.3). However, we do not know if a “for all partitions” guarantee is possible without the loglogn term. If it is possible, then a curious phenomenon will occur: we get a “for all partitions” and a “for all centers” guarantees, but not a “for all centers and partitions” guarantee. All our results are summarized in Table 1.

Candidate centers.

We consider also a more general variant of k-clustering, where the candidate centers are part of the input (given either explicitly or implicitly): Given a dataset P and candidate-centers set Q, the goal is to find CQ of size |C|k that minimizes pPdistz(p,C). When Q=d or Q=P, we obtain the continuous and discrete variants, respectively.

We observe a slightly different phenomenon in terms of the attainable contraction: to get (1ε)-contraction, one needs target dimension Θ(ε2log|Q|), and the lower bound holds even when both P and Q are doubling and k=1. We can still obtain claims analogous to Theorems 1.2 and 1.4, albeit with relaxed contraction: a “for all partitions and centers” using dimension t=O~(ε2(ddim(PQ)+logk+loglogn)), and a “for all centers” for the same target dimension but without the loglogn term. See Table 1 for references.

Table 1: Summary of our results for dimension reduction for k-clustering. The notions of “for all” centers and/or partitions, and of multiplicative/relaxed contraction are as explained in Section 1.2. Some lower bounds apply even for preserving the optimal value; for clarity, it is noted in the table they hold “even for value”. In the setting of candidate centers, the size of the candidate set is denoted by s. Suppressing log1ε terms and the dependence on α for α-relaxed contraction.
Problem Target dimension partitions centers contraction Reference
Continuous O(ε2logk) yes no multiplicative [32]
Ω(ε2logk) no no even for value [33]
>d1 no yes even for relaxed Thm 6.1
Discrete O(ε2(ddim+logk+loglogn)) yes yes multiplicative Thm 3.1
O(ε2(ddim+logk)) no yes relaxed Thm 5.1
? yes no any OPEN
Ω(ε2loglogn) yes yes even for relaxed Thm 6.3
Ω(ε2loglogn) no yes multiplicative Thm 6.2
Ω(ε2logk) no no even for value [33]
Ω(ε2ddim) no no even for value [4]
Candidate O(ε2logs) yes yes multiplicative Thm 4.1
centers O(ε2(ddim+logk+loglogn)) yes yes relaxed Thm 4.2
O(ε2(ddim+logk)) no yes relaxed Thm 5.1
Ω(ε2logs) no yes multiplicative Thm 6.4

1.3 Other related work

Besides the aforementioned results for “beyond JL” for clustering problems, there are also several improved bounds for other classes of problems such as Max-Cut [28, 27, 5], numerical linear algebra [31, 37, 7], and other applications [1, 15, 21].

The discrete k-median problem in Euclidean space was originally shown to be NP-hard by Papadimitriou, even for the case of d=2 [34]. In terms of hardness of approximation, the current state of the art is that one cannot approximate the discrete k-means or k-median problem beyond 1.07 and 1.17, respectively, assuming PNP [11, 12]. As for upper bounds, the best approximation factors known in polynomial time are 2+ε for any fixed ε>0 for discrete Euclidean k-median [10] and 5.912 for discrete Euclidean k-means [8]. There are also algorithms that achieve 1+ε approximation (again in the discrete case) in time that is doubly exponential in the doubling dimension, see [9] for a thorough discussion.

The discrete variant that we study may also be preferred over the continuous version in certain applications. First, it is thought to be less sensitive to outliers in practice than the continuous version [35, 25]. Second, in applications where cluster centers are used as data summarization, interpretability might require the centers to be part of the dataset. For example, in applications based on machine-learning embeddings of objects such as text [38], an arbitrary vector in the embedding space might not represent any actual object. A similar issue arises for structured data such as sparse data or images, e.g., the “average image” is visually random noise [30, 36] or the average of sparse vectors is not necessarily sparse. A discrete center, however, represents an actual underlying object, and thus preserves the underlying properties of the input points.

1.4 Technical overview

Since the dimension-error tradeoff behaves differently between the discrete and continuous settings, it is not surprising that our results for the discrete setting require new techniques. To simplify the discussion, we focus on the k-medoids (z=1) case, and an alternative guarantee that only preserves the optimal value, i.e.,

opt(G(P))(1±ε)opt(P), (1)

with target dimension bound t=O~(ε2(ddim(P)+logk)) which is the same as that in Theorem 1.2. While this is a weaker guarantee than both Theorem 1.1 and Theorem 1.2, it already introduces major technical challenges, and the techniques for this claim covers most of our new ideas.

We begin our discussion with the case k=1. We first argue that even for this case, a natural framework based on extension theorems (which has been used in previous works on dimension reduction for clustering) fails in our discrete case.

Failure of extension theorems in the discrete setting.

To prove (1) (and possibly more general claims), a natural framework based on extension theorems have been widely used in dimension reduction for clustering. Specifically, given an arbitrary center v in the target space (e.g., v is the optimal 1-median center of G(P)), one can define an “inverse image” u in the original space such that cost(P,u)(1+ε)cost(G(P),v), and this directly implies opt(G(P))11+εopt(P). The key step of defining “inverse image” is precisely what an extension theorem does. This framework is widely used in prior works such as [32, 22, 4], in the spirit of the classic Kirszbraun extension theorem [26] or the robust one-point extension theorem [32, Theorem 5.2]. However, such extension theorems are only known to work in the continuous setting, which require to pick the inverse image ud from the entire d and cannot be restricted only to the data points uP.111We note that the Kirszbraun theorem may be adapted to work for the discrete case when the target dimension t=O(logn), but this dimension bound is too large to be useful.

Our techniques.

We start with k=1 case (a detailed discussion can be found in Section 1.4.1). In this case, we first obtain a target dimension bound with an O(loglogn) factor, by utilizing the existence of a small movement-based coreset. A coreset is a small accurate proxy of the dataset, and the movement-based coreset additionally requires the existence of a “local” mapping such that each data point can be mapped to a nearby coreset point. The dimension reduction simply preserves the pairwise distance on the coreset, and (1) is argued via the local mapping. A conceptually similar coreset-to-dimension-reduction idea has been employed in [4] for continuous k-median, and one main difference is that we also utilize the locality property (movement-based) of the coreset, see a more detailed comparison in Section 1.4.1.

Then, to remove the O(loglogn) factor, we consider a weaker guarantee as in Theorem 1.2, where we prove the (1+ε) relative error only for near-optimal solutions, and for the other solutions we have a flat 100opt(P) error. This relaxed guarantee is strong enough for (1) (and many other applications), which may be of independent interest to further studies. Our analysis is crucially built on this small vs large cost case, albeit we also need to consider the middle ground of the mix of the two.

Finally, we discuss the generalization to k>1 in Section 1.4.2, which introduces several nontrivial technical complications from k=1.

1.4.1 The 𝒌 = 1 case

The easy side of (1) is the upper bound opt(G(P))(1+ε)opt(P), even for the general k case. The reason is that it suffices to preserve the cost w.r.t. an optimal center set C, and since C is a fixed solution, even a target dimension t=O(ε2log(1/ε)) will be sufficient. This is a standard argument also observed in prior works. The lower bound opt(G(P))(1ε)opt(P) is the major challenge. To prove this inequality, we want to preserve the clustering cost w.r.t. the optimal center set of G(P), denoted by C. Since C is a random set that depends on G, preserving its cost is almost the same as preserving the cost of all center sets, which is exactly the guarantee 2 of Theorems 1.1 and 1.2.

To introduce our new techniques, we first establish a weaker target dimension bound of O(ε2(ddim+loglogn)), and this part contains main ideas for proving Theorem 1.1. We then overview the key steps to eliminate the extra loglogn term, which also reflects how we prove Theorem 1.2.

The 𝑶(𝐥𝐨𝐠𝐥𝐨𝐠𝒏) bound: from coreset to dimension reduction.

To prove (1), we use an approach inspired by the movement-based coreset construction in Euclidean spaces [17]. Roughly speaking, a movement-based coreset222This definition is tailored to our need and may be slightly different to that in the literature. is a subset SP, such that there exists a mapping σ:PS satisfying pPpσ(p)O(ε)opt(P). Our framework is summarized as follows: we first construct a movement-based coreset S to compress the dataset P. Next, we apply the standard JL lemma to preserve pairwise distances in the coreset S within (1±ε), which requires O(ε2log|S|) target dimensions. After this step, the optimal value of S is already preserved, namely, opt(G(S))(1±ε)opt(S). Finally, it suffices to show that the cost of snapping data points to their nearest neighbor in S (i.e., pPpS(p) and pPGpGS(p)) is negligible in both original and target spaces.

The construction of the coreset is essentially the same as that in [17], except that [17] also assigns weight to the coreset points and here we only need the point set itself. We review the construction. This construction is based on a sequence of nets, a standard tool for discretizing metrics. Formally, a ρ-net of a point set P is a subset NP, such that 1) the interpoint distances in N are at least ρ, and 2) every point in P has a point in N within distance ρ. (See the more detailed definition in Definition 2.3). Denote cP as an optimal discrete 1-median center. We construct nets on a sequence of balls centered at c with geometrically decreasing radii. Denote r0:=opt(P) and r:=r0/2 for =1,2,,logn. Construct the level net N as an εr-net on the ball B(c,r), and denote N:==0lognN to be the union of all logn levels of nets.

By the standard packing property of doubling metrics, each net has size |N|O(εO(ddim)), thus |N|O(εO(ddim)logn), which implies a target dimension t=O(ε2(ddimlogε1+loglogn)). On the other hand, let G(c)G(P) be an optimal discrete 1-median center of G(P). Then the total cost of snapping c and all data points to the nearest neighbor in N (i.e., pP(pN(p)+cN(c))) can be bounded by O(ε)(opt(P)+cost(P,c)) in the original space. Based on results in [20], we further show that this snapping cost in the target space (i.e., pP(GpGN(p)+GcGN(c))) can increase by at most a constant factor.

Finally, we note that the above analysis can be applied to obtain the “for all centers” guarantee in Theorem 1.1, or even the stronger “for all centers and partitions” guarantee in Theorem 1.4.

Comparison to [4].

The locality property (movement-based) of the coreset is crucial to our analysis. To see this, let us investigate the proof in [4], which also uses coresets for dimension reduction but only works for the continuous setting. Specifically, it shows the existence of a coreset S of P which with high probability satisfies: (i) S is a coreset of P; (ii) the pairwise distances within S are preserved by G within (1±ε)-factor; and (iii) G(S) is a coreset of G(P). To see why this implies the desired bound opt(G(P))(1ε)opt(P), let vm be an optimal center for continuous 1-median of G(P). By property (iii), opt(G(P))(1ε)cost(G(S),v). Property (ii) and the Kirszbraun Extension Theorem [26] imply, as explained in [4], the existence of ud (intuitively, a pre-image of v) such that cost(G(S),v)(1ε)cost(S,u). Finally, by property (i) we can bound cost(S,u)(1ε)cost(P,u)(1ε)opt(P).

This elegant argument for the continuous 1-median breaks for discrete variant at the step of finding u that is a pre-image of an optimal center vG(P). This issue may seem surprising, because u:=G1(v)P is a natural candidate. However, it is not clear whether cost(G(S),v)(1ε)cost(S,u) holds, because property (ii) only preserves pairwise distances within S, and it is possible that vG(S). Another option is to define u via the Kirszbraun Extension Theorem, but then u is not guaranteed to be in P, i.e., might be an infeasible solution, hence we can no longer argue that cost(P,u)opt(P). Therefore, in the discrete setting we need to utilize more structures of the coreset, and the locality properties of movement-based coresets are key for resolving the issue.

Removing the 𝐥𝐨𝐠𝐥𝐨𝐠𝒏 term via relaxed guarantee

Let us first recall the cause of the loglogn term. We apply the JL Lemma to N, which is a union of logn nets, each of size εO(ddim). The loglogn thus comes from a union bound over all logn levels. To bypass this union bound, we use two technical ideas. First, we avoid touching cross-level pairs and only apply the union bound for each N separately. This requires us to always snap p and c to the same level of net when handling each pP. Second, for a single level, we analyze its maximum distance distortion which is a random variable, and bound the expectation. We remark that some levels will be distorted significantly, but the average distortion is (1+O(ε)). Similar ideas have been used by prior works (e.g., [14]).

Consider the following two extremes. First, suppose c is the closest point to c, say, pP,ccpc. For every pP, we can snap p to its nearest neighbor in net Np. Observe that c can also be covered by Np. The cost of snapping p and c can both be bounded by O(ε)pc, and we show that on average, the cost of snapping Gp and Gc is bounded by O(ε)pc as well, which adds up to O(ε)opt(P). The other extreme is that c is very far from c, i.e, cc>opt(P)/10. In this case, we can no longer snap c to the same net as p (like the previous case). We show that in this case, cost(G(P),Gc)100opt(P).

If c does not fall into any of the above two extremes, our analysis is a combination of them. Indeed, we show the relaxed “for all centers” guarantee,

cP,cost(G(P),Gc)min{(1ε)cost(P,c),100opt(P)}. (2)

Note that this is exactly the same as the guarantee 2 of Theorem 1.2, and that the two terms in the min correspond to the aforementioned two extremes, respectively. Specifically, we first specify a level and its corresponding radius r. If cc>r, then we fall into the second extreme and show that cost(G(P),Gc)100opt(P). Otherwise, ccr, then we handle each pP differently, depending on the distance pc. If pcr, then we use the same argument as the first extreme – snapping both p and c to Np, bounding the snapping cost, and analyzing the additive contraction. If pc<r, then we snap both p and c to N. Since is a fixed level, a union bound over N is affordable and we obtain cost(G(P),Gc)(1ε)cost(P,c) in this case.

1.4.2 Generalization to 𝒌>𝟏

Instead of directly generalizing (2), we first show a weaker guarantee: for target dimension t=O(ε2ddimlogk),

CP,|C|=k,pPGpGC(p)min{(1ε)cost(P,C),100opt(P)}, (3)

where C(p) is the center in C closest to p. Note that (3) is weaker than what we desire in Theorem 1.2, for the following two reasons. First, the target dimension is worse than the O(ε2(ddim+logk)) in Theorem 1.2. Second, the left hand side of (3) can be much larger than cost(G(P),G(C)), since the image of C(p) under G (i.e., GC(p)) is not necessarily the nearest neighbor of Gp in G(C). Nonetheless, the proof of (3) already captures most of our key ideas. In the end of this section, we briefly discuss how we obtain a sharper target dimension bound as well as a stronger guarantee.

Suppose CP is an optimal solution, which induces a clustering 𝒞={S1,S2,,Sk}. Our general proof framework is the same as the k=1 case – considering the “distance” between C and C, if C is “far from” C, then we show cost(G(P),G(C))100opt(P); otherwise we show cost(G(P),G(C))(1ε)cost(P,C).

However, an immediate issue is how to define that C and C are far from or close to each other. For each i[k], we specify a “threshold level” of cluster Si, denoted by i. We say C is “far from” C if there exists i[k], such that dist(ci,C)>10ri. In this case, the cost of connecting B(ci,ri) to C is already high. We further prove that cost(G(P),G(C))100opt(P), by careful analysis of the randomness of G.

Now suppose C is “close to” C, i.e., i[k],dist(ci,C)10ri. Our key observation is that for every pSi, C(p) should also be close to ci, i.e.,

pSi,C(p)ciO(max{pci,ri}). (4)

As a natural generalization of the k=1 case, we lower bound GpGC(p) for pSi differently, depending on the distances C(p)ci. If C(p)ciri, then we snap both p and C(p) to the (enlarged) net Np. (We can do this since (4) holds.) Otherwise, we snap both p and C(p) to the (enlarged) net Ni. The snapping cost and the distance contraction are bounded similarly to the k=1 case. This simply introduces an extra logk factor in the target dimension.

Decoupling 𝐝𝐝𝐢𝐦 from 𝐥𝐨𝐠𝒌.

So far, we only obtain an Oε(ddimlogk) bound, instead of Oε(ddim+logk). This is due to error accumulation: Recall we handle each (optimal) cluster Si separately, each of which incurs an O(ε)opt(P) additive error; hence, we have to rescale ε by a 1/k factor to compensate the accumulated error of k clusters, resulting in an O(ε2ddimlogk) target dimension (naïvely, that results in O~(ε2k2ddim) target dimension, but this is avoided by an easy adaptation).

To decouple these two factors, we need more delicate analysis for the error. For “far” points pSi with C(p)ciri, the snapping and distortion error is O(ε)pci in expectation, which adds up to O(ε)opt(P) and does not incur any error accumulation. However, the error accumulation happens for “close” points p with C(p)ci<ri, where the snapping cost within a single cluster Si, namely pSipN(p), is already O(ε)opt(P), which accumulates to O(kε)opt(P).

To reduce the error accumulation, we further divide the close points (i.e., C(p)ci<ri) into two ranges, namely, the close range C(p)ci<ri/k and the middle range C(p)ci[ri/k,ri], and handle these two ranges differently. The cost of points in the close range can be bounded by O(ε/k)opt(P), which adds up to O(ε)opt(P). For points in the middle range, we handle them in a point-by-point manner, at the cost of poly(k)eΩ(ε2t) per point. Since there are at most kO(logk) levels in the middle range, a union bound over all net points at these levels will be affordable.

Handling nearest neighbor assignment in the target space.

Recall that (4) conerns the cost GpGC(p), which is the cost in the target space with respect to the nearest neighbor assignment in the original space. However, what we really need is the nearest neighbor assignment in the target space. To capture such misalignment in the original and target spaces, we define a mapping f to be the assignment in the target space, i.e., f(p) is the center in C realizing dist(Gp,G(C)), so that cost(G(P),G(C))=pPGpGf(p), and f(p)=C(p) does not hold in general. We attempt to modify the previous analysis to lower bound each GpGf(p) instead of GpGC(p).

To lower bound this distance, we attempt to replace every C(p) with f(p) in our previous proof. The analysis becomes problematic, as our structural observation (4) no longer holds if we change C(p) to f(p), and this turns out to be the only place where our analysis does not go through. To resolve this issue, let us focus on the bad scenario where f(p) is sufficiently far from ci, i.e., f(p)cimax{pci,ri}. This implies f(p) is also far from p. We further show that GpGf(p)pci by careful analysis of G’s randomness. On the other hand, we have pC(p)O(pci) by (4). Therefore, we can directly lower bound GpGf(p) by pC(p) in this case.

2 Preliminaries

Consider a point set Pd. For every xd, denote by P(x) the point in P closest to x and dist(x,P):=xP(x) (recall that throughout is the Euclidean norm). Denote diam(P):=max{dist(p,q):p,qP} as the diameter of P. For xd and r>0, denote by B(x,r):={yd:|xyr} the ball centered at x with radius r. Recall that for k and z1, the (k,z)-clustering cost of P w.r.t. center set Cd,|C|k is costkz(P,C):=pPdist(p,C)z. The optimal discrete (k,z)-clustering cost of P w.r.t. a candidate center set Qd is denoted by optkz(P,Q):=minCQ,|C|kcostkz(P,C), and by opt(P,Q) for short when k,z are clear from the context. Denote opt(P):=opt(P,P) and optcont(P):=opt(P,d) for simplicity.

We use the following generalized triangle inequalities.

Lemma 2.1 (Generalized triangle inequalities [32]).

Let (X,dist) be a metric space. Then for every z1, ε(0,1) and p,q,rX,

dist(p,q)z(1zε)dist(p,r)zεzdist(q,r)z.
dist(p,q)z(1+ε)z1dist(p,r)z+(1+εε)z1dist(q,r)z.

2.1 Doubling dimension and nets

Definition 2.2 (Doubling dimension [16]).

The doubling dimension of a set Pd, denoted ddim(P), is the minimum m>0, such that r>0, every ball in P with radius r can be covered by at most 2m balls of radius r/2.

Our proof uses ρ-nets for doubling sets, whose definition and key properties are described here.

Definition 2.3 (ρ-net).

Let Pd and ρ>0. A subset NP is called a ρ-packing of P if u,vN,uv>ρ. The subset N is called a ρ-covering of P if xP, there exists uN such that xB(u,ρ). The subset N is called a ρ-net of P if N is both a ρ-packing and ρ-covering of P.

Lemma 2.4 (Packing property [16]).

Let Pd and NP be a ρ-packing of P. Then |N|(diam(P)/ρ)O(ddim(P)).

2.2 Dimension reduction

For simplicity, we only consider random linear maps defined by a matrix of iid Gaussians, which are known to satisfy the JL Lemma [19, 13].

Definition 2.5.

A Gaussian JL map is a t×d matrix with i.i.d. entries drawn from N(0,1t).

Recall the following concentration bound [20, Eq. (7)] (see also [33, Eq. (5)]), from which one can deduce the JL lemma.

Lemma 2.6 ([20, Eq. (7)]).

Let xd,ε>0 and a Gaussian JL map Gt×d. We have

Pr(Gx(1±ε)x)exp(ε2t/8).

The following two lemmas regard Gaussian JL maps when applied to doubling sets.

Lemma 2.7 ([20, Lemma 4.2]).

There exist universal constants A1,A2>0 such that for every subset PB(0,1) of the Euclidean unit ball in d, t>A1ddim(P)+1,D10, and a Gaussian JL map Gt×d,

Pr(xP,Gx>D)eA2tD2.
Lemma 2.8 ([18, Lemma 3.21]).

There exists universal constants A1,A2,L>1, such that for every PdB(0,1),ε>0,t>A1ddim(P), and a Gaussian JL map Gt×d,

Pr(xP,Gx<1L)eA2t.

3 The first upper bound

We prove Theorem 1.1 (a.k.a Theorem 1.4) in this section, formally stated below.

Theorem 3.1.

Let ε>0, z1 and d,ddim,k and a Gaussian JL map Gt×d with suitable t=O(z2ε2(ddimlog(z/ε)+logk+loglogn)). For every set Pd with ddim(P)ddim, with probability at least 2/3,

  1. 1.

    optkz(G(P))(1+ε)optkz(P), and

  2. 2.

    for all centers C=(c1,,ck)P and all partitions 𝒫=(S1,,Sk) of P,

    costkz(G(𝒫),G(C))(1ε)costkz(𝒫,C),

    where costkz(𝒫,C)=i=1kpSipciz.

We use the following lemma to bound the clustering cost of a fixed set of centers and partition of P. The proof can be found in the full version [23].

Lemma 3.2.

Let ε>0, z1 and d,k and a Gaussian JL map Gt×d with suitable t=O(z2ε2logε1). For every set Pd, every set of centers (c1,,ck)d and every partition 𝒫=(S1,,Sk) of P, with probability at least 9/10,

costkz(G(𝒫),G(C))(1+ε)costkz(𝒫,C).
Proof of Theorem 3.1.

Consider an optimal discrete k-median of P. Denote by C={c1,,ck}P and by S1,,Sk the centers and clusters (respectively) in that solution. Applying Lemma 3.2 to the optimal center set C and the partition 𝒫=(S1,,Sk), we have that with probability at least 9/10,

opt(G(P))cost(G(𝒫),G(C))(1+ε)cost(𝒫,C)=(1+ε)opt(P),

concluding the first part of the theorem.

Denote by r0 the largest radius of any cluster Si. Pick a suitable m=O(logn) such that 2m=n10. For i[0,m] and j[k], set ri=r0/2i, and Pij=SjB(cj,ri), i.e., for every cluster, we have a sequence of geometrically decreasing balls. Additionally, let Ni be an ε3ri-net of jPij. By Lemma 2.4, |Ni|kεO(ddim(P)).

For each x,yi[0,m]Ni, by Lemma 2.6,

Pr(GxGy>(1+ε)xy)exp(ε2t/8)εΩ(ddim(P))k2m2.

Thus, by a union bound, w.p. at least 9/10,

x,yi[0,m]Ni,GxGy(1+ε)xy. (5)

Furthermore, for each i[0,m],yNi, by Lemma 2.7,

Pr(pPB(y,ε3ri)s.t.G(py)>10ε3ri)exp(Ω(t)).

By a union bound, w.p. at least 9/10,

i[0,m],yNi,pPB(y,ε3ri),G(py)10ε3ri. (6)

By another union bound, Equations 5 and 6 hold with probability at least 2/3.

We are now ready to prove the second part of the theorem. Let C={c1,,ck}P and let a partition 𝒫=(S1,,Sk) of P. For every pP we denote by up the nearest net-point to p in the level such that PiPi+1 contains p, and the radius of that level is denoted rp. Denote by f(p) the center in C assigned to p according to the partition 𝒫. Recall that C(p) is a point in C that is nearest to p. Observe that

pPrpzn(r0n10)z+j=1ki=0m1pPi,jPi+1,j(2pcj)z=O(2z)opt(P),

and

(12rf(p))z f(p)C(f(p))z by definition
f(p)C(p)z C(f(p)) is nearest to f(p) from C
2z1pf(p)z+2z1pC(p)z by Lemma 2.1. (7)

Therefore,

cost(G(𝒫),G(C))
pPGpGf(p)z
pP(1zε)GupGuf(p)zεzGpGupzεzGf(p)Guf(p)z
pP(1zε)(1ε)zupuf(p)zεz(10ε3rp)zεz(10ε3rf(p))z
pP(1zε)2(1ε)zpf(p)zO(ε)zrpzO(ε)zrf(p)z
pP(13zε)pf(p)zO(ε)zrpzO(ε)z22z1(pf(p)z+pC(p)z)
(1O(zε))cost(𝒫,C)O(ε)opt(P),

where the first and the third inequalities follow from the generalized triangle inequality (Lemma 2.1), the second inequality follows from (5) and (6) and the fourth inequality follows from (7). Rescaling εε/z concludes the proof.

4 General candidate centers

We now consider a generalization of Theorem 3.1, to the setting where the centers are from a (possibly different than the input) candidate set Q. Unfortunately, to obtain multiplicative contraction in this setting, we have to pay Θ(ε2log|Q|) in the target dimension. We state the upper bound below, and the matching lower bound is provided in Theorem 6.4. The proofs appear in the full version [23].

Theorem 4.1.

Let ε>0, z1 and d,k,s and a Gaussian JL map Gt×d with suitable t=O(z2ε2(logs+zlog(z/ε))). For every set Pd and every candidate center set Qd with |Q|=sk, with probability at least 2/3,

  1. 1.

    optkz(G(P),G(Q))(1+ε)optkz(P,Q), and

  2. 2.

    for every C=(c1,,ck)Q and every partition 𝒫=(S1,,Sk) of P,

    costkz(G(𝒫),G(C))(1ε)costkz(𝒫,C),

    where costkz(𝒫,C)=i=1kpSipciz.

To bypass the O(ε2log|Q|) barrier in the target dimension, we consider relaxed contraction, and prove the following.

Theorem 4.2.

Let ε>0, z1 and d,ddim,k and a Gaussian JL map Gt×d with suitable t=O(z2ε2(ddimlog(z/ε)+logk+loglogα+loglogn)). For every n-point set Pd and every candidate center set Qd with ddim(PQ)ddim, with probability at least 2/3,

  1. 1.

    optkz(G(P),G(Q))(1+ε)optkz(P,Q), and

  2. 2.

    for every C=(c1,,ck)Q and every partition 𝒫=(S1,,Sk) of P,

    costkz(G(𝒫),G(C))min{αoptkz(P,Q),(1ε)costkz(𝒫,C)},

    where costkz(𝒫,C)=i=1kpSipciz.

5 Improved upper bound: Removing the 𝐥𝐨𝐠𝐥𝐨𝐠𝒏 term

We prove Theorem 1.2 in this section. In fact, we prove the following for the more general candidate centers setting, and get Theorem 1.2 by setting Q=P.

Theorem 5.1.

Let 0<ε<12, z1, α>2 and d,ddim and a Gaussian JL map Gt×d with suitable t=O(z2ε2(ddimlog(z/ε)+zlog(z/ε)+logk+loglogα)), the following holds. For every P,Qd with ddim(PQ)ddim, with probability at least 2/3,

  1. 1.

    optkz(G(P),G(Q))(1+ε)optkz(P,Q), and

  2. 2.

    CQ,|C|=k,

    costkz(G(P),G(C))min{αoptkz(P,Q),(1ε)costkz(P,C)}.

Consider an optimal discrete k-median of P w.r.t. candidate center set Q. Denote by C={c1,,ck}Q and by S1,,Sk the centers and clusters (respectively) in that solution. Denote r0:=opt(P,Q)1/z. For and i[k], set r=r0/2, and Pi=SiB(ci,r), i.e., for every cluster, we have a sequence of geometrically decreasing balls. Additionally, let Ni be an ε3r-net of (PQ)B(ci,rlogε1). Let N:=iNi.

For pPQ, recall C(p) is the closest center to p in C. Let jp be the level satisfying rjp+1pC(p)rjp. Denote rp:=rjp for simplicity. We have the following claim.

Lemma 5.2.

pPrpz2zoptkz(P,Q).

For CQ and pP, recall we denote by C(p) the point closest to p in C. We have the following lemma that upper bounds the distance from C(p) to C(p) (and also the distance from C(p) to p).

Lemma 5.3.

Let CQ. Then for every i[k] and pSi, it holds that C(p)ci4max{rp,ciC(ci)}.

Proof.
C(p)ci C(p)p+pci by triangle inequality
C(ci)p+pci C(p) is the point closest to p in C
C(ci)ci+cip+pci by triangle inequality
4max{rp,ciC(ci)}.

Proof of Theorem 5.1.

The first guarantee is the same as Theorem 3.1, so we omit its proof and focus on the second guarantee. For a generic solution CQ,|C|=k, denote C={c1,c2,,ck}. Denote f(p):=G1(GC(Gp)), i.e., f(p) is a center in C realizing dist(Gp,G(C)). For j[k], denote Sj:={pP:f(p)=cj} as the cluster induced by cj.

For every i[k], define the “threshold level” of cluster i as

i:=max{:|Pi|rz>αopt(P,Q)}. (8)

We also define the i-th “buffer” as Ii:=[ilog(2000L2),i+log(αk)], where L is the (sufficiently large) constant in Lemma 2.8.

For 0m, denote random variable β to be the minimum real, such that u,vN,GuGv(1εβε)uv. Denote random variable γ to be the minimum real, such that uN,vB(u,ε3r),GuGvγε3r. For pPQ, write βp:=βjp and γp:=γjp for simplicity.

In the following lemma, we define our good events and bound their success probability. The proof can be found in the full version [23].

Lemma 5.4.

With probability at least 0.99, the following events happen simultaneously.

  1. (a)

    pPβprpzeΩ(ε2t)opt(P,Q), and pPγpzrpz10zO(opt(P,Q)).

  2. (b)

    i[k],Ii,uN,vB(u,ε3r),GuGv10ε3r.

  3. (c)

    i[k],Ii, every net point uN satisfies that PP,

    pPGpGuz(1ε)3zpPpuzεk2opt(P,Q).
  4. (d)

    i[k],yB(ci,40Lri),GyGci400Lri.

  5. (e)

    i[k],y(PQ)B(ci,2000L2ri),GyGci>2000Lri.

  6. (f)

    For pP, denote by random variable ξp:=miny:yp>9LriGyGp. Then i[k],

    pPiiξpz>αopt(P,Q).
  7. (g)

    For pP, denote ηp:=miny:yp>9LrpGyGp. Then i[k],

    pSimax{0,(9rp)zηpz}eΩ(t)pSirpz.

The proof proceeds by a careful case analysis.

Case 1, one cluster with no cover: 𝐦𝐚𝐱𝟏𝒊𝒌{𝒄𝒊𝑪(𝒄𝒊)𝟏𝟎𝑳𝒓𝒊}>𝟎.

Then there exists i[k], such that ciC(ci)>10Lri. Intuitively, this means all points in C are far away from ci. Write

cost(G(P),G(C))cost(G(Pii),G(C))=pPiiGpGf(p)z. (9)

Note that for every pPii,

pf(p) pC(p)
C(p)cipci
ciC(ci)pci
>10Lrirp
9Lri.

Therefore, GpGf(p)ξp. Combining with (9) yields

cost(G(P),G(C))pPiiGpGf(p)zpPiiξpz>αopt(P,Q),

where the last inequality follows from event f.

Case 2, 𝐦𝐚𝐱𝟏𝒊𝒌{𝒄𝒊𝑪(𝒄𝒊)𝟏𝟎𝑳𝒓𝒊}𝟎.

Then for every i[k], ciC(ci)10Lri, which intuitively means every center in C has a nearby neighbor in C.

Comparing “fake” centers to optimal centers.

Let i[k]. For every pSi, we consider the distance of p’s “fake” center f(p) (recall, Gf(p) realizes dist(Gp,G(C))) from p’s optimal center ci. There are three ranges we consider for f(p)ci.

Define Ri:={pSi:ri/(αk)f(p)ci2000L2ri}, and denote R:=i=1kRi (called “the middle range”). Moreover, define Ti:={pSi:f(p)ciri/(αk)}, and denote T:=i=1kTi (called “the close range”).

Case 2.1, the middle range 𝒑𝑹.

Let us first lower bound GpGf(p) for pR. Assume C(p)=ci and f(p)=cj, where i,j[k]. Since pRi, we can assume r+1<cjcir for some level Ii. Let ui,j be the net point in N closest to cj. Then

GpGf(p)z (1zε)GpGui,jzεzGcjGui,jz
(1zε)GpGui,jzεz(10ε3r)z
(1zε)GpGui,jzO(ε)2zcjciz
(1zε)GpGui,jzO(ε)2zpcjzO(ε)2zpciz
=(1zε)GpGui,jzO(ε)2zpf(p)zO(ε)2zpC(p)z,

where the first and the last inequalities follow from Lemma 2.1 and the second inequality follows from event b. Summing over pR, we have

pRGpGf(p)z
=i=1kj=1kpRiSjGpGcjz
i=1kj=1kpRiSj((1zε)GpGui,jzO(ε)2zpf(p)zO(ε)2zpC(p)z)
i=1kj=1kpRiSj(1zε)GpGui,jzO(ε)2zpRpf(p)zO(ε)2zopt(P,Q)
Applying event c to net point ui,j and subset RiSj, we have
i=1kj=1k((1O(zε))pRiSjpui,jzεk2opt(P,Q))
O(ε)pRpf(p)zO(ε)opt(P,Q)
(1O(zε))pRpf(p)zO(ε)opt(P,Q)
(1O(zε))pRpC(p)zO(ε)opt(P,Q). (10)
Case 2.2, the close range 𝒑𝑻.

This is somewhat of a special case of Case 2.1. Assume C(p)=ci and f(p)=cj, where i,j[k]. Since pRi, we have cjcir for =i+log(αk). Let ui,j be the net point in N closest to cj. We have,

GpGf(p)z (1zε)GpGui,jzεzGcjGui,jz
(1zε)GpGui,jzO(ε)2zrz.

If pB(ci,ri+1), then

r12kpci12k(pcj+cjci)12k(pcj+r).

Rearranging, we obtain rpcj. Summing over pT, we have

pTGpGf(p)z
=i=1kj=1kpTiSjGpGcjz
i=1kj=1kpTiSj((1zε)GpGui,jz
O(ε)2zpcjz)O(ε)2z|PB(ci,ri+1)|rz
i=1kj=1kpTiSj((1zε)GpGui,jzO(ε)2zpcjz)O(εopt)
Applying event c to net point ui,j and subset TiSj, we have
i=1kj=1kpTiSj((1O(zε))pui,jzεk2optO(ε)2zpcjz)O(εopt)
i=1kj=1kpTiSj((1O(zε))pcjzεk2opt)O(εopt)
(1O(zε))pTpC(p)zO(ε)opt(P,Q). (11)
Case 2.3, the far range 𝒑𝑹𝑻.

We now consider points pSi(RT), i.e., f(p)ci2000L2ri. Suppose f(p)=cj. By e, GcjGci2000Lri.

Claim 5.5.

In this case, rp10Lri.

Proof.

Assume by contradiction that rp<10Lri. By Lemma 5.3, C(p)ci4max{rp,ciC(ci)}40Lri. Thus by d, Gp,GC(p)B(Gci,400Lri). Therefore,

GcjGciGcjGp+GpGciGC(p)Gp+GpGci800Lri,

contradiction. Therefore, by Lemma 5.3, C(p)ci4rp and hence

pC(p)pci+C(p)ci5rp. (12)

On a high level, as can be seen by the claim, we have that both f(p) and p are far from ci. We split into cases depending on which of p or f(p) is farther from ci (up to a constant), as follows.

Case 2.3.1, 𝒑𝑺𝒊(𝑹𝑻), and 𝒇(𝒑)𝒄𝒊>𝟏𝟎𝑳𝒓𝒑.

By triangle inequality,

pf(p)f(p)cipci9Lrp.

By the definition of ηp, we have GpGf(p)ηp. Therefore,

pSi(RT)f(p)ci>10LrpGpGf(p)z
pSi(RT)f(p)ci>10Lrpηpz since pf(p)9Lrp
pSi(RT)f(p)ci>10Lrp(9rp)zeΩ(t)pSirpz by event g
pSi(RT)f(p)ci>10Lrp(5rp)zeΩ(t)pSirpz
pSi(RT)f(p)ci>10LrppC(p)zeΩ(t)pSirpz by (12) (13)
Case 2.3.2, 𝒑𝑺𝒊(𝑹𝑻), and 𝒇(𝒑)𝒄𝒊𝟏𝟎𝑳𝒓𝒑.

Denote up and uf(p) to be the net points in Njp that are closest to p and f(p), respectively. Then

GpGf(p)z
(12zε)GupGuf(p)zεzGpGupzεzGf(p)Guf(p)z
(12zε)(1εβpε)zupuf(p)z2εz(γpε3rp)z
(13zεβpzε)upuf(p)zO(ε)2zγpzrpz
(13zεβpzε)pf(p)zO(ε)2zrpzO(ε)2zγpzrpz
Since pf(p)pci+f(p)cirp+10Lrp20Lrp, we have
(13zε)pf(p)zβpzε(20L)zrpzO(ε)2zrpzO(ε)2zγpzrpz.

Therefore,

pSi(RT)f(p)ci10LrpGpGf(p)z
(13zε)pSi(RT)f(p)ci10Lrppf(p)zzε(20L)zpSiβprpzO(ε)2zpSi(1+γpz)rpz. (14)
Wrap Up.

Combining (14) and (13), we have

pSi(RT)GpGf(p)z (13zε)pSi(RT)pC(p)z
zε(20L)zpSiβprpzO(ε)2zpSi(1+γpz)rpz.

Summing over i[k] yields

pP(RT)GpGf(p)z
(13zε)pP(RT)pC(p)zzε(20L)zpPβprpzO(ε)2zpP(1+γpz)rpz
(13zε)pP(RT)pC(p)zzε(20L)zeΩ(ε2t)opt(P,Q)O(ε)2zopt(P,Q)
(13zε)pP(RT)pC(p)zO(ε)opt(P,Q), (15)

where the second last inequality follows from event a and Lemma 5.2. Finally, we combine (10),(11) and (15) and obtain

cost(G(P),G(C))(1O(zε))cost(P,C)O(ε)opt(P,Q)(1O(zε))cost(P,C).

Rescaling εε/z concludes the proof.

6 Lower bounds

In this section, we state our lower bounds. For simplicity, we do not try to optimize the dependence on z. All lower bounds are presented for z=1. All the proofs are provided in the full version [23]. Denote by 0d the origin of d. For ease of presentation, we allow P to be a multi-set.

Theorem 6.1 (Continuous, for all centers).

Let n,d, and P={0d}n. Let G(d1)×d be any linear map. Then, there exists cd such that pPGpGc=0 and pPpc=n.

The next theorem shows that in order to bound the (multiplicative) contraction for all centers, we need either dimension Ω(loglogn), or to relax the definition of contraction (as is done in Theorem 5.1).

Theorem 6.2 (Discrete, for all centers).

Let n,d and ε(0,12). There exists Pd of size |P|=n and ddim(P)=Θ(1), such that if G is a Gaussian JL map onto dimension taε2loglogn for a sufficiently small constant a>0, then with probability at least 2/3, there exists cP such that pPGpGc(1ε)pPpc.

Next, we show that dimension Ω(loglogn) is necessary, even for the relaxed notion of contraction, for preserving all partitions and centers.

Theorem 6.3 (Discrete, for all partitions and centers).

Let n,d and ε(0,12). There exists Pd of size |P|=n and ddim(P)=Θ(1), such that if G is a Gaussian JL map onto dimension 1aε2taε2loglogn for a sufficiently small constant a>0, then with probability at least 2/3, there exists (c1,c2)P and a partition (P1,P2) of P such that

i{1,2}pPiGpGci<min{(1ε)i{1,2}pPipci,100opt(P)}.
Theorem 6.4 (Discrete, for all centers, with candidate center set).

Let n,s,d and ε(0,12). There exists P,Qd of sizes |P|=n,|Q|=s, and ddim(PQ)=O(1), such that if G is a Gaussian JL map onto dimension taε2logs for a sufficiently small constant a>0, then with probability at least 2/3, there exists cQ such that pPGpGc(1ε)pPpc.

References

  • [1] Yair Bartal, Ben Recht, and Leonard J Schulman. Dimensionality reduction: beyond the johnson-lindenstrauss bound. In Proceedings of the twenty-second annual ACM-SIAM symposium on Discrete Algorithms, pages 868–887. SIAM, 2011. doi:10.1137/1.9781611973082.68.
  • [2] Luca Becchetti, Marc Bury, Vincent Cohen-Addad, Fabrizio Grandoni, and Chris Schwiegelshohn. Oblivious dimension reduction for k-means: beyond subspaces and the Johnson-Lindenstrauss Lemma. In Proceedings of the 51st Annual ACM SIGACT Symposium on Theory of Computing, STOC, pages 1039–1050, 2019. doi:10.1145/3313276.3316318.
  • [3] Christos Boutsidis, Anastasios Zouzias, and Petros Drineas. Random projections for k-means clustering. In 24th Annual Conference on Neural Information Processing Systems, NeurIPS, pages 298–306. Curran Associates, Inc., 2010. URL: https://proceedings.neurips.cc/paper/2010/hash/73278a4a86960eeb576a8fd4c9ec6997-Abstract.html.
  • [4] Moses Charikar and Erik Waingarten. The Johnson-Lindenstrauss Lemma for clustering and subspace approximation: From coresets to dimension reduction. In SODA, pages 3172–3209. SIAM, 2025. doi:10.1137/1.9781611978322.102.
  • [5] Xiaoyu Chen, Shaofeng H.-C. Jiang, and Robert Krauthgamer. Streaming Euclidean Max-Cut: Dimension vs data reduction. In Proceedings of the 55th Annual ACM Symposium on Theory of Computing, STOC, pages 170–182, 2023. doi:10.1145/3564246.3585170.
  • [6] Michael B. Cohen, Sam Elder, Cameron Musco, Christopher Musco, and Madalina Persu. Dimensionality reduction for k-means clustering and low rank approximation. In Proceedings of the Forty-Seventh Annual ACM on Symposium on Theory of Computing, STOC, pages 163–172, 2015. doi:10.1145/2746539.2746569.
  • [7] Michael B. Cohen, Jelani Nelson, and David P. Woodruff. Optimal approximate matrix product in terms of stable rank. In 43rd International Colloquium on Automata, Languages, and Programming (ICALP 2016), volume 55 of LIPIcs, pages 11:1–11:14. Schloss Dagstuhl – Leibniz-Zentrum für Informatik, 2016. doi:10.4230/LIPIcs.ICALP.2016.11.
  • [8] Vincent Cohen-Addad, Hossein Esfandiari, Vahab Mirrokni, and Shyam Narayanan. Improved approximations for Euclidean k-means and k-median, via nested quasi-independent sets. In Proceedings of the 54th Annual ACM SIGACT Symposium on Theory of Computing, pages 1621–1628, 2022. doi:10.1145/3519935.3520011.
  • [9] Vincent Cohen-Addad, Andreas Emil Feldmann, and David Saulpic. Near-linear time approximation schemes for clustering in doubling metrics. Journal of the ACM, 68(6):1–34, 2021. doi:10.1145/3477541.
  • [10] Vincent Cohen-Addad, Fabrizio Grandoni, Euiwoong Lee, Chris Schwiegelshohn, and Ola Svensson. A (2+ ε)-approximation algorithm for metric k-median. In Proceedings of the 57th Annual ACM Symposium on Theory of Computing, pages 615–624, 2025. doi:10.1145/3717823.3718299.
  • [11] Vincent Cohen-Addad and C. S. Karthik. Inapproximability of clustering in Lp metrics. In 2019 IEEE 60th Annual Symposium on Foundations of Computer Science (FOCS), pages 519–539. IEEE, 2019. doi:10.1109/FOCS.2019.00040.
  • [12] Vincent Cohen-Addad, C. S. Karthik, and Euiwoong Lee. Johnson coverage hypothesis: Inapproximability of k-means and k-median in p-metrics. In Proceedings of the 2022 Annual ACM-SIAM Symposium on Discrete Algorithms (SODA), pages 1493–1530. SIAM, 2022. doi:10.1137/1.9781611977073.63.
  • [13] Sanjoy Dasgupta and Anupam Gupta. An elementary proof of a theorem of Johnson and Lindenstrauss. Random Struct. Algorithms, 22(1):60–65, 2003. doi:10.1002/rsa.10073.
  • [14] Jie Gao, Rajesh Jayaram, Benedikt Kolbe, Shay Sapir, Chris Schwiegelshohn, Sandeep Silwal, and Erik Waingarten. Randomized dimensionality reduction for Euclidean maximization and diversity measures. In Forty-second International Conference on Machine Learning, 2025. URL: https://openreview.net/forum?id=Rcivp36KzO.
  • [15] Lee-Ad Gottlieb and Robert Krauthgamer. A nonlinear approach to dimension reduction. Discrete & Computational Geometry, 54(2):291–315, 2015. doi:10.1007/s00454-015-9707-9.
  • [16] Anupam Gupta, Robert Krauthgamer, and James R. Lee. Bounded geometries, fractals, and low-distortion embeddings. In 44th Symposium on Foundations of Computer Science, FOCS, pages 534–543. IEEE Computer Society, 2003. doi:10.1109/SFCS.2003.1238226.
  • [17] Sariel Har-Peled and Soham Mazumdar. On coresets for k-means and k-median clustering. In STOC, pages 291–300. ACM, 2004. doi:10.1145/1007352.1007400.
  • [18] Lingxiao Huang, Shaofeng H.-C. Jiang, Robert Krauthgamer, and Di Yue. Near-optimal dimension reduction for facility location. In Proceedings of the 57th Annual ACM Symposium on Theory of Computing, STOC, pages 665–676, 2025. doi:10.1145/3717823.3718214.
  • [19] Piotr Indyk and Rajeev Motwani. Approximate nearest neighbors: Towards removing the curse of dimensionality. In Proceedings of the Thirtieth Annual ACM Symposium on the Theory of Computing, STOC, pages 604–613, 1998. doi:10.1145/276698.276876.
  • [20] Piotr Indyk and Assaf Naor. Nearest-neighbor-preserving embeddings. ACM Trans. Algorithms, 3(3):31, 2007. doi:10.1145/1273340.1273347.
  • [21] Zachary Izzo, Sandeep Silwal, and Samson Zhou. Dimensionality reduction for Wasserstein barycenter. Advances in neural information processing systems, 34:15582–15594, 2021. URL: https://proceedings.neurips.cc/paper/2021/hash/8346db44a721fa863ca38180638bad3d-Abstract.html.
  • [22] Shaofeng H.-C. Jiang, Robert Krauthgamer, and Shay Sapir. Moderate dimension reduction for k-center clustering. In 40th International Symposium on Computational Geometry (SoCG 2024), volume 293 of Leibniz International Proceedings in Informatics (LIPIcs), pages 64:1–64:16. Schloss Dagstuhl – Leibniz-Zentrum für Informatik, 2024. doi:10.4230/LIPIcs.SoCG.2024.64.
  • [23] Shaofeng H. C. Jiang, Robert Krauthgamer, Shay Sapir, Sandeep Silwal, and Di Yue. Dimension reduction for clustering: The curious case of discrete centers, 2025. Full version. doi:10.48550/arXiv.2509.07444.
  • [24] William B. Johnson and Joram Lindenstrauss. Extensions of Lipschitz mappings into Hilbert space. Contemporary mathematics, 26:189–206, 1984. doi:10.1090/conm/026/737400.
  • [25] Leonard Kaufman and Peter J. Rousseeuw. Finding groups in data: an introduction to cluster analysis. John Wiley & Sons, 2009. doi:10.1002/9780470316801.
  • [26] M. Kirszbraun. Über die zusammenziehende und lipschitzsche transformationen. Fundamenta Mathematicae, 22(1):77–108, 1934. doi:10.4064/fm-22-1-77-108.
  • [27] Christiane Lammersen. Approximation Techniques for Facility Location and Their Applications in Metric Embeddings. PhD thesis, Technische Universität Dortmund, 2010. doi:10.17877/DE290R-8506.
  • [28] Christiane Lammersen, Anastasios Sidiropoulos, and Christian Sohler. Streaming embeddings with slack. In 11th International Symposium on Algorithms and Data Structures, WADS, volume 5664 of Lecture Notes in Computer Science, pages 483–494. Springer, 2009. doi:10.1007/978-3-642-03367-4_42.
  • [29] Kasper Green Larsen and Jelani Nelson. Optimality of the Johnson-Lindenstrauss Lemma. In 58th IEEE Annual Symposium on Foundations of Computer Science, FOCS, pages 633–638, 2017. doi:10.1109/FOCS.2017.64.
  • [30] Jure Leskovec, Anand Rajaraman, and Jeffrey David Ullman. Mining of massive data sets. Cambridge University Press, 3rd edition, 2020. doi:10.1017/9781108684163.
  • [31] Michael W. Mahoney. Randomized algorithms for matrices and data. Foundations and Trends in Machine Learning, 3(2):123–224, 2011. doi:10.1561/2200000035.
  • [32] Konstantin Makarychev, Yury Makarychev, and Ilya P. Razenshteyn. Performance of Johnson-Lindenstrauss transform for k-means and k-medians clustering. In Proceedings of the 51st Annual Symposium on Theory of Computing, STOC, pages 1027–1038, 2019. doi:10.1145/3313276.3316350.
  • [33] Shyam Narayanan, Sandeep Silwal, Piotr Indyk, and Or Zamir. Randomized dimensionality reduction for facility location and single-linkage clustering. In Proceedings of the 38th International Conference on Machine Learning, ICML, volume 139 of Proceedings of Machine Learning Research, pages 7948–7957. PMLR, 2021. URL: http://proceedings.mlr.press/v139/narayanan21b.html.
  • [34] Christos H. Papadimitriou. Worst-case and probabilistic analysis of a geometric location problem. SIAM Journal on Computing, 10(3):542–557, 1981. doi:10.1137/0210040.
  • [35] Hae-Sang Park and Chi-Hyuck Jun. A simple and fast algorithm for K-medoids clustering. Expert Syst. Appl., 36(2):3336–3341, 2009. doi:10.1016/J.ESWA.2008.01.039.
  • [36] Mo Tiwari, Martin J Zhang, James Mayclin, Sebastian Thrun, Chris Piech, and Ilan Shomorony. Banditpam: Almost linear time k-medoids clustering via multi-armed bandits. Advances in Neural Information Processing Systems, 33:10211–10222, 2020.
  • [37] David P. Woodruff. Sketching as a tool for numerical linear algebra. Foundations and Trends in Theoretical Computer Science, 10(1–-2):1–157, 2014. doi:10.1561/0400000060.
  • [38] Junyuan Xie, Ross Girshick, and Ali Farhadi. Unsupervised deep embedding for clustering analysis. In Proceedings of The 33rd International Conference on Machine Learning, pages 478–487. PMLR, 2016. URL: https://proceedings.mlr.press/v48/xieb16.html.