HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Deep Adaptive Image Clustering

{Lingfeng Wang Jianlong Chang Shiming Xiang Gaofeng Meng Chunhong Pan}

Deep Adaptive Image Clustering

Abstract

Image clustering is a crucial but challenging task in machine learning and computer vision. Existing methods often ignore the combination between feature learning and clustering. To tackle this problem, we propose Deep Adaptive Clustering (DAC) that recasts the clustering problem into a binary pairwise-classification framework to judge whether pairs of images belong to the same clusters. In DAC, the similarities are calculated as the cosine distance between label features of images which are generated by a deep convolutional network (ConvNet). By introducing a constraint into DAC, the learned label features tend to be one-hot vectors that can be utilized for clustering images. The main challenge is that the ground-truth similarities are unknown in image clustering. We handle this issue by presenting an alternating iterative Adaptive Learning algorithm where each iteration alternately selects labeled samples and trains the ConvNet. Conclusively, images are automatically clustered based on the label features. Experimental results show that DAC achieves state-of-the-art performance on five popular datasets, e.g., yielding 97.75% clustering accuracy on MNIST, 52.18% on CIFAR-10 and 46.99% on STL-10.

Benchmarks

BenchmarkMethodologyMetrics
image-clustering-on-cifar-10DAC
ARI: 0.301
Accuracy: 0.522
Backbone: ConvNet
NMI: 0.4
Train set: Train+Test
image-clustering-on-cifar-100DAC
Accuracy: 0.238
NMI: 0.185
Train Set: Train+Test
image-clustering-on-imagenet-10DAC
Accuracy: 0.527
NMI: 0.394
image-clustering-on-imagenet-dog-15DAC
Accuracy: 0.275
NMI: 0.219
image-clustering-on-stl-10DAC
Accuracy: 0.470
Backbone: ConvNet
NMI: 0.366
Train Split: Train+Test
image-clustering-on-tiny-imagenetDAC
Accuracy: 0.066
NMI: 0.190

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Deep Adaptive Image Clustering | Papers | HyperAI