Abstract:Clustering is a long-standing problem area in data mining. The centroid-based classical approaches to clustering mainly face difficulty in the case of high dimensional inputs such as images. With the advent of deep neural networks, a common approach to this problem is to map the data to some latent space of comparatively lower dimensions and then do the clustering in that space. Network architectures adopted for this are generally autoencoders that reconstruct a given input in the output. To keep the input in some compact form, the encoder in AE's learns to extract useful features that get decoded at the reconstruction end. A well-known centroid-based clustering algorithm is K-means. In the context of deep feature learning, recent works have empirically shown the importance of learning the representations and the cluster centroids together. However, in this aspect of joint learning, recently a continuous variant of K-means has been proposed; where the softmax function is used in place of argmax to learn the clustering and network parameters jointly using stochastic gradient descent (SGD). However, unlike K-means, where the input space stays constant, here the learning of the centroid is done in parallel to the learning of the latent space for every batch of data. Such batch updates disagree with the concept of classical K-means, where the clustering space remains constant as it is the input space itself. To this end, we propose to alternatively learn a clustering-friendly data representation and K-means based cluster centers. Experiments on some benchmark datasets have shown improvements of our approach over the previous approaches.

Transformed K-means Clustering

Combining multiple clusterings via k-modes algorithm

Contrastive Deep Convolutional Transform K-Means Clustering

Deep Transformation-Invariant Clustering

An Approach Towards Learning K-means-friendly Deep Latent Representation

Kernel K-means Based Framework for Aggregate Outputs Classification

Clustering transformed compositional data using K-means, with applications in gene expression and bicycle sharing system data

Sub-One Quasi-Norm-Based k-Means Clustering Algorithm and Analyses

Rethinking k-means from manifold learning perspective

Sparse subspace clustering incorporated deep convolutional transform learning for hyperspectral band selection

Clustering Via Multiple Kernel K-Means Coupled Graph and Enhanced Tensor Learning

Integrating Curriculum Learning With k-Means: A Data-Centric Approach to Faster Clustering

Clustering Large Datasets by Merging K-Means Solutions

Merging $K$-means with hierarchical clustering for identifying general-shaped groups

Learning the Shared Subspace for Multi-task Clustering and Transductive Transfer Classification.

A framework for simultaneous co-clustering and learning from complex data

Multiple Kernel Clustering with Kernel k-Means Coupled Graph Tensor Learning

Robust and Automatic Data Clustering: Dirichlet Process meets Median-of-Means

Hybridization of Expectation-Maximization and K-Means Algorithms for Better Clustering Performance

Federated K-means Clustering

K-Means Clustering with Bagging and MapReduce