arXiv · 2601.00963
Deep Clustering with Associative Memories
Abstract
Deep clustering - joint representation learning and latent space clustering - is a well studied problem especially in computer vision and text processing under the deep learning framework. While the representation learning is generally differentiable, clustering is an inherently discrete optimization task, requiring various approximations and regularizations to fit in a standard differentiable pipeline. This leads to a somewhat disjointed representation learning and clustering. In this work, we propose a novel loss function utilizing energy-based dynamics via Associative Memories to formulate a new deep clustering method, DCAM, which ties together the representation learning and clustering aspects more intricately in a single objective. Our experiments showcase the advantage of DCAM, producing improved clustering quality for various architecture choices (convolutional, residual or fully-connected) and data modalities (images or text).
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Bishwajit Saha, Dmitry Krotov, Mohammed J. Zaki, Parikshit Ram. 2026-01-02. Deep Clustering with Associative Memories. https://arxiv.org/abs/2601.00963
Cite the original work for its findings. Save a collection to share your selection of sources.