Command Palette
Search for a command to run...
Yan Yuxuan ; Lu Na ; Yan Ruofan

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.
Code Repositories
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| image-clustering-on-cifar-10 | DPAC | ARI: 0.866 Accuracy: 0.934 Backbone: ResNet-34 NMI: 0.87 |
| image-clustering-on-cifar-100 | DPAC | ARI: 0.393 Accuracy: 0.555 Backbone: ResNet-34 NMI: 0.542 |
| image-clustering-on-imagenet-10 | DPAC | ARI: 0.935 Accuracy: 0.97 Backbone: ResNet-34 NMI: 0.925 |
| image-clustering-on-imagenet-dog-15 | DPAC | ARI: 0.598 Accuracy: 0.726 Backbone: ResNet-34 NMI: 0.667 |
| image-clustering-on-stl-10 | DPAC | 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.