Sensitivity Sampling for $k$-Means: Worst Case and Stability Optimal Coreset Bounds

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Bansal, Nikhil, Cohen-Addad, Vincent, Prabhu, Milind, Saulpic, David, Schwiegelshohn, Chris
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866916232536522752
author Bansal, Nikhil
Cohen-Addad, Vincent
Prabhu, Milind
Saulpic, David
Schwiegelshohn, Chris
author_facet Bansal, Nikhil
Cohen-Addad, Vincent
Prabhu, Milind
Saulpic, David
Schwiegelshohn, Chris
contents Coresets are arguably the most popular compression paradigm for center-based clustering objectives such as $k$-means. Given a point set $P$, a coreset $Ω$ is a small, weighted summary that preserves the cost of all candidate solutions $S$ up to a $(1\pm \varepsilon)$ factor. For $k$-means in $d$-dimensional Euclidean space the cost for solution $S$ is $\sum_{p\in P}\min_{s\in S}\|p-s\|^2$. A very popular method for coreset construction, both in theory and practice, is Sensitivity Sampling, where points are sampled in proportion to their importance. We show that Sensitivity Sampling yields optimal coresets of size $\tilde{O}(k/\varepsilon^2\min(\sqrt{k},\varepsilon^{-2}))$ for worst-case instances. Uniquely among all known coreset algorithms, for well-clusterable data sets with $Ω(1)$ cost stability, Sensitivity Sampling gives coresets of size $\tilde{O}(k/\varepsilon^2)$, improving over the worst-case lower bound. Notably, Sensitivity Sampling does not have to know the cost stability in order to exploit it: It is appropriately sensitive to the clusterability of the data set while being oblivious to it. We also show that any coreset for stable instances consisting of only input points must have size $Ω(k/\varepsilon^2)$. Our results for Sensitivity Sampling also extend to the $k$-median problem, and more general metric spaces.
format Preprint
id arxiv_https___arxiv_org_abs_2405_01339
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Sensitivity Sampling for $k$-Means: Worst Case and Stability Optimal Coreset Bounds
Bansal, Nikhil
Cohen-Addad, Vincent
Prabhu, Milind
Saulpic, David
Schwiegelshohn, Chris
Data Structures and Algorithms
Coresets are arguably the most popular compression paradigm for center-based clustering objectives such as $k$-means. Given a point set $P$, a coreset $Ω$ is a small, weighted summary that preserves the cost of all candidate solutions $S$ up to a $(1\pm \varepsilon)$ factor. For $k$-means in $d$-dimensional Euclidean space the cost for solution $S$ is $\sum_{p\in P}\min_{s\in S}\|p-s\|^2$. A very popular method for coreset construction, both in theory and practice, is Sensitivity Sampling, where points are sampled in proportion to their importance. We show that Sensitivity Sampling yields optimal coresets of size $\tilde{O}(k/\varepsilon^2\min(\sqrt{k},\varepsilon^{-2}))$ for worst-case instances. Uniquely among all known coreset algorithms, for well-clusterable data sets with $Ω(1)$ cost stability, Sensitivity Sampling gives coresets of size $\tilde{O}(k/\varepsilon^2)$, improving over the worst-case lower bound. Notably, Sensitivity Sampling does not have to know the cost stability in order to exploit it: It is appropriately sensitive to the clusterability of the data set while being oblivious to it. We also show that any coreset for stable instances consisting of only input points must have size $Ω(k/\varepsilon^2)$. Our results for Sensitivity Sampling also extend to the $k$-median problem, and more general metric spaces.
title Sensitivity Sampling for $k$-Means: Worst Case and Stability Optimal Coreset Bounds
topic Data Structures and Algorithms
url https://arxiv.org/abs/2405.01339