HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Deep Online Probability Aggregation Clustering

Yan Yuxuan ; Lu Na ; Yan Ruofan

Deep Online Probability Aggregation Clustering

Abstract

Combining machine clustering with deep models has shown remarkablesuperiority in deep clustering. It modifies the data processing pipeline intotwo alternating phases: feature clustering and model training. However, suchalternating schedule may lead to instability and computational burden issues.We propose a centerless clustering algorithm called Probability AggregationClustering (PAC) to proactively adapt deep learning technologies, enabling easydeployment in online deep clustering. PAC circumvents the cluster center andaligns the probability space and distribution space by formulating clusteringas an optimization problem with a novel objective function. Based on thecomputation mechanism of the PAC, we propose a general online probabilityaggregation module to perform stable and flexible feature clustering overmini-batch data and further construct a deep visual clustering framework deepPAC (DPAC). Extensive experiments demonstrate that PAC has superior clusteringrobustness and performance and DPAC remarkably outperforms the state-of-the-artdeep clustering methods.

Benchmarks

BenchmarkMethodologyMetrics
image-clustering-on-cifar-10DPAC
ARI: 0.866
Accuracy: 0.934
Backbone: ResNet-34
NMI: 0.87
image-clustering-on-cifar-100DPAC
ARI: 0.393
Accuracy: 0.555
Backbone: ResNet-34
NMI: 0.542
image-clustering-on-imagenet-10DPAC
ARI: 0.935
Accuracy: 0.97
Backbone: ResNet-34
NMI: 0.925
image-clustering-on-imagenet-dog-15DPAC
ARI: 0.598
Accuracy: 0.726
Backbone: ResNet-34
NMI: 0.667
image-clustering-on-stl-10DPAC
ARI: 0.861
Accuracy: 0.934
Backbone: ResNet-34
NMI: 0.863

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