Web-scale K-means Clustering

Abstract

We present two modifications to the popular <i>k</i>-means clustering algorithm to address the extreme requirements for latency, scalability, and sparsity encountered in user-facing web applications. First, we propose the use of mini-batch optimization for <i>k</i>-means clustering. This reduces computation cost by orders of magnitude compared to the classic batch algorithm while yielding significantly better solutions than online stochastic gradient descent. Second, we achieve sparsity with projected gradient descent, and give a fast ε-accurate projection onto the <i>L</i>1-ball. Source code is freely available: http://code.google.com/p/sofia-ml