Scaling Up Deep Clustering Methods Beyond ImageNet-1K

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Adaloglou, Nikolas, Michels, Felix, Senft, Kaspar, Petrusheva, Diana, Kollmann, Markus
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929371157102592
author Adaloglou, Nikolas
Michels, Felix
Senft, Kaspar
Petrusheva, Diana
Kollmann, Markus
author_facet Adaloglou, Nikolas
Michels, Felix
Senft, Kaspar
Petrusheva, Diana
Kollmann, Markus
contents Deep image clustering methods are typically evaluated on small-scale balanced classification datasets while feature-based $k$-means has been applied on proprietary billion-scale datasets. In this work, we explore the performance of feature-based deep clustering approaches on large-scale benchmarks whilst disentangling the impact of the following data-related factors: i) class imbalance, ii) class granularity, iii) easy-to-recognize classes, and iv) the ability to capture multiple classes. Consequently, we develop multiple new benchmarks based on ImageNet21K. Our experimental analysis reveals that feature-based $k$-means is often unfairly evaluated on balanced datasets. However, deep clustering methods outperform $k$-means across most large-scale benchmarks. Interestingly, $k$-means underperforms on easy-to-classify benchmarks by large margins. The performance gap, however, diminishes on the highest data regimes such as ImageNet21K. Finally, we find that non-primary cluster predictions capture meaningful classes (i.e. coarser classes).
format Preprint
id arxiv_https___arxiv_org_abs_2406_01203
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Scaling Up Deep Clustering Methods Beyond ImageNet-1K
Adaloglou, Nikolas
Michels, Felix
Senft, Kaspar
Petrusheva, Diana
Kollmann, Markus
Computer Vision and Pattern Recognition
Artificial Intelligence
Machine Learning
Deep image clustering methods are typically evaluated on small-scale balanced classification datasets while feature-based $k$-means has been applied on proprietary billion-scale datasets. In this work, we explore the performance of feature-based deep clustering approaches on large-scale benchmarks whilst disentangling the impact of the following data-related factors: i) class imbalance, ii) class granularity, iii) easy-to-recognize classes, and iv) the ability to capture multiple classes. Consequently, we develop multiple new benchmarks based on ImageNet21K. Our experimental analysis reveals that feature-based $k$-means is often unfairly evaluated on balanced datasets. However, deep clustering methods outperform $k$-means across most large-scale benchmarks. Interestingly, $k$-means underperforms on easy-to-classify benchmarks by large margins. The performance gap, however, diminishes on the highest data regimes such as ImageNet21K. Finally, we find that non-primary cluster predictions capture meaningful classes (i.e. coarser classes).
title Scaling Up Deep Clustering Methods Beyond ImageNet-1K
topic Computer Vision and Pattern Recognition
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2406.01203