Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Deep Online Probability Aggregation Clustering

About

Combining machine clustering with deep models has shown remarkable superiority in deep clustering. It modifies the data processing pipeline into two alternating phases: feature clustering and model training. However, such alternating schedule may lead to instability and computational burden issues. We propose a centerless clustering algorithm called Probability Aggregation Clustering (PAC) to proactively adapt deep learning technologies, enabling easy deployment in online deep clustering. PAC circumvents the cluster center and aligns the probability space and distribution space by formulating clustering as an optimization problem with a novel objective function. Based on the computation mechanism of the PAC, we propose a general online probability aggregation module to perform stable and flexible feature clustering over mini-batch data and further construct a deep visual clustering framework deep PAC (DPAC). Extensive experiments demonstrate that PAC has superior clustering robustness and performance and DPAC remarkably outperforms the state-of-the-art deep clustering methods.

Yuxuan Yan, Na Lu, Ruofan Yan• 2024

Related benchmarks

TaskDatasetResultRank
Image ClusteringSTL-10
ACC93.4
282
Image ClusteringImageNet-10
NMI0.925
201
ClusteringCIFAR-10 (test)
Accuracy93.4
190
ClusteringCIFAR-100 (test)
ACC55.5
123
ClusteringImagenet Dogs
NMI66.7
85
ClusteringCOIL-100
ACC65.1
28
Clusteringpendigits
Average Time153.6
14
ClusteringIsolet
ACC61.8
8
ClusteringMNIST
ACC59.7
8
ClusteringCIFAR-10 deep features (test)
Accuracy87.1
7
Showing 10 of 13 rows

Other info

Code

Follow for update