SOTAVerified

Unsupervised Deep Embedding for Clustering Analysis

2015-11-19Code Available1· sign in to hype

Junyuan Xie, Ross Girshick, Ali Farhadi

Code Available — Be the first to reproduce this paper.

Reproduce

Code

Abstract

Clustering is central to many data-driven application domains and has been studied extensively in terms of distance functions and grouping algorithms. Relatively little work has focused on learning representations for clustering. In this paper, we propose Deep Embedded Clustering (DEC), a method that simultaneously learns feature representations and cluster assignments using deep neural networks. DEC learns a mapping from the data space to a lower-dimensional feature space in which it iteratively optimizes a clustering objective. Our experimental evaluations on image and text corpora show significant improvement over state-of-the-art methods.

Tasks

Benchmark Results

DatasetModelMetricClaimedVerifiedStatus
CIFAR-10DECAccuracy0.3Unverified
CIFAR-100DECAccuracy0.19Unverified
CMU-PIEDEC (KL based)NMI0.92Unverified
ImageNet-10DECNMI0.28Unverified
Imagenet-dog-15DECAccuracy0.2Unverified
STL-10DECAccuracy0.36Unverified
Tiny ImageNetDECAccuracy0.04Unverified
YouTube Faces DBDEC (KL based)NMI0.45Unverified

Reproductions