Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

A Simple Approach to Automated Spectral Clustering

About

The performance of spectral clustering heavily relies on the quality of affinity matrix. A variety of affinity-matrix-construction (AMC) methods have been proposed but they have hyperparameters to determine beforehand, which requires strong experience and leads to difficulty in real applications, especially when the inter-cluster similarity is high and/or the dataset is large. In addition, we often need to choose different AMC methods for different datasets, which still depends on experience. To solve these two challenging problems, in this paper, we present a simple yet effective method for automated spectral clustering. First, we propose to find the most reliable affinity matrix via grid search or Bayesian optimization among a set of candidates given by different AMC methods with different hyperparameters, where the reliability is quantified by the \textit{relative-eigen-gap} of graph Laplacian introduced in this paper. Second, we propose a fast and accurate AMC method based on least squares representation and thresholding and prove its effectiveness theoretically. Finally, we provide a large-scale extension for the automated spectral clustering method, of which the time complexity is linear with the number of data points. Extensive experiments of natural image clustering show that our method is more versatile, accurate, and efficient than baseline methods.

Jicong Fan, Yiheng Tu, Zhao Zhang, Mingbo Zhao, Haijun Zhang• 2021

Related benchmarks

TaskDatasetResultRank
ClusteringMNIST (full)
Accuracy75.5
98
ClusteringMNIST--
92
ClusteringCOIL20
AC87.8
28
Subspace ClusteringYale-B
ACC90.9
21
Subspace ClusteringORL
NMI90.7
19
ClusteringORL--
11
ClusteringAR
Accuracy82.6
10
ClusteringMNIST 1K
Accuracy61.9
10
ClusteringFashion-MNIST 1k
ACC58.4
10
Subspace ClusteringAR
NMI90.4
9
Showing 10 of 18 rows

Other info

Code

Follow for update