Spectral clustering

In multivariate statistics and the clustering of data, spectral clustering techniques make use of the spectrum (eigenvalues) of the similarity matrix of the data to perform dimensionality reduction before clustering in fewer dimensions. The similarity matrix is provided as an input and consists of a quantitative assessment of the relative similarity of each pair of points in the dataset.

In application to image segmentation, spectral clustering is known as segmentation-based object categorization.

Algorithms

Given an enumerated set of data points, the similarity matrix may be defined as a symmetric matrix A, where A_{ij}\geq 0 represents a measure of the similarity between data points with indexes i and j.

One spectral clustering technique is the normalized cuts algorithm or Shi–Malik algorithm introduced by Jianbo Shi and Jitendra Malik,[1] commonly used for image segmentation. It partitions points into two sets (B_1,B_2) based on the eigenvector v corresponding to the second-smallest eigenvalue of the symmetric normalized Laplacian defined as

L^{norm}:=I-D^{-1/2}AD^{-1/2},

where D is the diagonal matrix

D_{ii} = \sum_j A_{ij}.

A mathematically equivalent algorithm [2] takes the eigenvector corresponding to the largest eigenvalue of the random walk normalized Laplacian matrix P = D^{-1}A.

Another possibility is to use the Laplacian matrix defined as

L:=D-A

rather than the symmetric normalized Laplacian matrix.

Partitioning may be done in various ways, such as by computing the median m of the components of the second smallest eigenvector v, and placing all points whose component in v is greater than m in B_1, and the rest in B_2. The algorithm can be used for hierarchical clustering by repeatedly partitioning the subsets in this fashion.

Alternatively to computing just one eigenvector, k eigenvectors for some k, are computed, and then another algorithm (e.g. k-means clustering) is used to cluster points by their respective k components in these eigenvectors.

The efficiency of spectral clustering may be improved if the solution to the corresponding eigenvalue problem is performed in a matrix-free fashion, i.e., without explicitly manipulating or even computing the similarity matrix, as, e.g., in the Lanczos algorithm.

For large-sized graphs, the second eigenvalue of the (normalized) graph Laplacian matrix is often ill-conditioned, leading to slow convergence of iterative eigenvalue solvers. Preconditioning is a key technology accelerating the convergence, e.g., in the matrix-free LOBPCG method. Spectral clustering has been successfully applied on large graphs by first identifying their community structure, and then clustering communities.[3]

Spectral clustering is closely related to Nonlinear dimensionality reduction, and dimension reduction techniques such as locally-linear embedding can be used to reduce errors from noise or outliers.[4]

Relationship with k-means

The kernel k-means problem is an extension of the k-means problem where the input data points are mapped non-linearly into a higher-dimensional feature space via a kernel function k(x_i,x_j) = \phi^T(x_i)\phi(x_j). The weighted kernel k-means problem further extends this problem by defining a weight w_r for each cluster as the reciprocal of the number of elements in the cluster,


\max_{\{C_s\}} \sum_{r=1}^k w_r \sum_{x_i,x_j \in C_r} k(x_i,x_j).

Suppose F is a matrix of the normalizing coefficients for each point for each cluster F_{ij} = w_r if i,j \in C_r and zero otherwise. Suppose K is the kernel matrix for all points. The weighted kernel k-means problem with n points and k clusters is given as,


\max_{F} \operatorname{ trace } \left(KF\right)

such that,


F = G_{n\times k}G_{n\times k}^T

G^TG = I

such that \text{rank}(G) = k. In addition, there are identity constrains on F given by,


F\cdot \mathbb{I} = \mathbb{I}

where \mathbb{I} represents a vector of ones.


F^T\mathbb{I} = \mathbb{I}

This problem can be recast as,


\max_G \text{ trace }\left(G^TG\right).

This problem is equivalent to the spectral clustering problem when the identity constraints on F are relaxed. In particular, the weighted kernel k-means problem can be reformulated as a spectral clustering (graph partitioning) problem and vice versa. The output of the algorithms are eigenvectors which do not satisfy the identity requirements for indicator variables defined by F. Hence, post-processing of the eigenvectors is required for the equivalence between the problems.[5] Transforming the spectral clustering problem into a weighted kernel k-means problem greatly reduces the computational burden.[6]

See also

References

  1. Jianbo Shi and Jitendra Malik, "Normalized Cuts and Image Segmentation", IEEE Transactions on PAMI, Vol. 22, No. 8, Aug 2000.
  2. Marina Meilă & Jianbo Shi, "Learning Segmentation by Random Walks", Neural Information Processing Systems 13 (NIPS 2000), 2001, pp. 873–879.
  3. Zare, Habil; P. Shooshtari, A. Gupta and R. Brinkman (2010). "Data reduction for spectral clustering to analyze high throughput flow cytometry data". BMC Bioinformatics.
  4. Arias-Castro, E. and Chen, G. and Lerman, G. (2011), "Spectral clustering based on local linear approximations.", Electronic Journal of Statistics 5: 1537–1587, doi:10.1214/11-ejs651
  5. Dhillon, I.S. and Guan, Y. and Kulis, B. (2004). "Kernel k-means: spectral clustering and normalized cuts". Proceedings of the tenth ACM SIGKDD international conference on Knowledge discovery and data mining. pp. 551–556.
  6. Dhillon, Inderjit; Yuqiang Guan; Brian Kulis (November 2007). "Weighted Graph Cuts without Eigenvectors: A Multilevel Approach". IEEE Transactions on Pattern Analysis and Machine Intelligence 29 (11): 1–14.