asa058, a FORTRAN90 code which handles the K-Means problem, by David Sparks.
The code is Applied Statistics Algorithm 58.
In the K-Means problem, a set of N points X(I) in M-dimensions is given. The goal is to arrange these points into K clusters, with each cluster having a representative point Z(J), usually chosen as the centroid of the points in the cluster. The energy of each cluster is
E(J) = Sum ( all points X(I) in cluster J ) || X(I) - Z(J) ||^2
For a given set of clusters, the total energy is then simply the sum of the cluster energies E(J). The goal is to choose the clusters in such a way that the total energy is minimized. Usually, a point X(I) goes into the cluster with the closest representative point Z(J). So to define the clusters, it's enough simply to specify the locations of the cluster representatives.
This is actually a fairly hard problem. Most algorithms do reasonably well, but cannot guarantee that the best solution has been found. It is very common for algorithms to get stuck at a solution which is merely a "local minimum". For such a local minimum, every slight rearrangement of the solution makes the energy go up; however a major rearrangement would result in a big drop in energy.
A simple algorithm for the problem is known as "H-Means". It alternates between two procedures:
A more sophisticated algorithm, known as "K-Means", takes advantage of the fact that it is possible to quickly determine the decrease in energy caused by moving a point from its current cluster to another. It repeats the following procedure:
Note: the original reference lists the input variable F as an integer workspace array. However, F is used in the CLUSTR routine exclusively as a real array. Even in single precision, this causes the routine to compute incorrect results (try it, please!); in double precision it also causes memory overwrites. The code presented here has corrected this mistake.
The computer code and data files described and made available on this web page are distributed under the MIT license
asa058 is available in a C version and a C++ version and a FORTRAN90 version and a MATLAB version.
ASA113, a FORTRAN90 code which implements the Banfield and Bassill clustering algorithm using transfers and swaps.
ASA136, a FORTRAN90 code which implements the Hartigan and Wong clustering algorithm.
CITIES, a FORTRAN90 code which contains various problems associated with a set of "cities" on a map.
CITIES, a dataset directory which contains sets of data defining groups of cities.
KMEANS, a FORTRAN90 code which contains several different algorithms for the K-Means problem.
LAU_NP, a FORTRAN90 code which contains heuristic algorithms for the K-center and K-median problems.
SPAETH, a FORTRAN90 code which clusters data according to various principles.
SPAETH, a dataset directory which contains sets of test data for clustering.
SPAETH2, a FORTRAN90 code which clusters data according to various principles.
SPAETH2, a dataset collection which contains sets of test data for clustering.
Original FORTRAN77 version by David Sparks; FORTRAN90 version by John Burkardt.