A fast and recursive algorithm for clustering large datasets with k-medians

Hervé Cardot (1) , Peggy Cénac (1) , Jean-Marie Monnez (2, 3)

1 IMB - Institut de Mathématiques de Bourgogne [Dijon]
2 IECN - Institut Élie Cartan de Nancy
3 BIGS - Biology, genetics and statistics

Clustering with fast algorithms large samples of high dimensional data is an important challenge in computational statistics. A new class of recursive stochastic gradient algorithms designed for the k-medians loss criterion is proposed. By their recursive nature, these algorithms are very fast and are well adapted to deal with large samples of data that are allowed to arrive sequentially. It is proved that the stochastic gradient algorithm converges almost surely to the set of stationary points of the underlying loss criterion. A particular attention is paid to the averaged versions which are known to have better performances. A data-driven procedure that permits a fully automatic selection of the value of the descent step is also proposed. The performance of the averaged sequential estimator is compared on a simulation study, both in terms of computation speed and accuracy of the estimations, with more classical partitioning techniques such as k-means, trimmed k-means and PAM (partitioning around medoids). Finally, this new online clustering technique is illustrated on determining television audience profiles with a sample of more than 5000 individual television audiences measured every minute over a period of 24 hours.

Keywords: k-medoids; online clustering; stochastic gradient; stochastic approximation; Robbins Monro; recursive estimators; partitioning around medoids; averaging; high dimensional data

Domains: Statistics [math.ST]; Statistics Theory [stat.TH]

Hervé Cardot, Peggy Cénac, Jean-Marie Monnez. A fast and recursive algorithm for clustering large datasets with k-medians. Computational Statistics and Data Analysis, 2012, 56, pp.1434-1449. ⟨10.1016/j.csda.2011.11.019⟩. ⟨hal-00644683⟩