Do you know what q-means?

TL;DR

This paper introduces classical and quantum algorithms for approximate k-means clustering, achieving significant runtime improvements and matching quantum lower bounds, with the quantum approach avoiding complex linear algebra primitives.

Contribution

It presents a classical $ ext{ extasciicircum}$$ ext{ extasciicircum}$-k-means algorithm with improved complexity and a novel quantum $q$-means algorithm that outperforms classical methods, both matching lower bounds.

Findings

Classical $ ext{ extasciicircum}$$ ext{ extasciicircum}$-k-means has reduced runtime complexity.

Quantum $q$-means achieves quadratic speedup over classical algorithms.

Both algorithms are shown to be optimal in most parameters.

Abstract

Clustering is one of the most important tools for analysis of large datasets, and perhaps the most popular clustering algorithm is Lloyd's algorithm for $k$-means. This algorithm takes $n$ vectors $V=[v_1,\dots,v_n]\in\mathbb{R}^{d\times n}$ and outputs $k$ centroids $c_1,\dots,c_k\in\mathbb{R}^d$; these partition the vectors into clusters based on which centroid is closest to a particular vector. We present a classical $\varepsilon$-$k$-means algorithm that performs an approximate version of one iteration of Lloyd's algorithm with time complexity $\tilde{O}\big(\frac{\|V\|_F^2}{n}\frac{k^{2}d}{\varepsilon^2}(k + \log{n})\big)$, exponentially improving the dependence on the data size $n$ and matching that of the "$q$-means" quantum algorithm originally proposed by Kerenidis, Landman, Luongo, and Prakash (NeurIPS'19). Moreover, we propose an improved $q$-means quantum algorithm with time complexity $\tilde{O}\big(\frac{\|V\|_F}{\sqrt{n}}\frac{k^{3/2}d}{\varepsilon}(\sqrt{k}+\sqrt{d})(\sqrt{k} + \log{n})\big)$ that quadratically improves the runtime of our classical $\varepsilon$-$k$-means algorithm in several parameters. Our quantum algorithm does not rely on quantum linear algebra primitives of prior work, but instead only uses QRAM to prepare simple states based on the current iteration's clusters and multivariate quantum amplitude estimation. Finally, we provide classical and quantum query lower bounds, showing that our algorithms are optimal in most parameters.