{"ID":2897832,"CreatedAt":"2026-06-01T04:54:23.091178241Z","UpdatedAt":"2026-06-01T04:54:23.091178241Z","DeletedAt":null,"paper_url":"https://arxiv.org/abs/2507.04283","arxiv_id":"2507.04283","title":"Clustering via Self-Supervised Diffusion","abstract":"Diffusion models, widely recognized for their success in generative tasks, have not yet been applied to clustering. We introduce Clustering via Diffusion (CLUDI), a self-supervised framework that combines the generative power of diffusion models with pre-trained Vision Transformer features to achieve robust and accurate clustering. CLUDI is trained via a teacher-student paradigm: the teacher uses stochastic diffusion-based sampling to produce diverse cluster assignments, which the student refines into stable predictions. This stochasticity acts as a novel data augmentation strategy, enabling CLUDI to uncover intricate structures in high-dimensional data. Extensive evaluations on challenging datasets demonstrate that CLUDI achieves state-of-the-art performance in unsupervised classification, setting new benchmarks in clustering robustness and adaptability to complex data distributions. Our code is available at https://github.com/BGU-CS-VIL/CLUDI.","short_abstract":"Diffusion models, widely recognized for their success in generative tasks, have not yet been applied to clustering. We introduce Clustering via Diffusion (CLUDI), a self-supervised framework that combines the generative power of diffusion models with pre-trained Vision Transformer features to achieve robust and accurat...","url_abs":"https://arxiv.org/abs/2507.04283","url_pdf":"https://arxiv.org/pdf/2507.04283v2","authors":"[\"Roy Uziel\",\"Irit Chelly\",\"Oren Freifeld\",\"Ari Pakman\"]","published":"2025-07-06T07:57:08Z","proceeding":"cs.AI","tasks":"[\"cs.AI\",\"cs.CV\"]","methods":"[\"Vision Transformer\",\"Diffusion Model\",\"Transformer\"]","has_code":false,"code_links":[{"ID":612371,"CreatedAt":"2026-06-01T04:54:23.091178241Z","UpdatedAt":"2026-06-01T04:54:23.091178241Z","DeletedAt":null,"paper_id":2897832,"paper_url":"https://arxiv.org/abs/2507.04283","paper_title":"Clustering via Self-Supervised Diffusion","repo_url":"https://github.com/BGU-CS-VIL/CLUDI","is_official":false,"mentioned_in_paper":false,"mentioned_in_github":true,"github_stars":0}]}