View source for K-means++

{{short description |Algorithm in data mining}}
{{DISPLAYTITLE:''k''-means++}}
In [[data mining]], '''''k''-means++'''<ref>{{Cite conference
 | title=''k''-means++: the advantages of careful seeding
 | url=http://ilpubs.stanford.edu:8090/778/1/2006-13.pdf
 |author1=Arthur, D.  |author2=Vassilvitskii, S.
 | book-title=Proceedings of the eighteenth annual ACM-SIAM symposium on Discrete algorithms
 | pages=1027&ndash;1035
 | year=2007
 | publisher=Society for Industrial and Applied Mathematics Philadelphia, PA, USA
}}</ref><ref>http://theory.stanford.edu/~sergei/slides/BATS-Means.pdf Slides for presentation of method by Arthur, D. and Vassilvitskii, S.
</ref> is an algorithm for choosing the initial values (or "seeds") for the [[k-means clustering|''k''-means clustering]] algorithm.  It was proposed in 2007 by David Arthur and Sergei Vassilvitskii, as an approximation algorithm for the [[NP-hard]] ''k''-means problem&mdash;a way of avoiding the sometimes poor clusterings found by the standard ''k''-means algorithm. It is similar to the first of three seeding methods proposed, in  independent work, in 2006<ref>{{Cite conference
 | title=The Effectiveness of Lloyd-Type Methods for the k-Means Problem
 |author=Ostrovsky, R. |author2=Rabani, Y. |author3=Schulman, L. J. |author4=Swamy, C.
 | book-title=Proceedings of the 47th Annual IEEE Symposium on Foundations of Computer Science (FOCS'06)
 | pages=165&ndash;174
 | year=2006
 | publisher=IEEE
}}</ref> by [[Rafail Ostrovsky]], Yuval Rabani, [[Leonard Schulman]] and Chaitanya Swamy. (The distribution of the first seed is different.)

==Background==
The ''k''-means problem is to find cluster centers that minimize  the intra-class variance, i.e. the sum of squared distances from each data point being clustered to its cluster center (the center that is closest to it).
Although finding an exact solution to the ''k''-means problem for arbitrary input is NP-hard,<ref>{{cite journal
 | doi=10.1023/B:MACH.0000033113.59016.96
 | title=Clustering Large Graphs via the Singular Value Decomposition
 |author=Drineas, P. |author2=Frieze, A. |author3=Kannan, R. |author4=Vempala, S. |author5=Vinay, V.
 | journal=Machine Learning
 | volume=56
 | issue=1–3
 | pages=9&ndash;33
 | year=2004
 | doi-access=free
 }}</ref> the standard approach to finding an approximate solution (often called [[Lloyd's algorithm]] or the ''k''-means algorithm) is used widely and frequently finds reasonable solutions quickly.

However, the ''k''-means algorithm has at least two major theoretic shortcomings:
* First, it has been shown that the worst case running time of the algorithm is super-polynomial in the input size.<ref>{{Cite conference
 | title=How slow is the ''k''-means method?
 |author1=Arthur, D.  |author2=Vassilvitskii, S.
 | book-title=Proceedings of the twenty-second annual symposium on Computational geometry
 | pages=144&ndash;153
 | year=2006
 | work=ACM New York, NY, USA
}}</ref>
* Second, the approximation found can be arbitrarily bad with respect to the objective function compared to the optimal clustering.

The ''k''-means++ algorithm addresses the second of these obstacles by specifying a procedure to initialize the cluster centers before proceeding with the standard ''k''-means optimization iterations.
With the ''k''-means++ initialization, the algorithm is guaranteed to find a solution that is O(log&nbsp;''k'') competitive to the optimal ''k''-means solution.

=== Example of a suboptimal clustering ===
[[File:Bad K-Means clustering.png|alt=Bad clustering of a rectangle|thumb|This is a bad clustering where the points A and D are in the red cluster of centroid E and the points B and C are in the blue cluster  of centroid F, as the intra-cluster distance in not minimal]]
To illustrate the potential of the ''k''-means algorithm to perform arbitrarily poorly with respect to the objective function of minimizing the sum of squared distances of cluster points to the centroid of their assigned clusters, consider the example of four points in <math>\mathbb{R}^2</math> that form an axis-aligned rectangle whose width is greater than its height.
[[File:Good clustering.png|thumb|Optimal clustering for the problem.]]
If <math>k = 2</math> and the two initial cluster centers lie at the midpoints of the top and bottom line segments of the rectangle formed by the four data points, the ''k''-means algorithm converges immediately, without moving these cluster centers. Consequently, the two bottom data points are clustered together and the two data points forming the top of the rectangle are clustered together&mdash;a suboptimal clustering because the width of the rectangle is greater than its height.

Consider now extending the rectangle in a horizontal direction to any desired width. The standard ''k''-means algorithm will continue to cluster the points suboptimally, and by increasing the horizontal distance between the two data points in each cluster, we can make the algorithm perform arbitrarily poorly with respect to the ''k''-means objective function.

==Improved initialization algorithm==
The intuition behind this approach is that spreading out the ''k'' initial cluster centers is a good thing: the first cluster center is chosen uniformly at random from the data points that are being clustered, after which each subsequent cluster center is chosen from the ''remaining'' data points with probability proportional to its squared distance from the point's closest existing cluster center.

The exact algorithm is as follows:
# Choose one center uniformly at random among the data points.
# For each data point {{math|<var>x</var>}} not chosen yet, compute D({{math|<var>x</var>}}), the distance between {{math|<var>x</var>}} and the nearest center that has already been chosen.
# Choose one new data point at random as a new center, using a weighted probability distribution where a point {{math|<var>x</var>}} is chosen with probability proportional to D({{math|<var>x</var>}})<sup>2</sup>.
# Repeat Steps 2 and 3 until {{math|<var>k</var>}} centers have been chosen.
# Now that the initial centers have been chosen, proceed using standard [[k-means clustering|''k''-means clustering]].

This seeding method yields considerable improvement in the final error of {{math|<var>k</var>}}-means. Although the initial selection in the algorithm takes extra time, the {{math|<var>k</var>}}-means part itself converges very quickly after this seeding and thus the algorithm actually lowers the computation time. The authors tested their method with real and synthetic datasets and obtained typically 2-fold improvements in speed, and for certain datasets, close to 1000-fold improvements in error. In these simulations the new method almost always performed at least as well as [[Vanilla (computing)|vanilla]] {{math|<var>k</var>}}-means in both speed and error.

Additionally, the authors calculate an approximation ratio for their algorithm. The {{math|<var>k</var>}}-means++ algorithm guarantees an approximation ratio O(log&nbsp;''k'') in expectation (over the randomness of the algorithm), where <math>k</math> is the number of clusters used. This is in contrast to vanilla {{math|<var>k</var>}}-means, which can generate clusterings arbitrarily worse than the optimum.<ref name=kanungo>{{citation
 |last1       = Kanungo
 |first1      = T.
 |last2       = Mount
 |first2      = D.
 |last3       = Netanyahu
 |first3      = N.
 |last4       = Piatko
 |first4      = C.
 |author4-link = Christine Piatko
 |last5       = Silverman
 |first5      = R.
 |last6       = Wu
 |first6      = A.
 |journal     = [[Computational Geometry (journal)|Computational Geometry: Theory and Applications]]
 |volume      = 28
 |issue      = 2–3
 |pages      = 89–112
 |title       = A Local Search Approximation Algorithm for ''k''-Means Clustering
 |year        = 2004
 |doi      = 10.1016/j.comgeo.2004.03.003
 |doi-access= free
 }}.</ref>
A generalization of the performance of k-means++ with respect to any arbitrary distance is provided in
.<ref name=nn>{{citation
 |last1       = Nielsen
 |first1      = Frank
 |last2       = Nock
 |first2      = Richard
 |title       = 2015 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP)
 |chapter = Total Jensen divergences: Definition, properties and clustering
 |year        = 2013
 |pages = 2016–2020
 |doi = 10.1109/ICASSP.2015.7178324
 |arxiv = 1309.7109
 |bibcode = 2013arXiv1309.7109N
 |isbn = 978-1-4673-6997-8
 |s2cid = 463728
 }}.</ref>

==Applications==
The ''k''-means++ approach has been applied since its initial proposal. In a review by Shindler,<ref>https://web.archive.org/web/20110927100642/http://www.cs.ucla.edu/~shindler/shindler-kMedian-survey.pdf Approximation Algorithms for the Metric ''k''-Median
Problem</ref> which includes many types of clustering algorithms, the method is said to successfully overcome some of the problems associated with other ways of defining initial cluster-centres for ''k''-means clustering. Lee et al.<ref>http://sir-lab.usc.edu/publications/2008-ICWSM2LEES.pdf {{Webarchive|url=https://web.archive.org/web/20160303223350/http://sir-lab.usc.edu/publications/2008-ICWSM2LEES.pdf |date=2016-03-03 }} Discovering Relationships among Tags and Geotags, 2007</ref> report an application of ''k''-means++ to create geographical cluster of photographs based on the latitude and longitude information attached to the photos. An application to financial diversification is reported by Howard and Johansen.<ref>http://www.cse.ohio-state.edu/~johansek/clustering.pdf{{dead link|date=December 2017 |bot=InternetArchiveBot |fix-attempted=yes }} Clustering Techniques for Financial Diversification, March 2009</ref> Other support for the method and ongoing discussion is also available online.<ref>{{usurped|1=[https://web.archive.org/web/20090327235840/http://lingpipe-blog.com/2009/03/23/arthur-vassilvitskii-2007-kmeans-the-advantages-of-careful-seeding/ Article title]}} Lingpipe Blog</ref> 
Since the k-means++ initialization needs k passes over the data, it does not scale very well to large data sets. Bahman Bahmani et al. have proposed a scalable variant of k-means++ called k-means|| which provides the same theoretical guarantees and yet is highly scalable.<ref name=kmeans||>B. Bahmani, B. Moseley, A. Vattani, R. Kumar, S. Vassilvitskii [http://vldb.org/pvldb/vol5/p622_bahmanbahmani_vldb2012.pdf "Scalable K-means++"] 2012 Proceedings of the VLDB Endowment.</ref>

== Software ==
* [[Accord.NET]] contains C# implementations for ''k''-means, ''k''-means++ and ''k''-modes.
* [[ALGLIB]] contains parallelized C++ and C# implementations for ''k''-means and ''k''-means++.
* [[Apache Commons|Apache Commons Math]] contains k-means++
* [[ELKI]] data-mining framework contains multiple k-means variations, including k-means++ for seeding.
* [[MATLAB]] has a K-Means implementation that uses k-means++ as default for seeding.
* [[OpenCV]] contains C++ and Python K-means implementation (with optional k-means seed initialization).
* [[Orange (software)|Orange]] includes k-means++ UI widget and API support
* [[GNU R|R]] includes k-means, and the "flexclust" package can do k-means++
* [[scikit-learn]] has a K-Means implementation that uses k-means++ by default.
* [[Weka (machine learning)|Weka]] contains k-means (with optional k-means++) and x-means clustering.

==References==
{{Reflist}}

[[Category:Cluster analysis algorithms]]