Saved in:
Bibliographic Details
Main Authors: Sadeghi, Mohammadreza, Armanfard, Narges
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2405.03590
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909191309885440
author Sadeghi, Mohammadreza
Armanfard, Narges
author_facet Sadeghi, Mohammadreza
Armanfard, Narges
contents Deep clustering incorporates embedding into clustering to find a lower-dimensional space appropriate for clustering. In this paper, we propose a novel deep clustering framework with self-supervision using pairwise similarities (DCSS). The proposed method consists of two successive phases. In the first phase, we propose to form hypersphere-like groups of similar data points, i.e. one hypersphere per cluster, employing an autoencoder that is trained using cluster-specific losses. The hyper-spheres are formed in the autoencoder's latent space. In the second phase, we propose to employ pairwise similarities to create a $K$-dimensional space that is capable of accommodating more complex cluster distributions, hence providing more accurate clustering performance. $K$ is the number of clusters. The autoencoder's latent space obtained in the first phase is used as the input of the second phase. The effectiveness of both phases is demonstrated on seven benchmark datasets by conducting a rigorous set of experiments.
format Preprint
id arxiv_https___arxiv_org_abs_2405_03590
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Deep Clustering with Self-Supervision using Pairwise Similarities
Sadeghi, Mohammadreza
Armanfard, Narges
Machine Learning
Deep clustering incorporates embedding into clustering to find a lower-dimensional space appropriate for clustering. In this paper, we propose a novel deep clustering framework with self-supervision using pairwise similarities (DCSS). The proposed method consists of two successive phases. In the first phase, we propose to form hypersphere-like groups of similar data points, i.e. one hypersphere per cluster, employing an autoencoder that is trained using cluster-specific losses. The hyper-spheres are formed in the autoencoder's latent space. In the second phase, we propose to employ pairwise similarities to create a $K$-dimensional space that is capable of accommodating more complex cluster distributions, hence providing more accurate clustering performance. $K$ is the number of clusters. The autoencoder's latent space obtained in the first phase is used as the input of the second phase. The effectiveness of both phases is demonstrated on seven benchmark datasets by conducting a rigorous set of experiments.
title Deep Clustering with Self-Supervision using Pairwise Similarities
topic Machine Learning
url https://arxiv.org/abs/2405.03590