arXiv · 2206.05093
Federated Momentum Contrastive Clustering
Abstract
We present federated momentum contrastive clustering (FedMCC), a learning framework that can not only extract discriminative representations over distributed local data but also perform data clustering. In FedMCC, a transformed data pair passes through both the online and target networks, resulting in four representations over which the losses are determined. The resulting high-quality representations generated by FedMCC can outperform several existing self-supervised learning methods for linear evaluation and semi-supervised learning tasks. FedMCC can easily be adapted to ordinary centralized clustering through what we call momentum contrastive clustering (MCC). We show that MCC achieves state-of-the-art clustering accuracy results in certain datasets such as STL-10 and ImageNet-10. We also present a method to reduce the memory footprint of our clustering schemes.
Explore related subjects
Keep this discovery
Runxuan Miao, Erdem Koyuncu. 2022-06-10. Federated Momentum Contrastive Clustering. https://arxiv.org/abs/2206.05093
Cite the original work for its findings. Save a collection to share your selection of sources.