Benchmarking Deep Clustering Algorithms With ClustPy

Leiber, Collin ; Miklautz, Lukas; Plant, Claudia und Böhm, Christian (2023): Benchmarking Deep Clustering Algorithms With ClustPy. 2023 IEEE International Conference on Data Mining Workshops (ICDMW), Shanghai, China, 01-04 December 2023. Wang, Jihe; He, Yi; Dinh, Thang N.; Gran, Christan; Qiu, Meikang und Pedrycz, Witold (Hrsg.): In: 2023 IEEE International Conference on Data Mining Workshops (ICDMW), Piscataway, NJ: IEEE. S. 625-632

Abstract

Deep clustering algorithms have gained popularity as they are able to cluster complex large-scale data, like images. Yet these powerful algorithms require many decisions w.r.t. architecture, learning rate and other hyperparameters, making it difficult to compare different methods. A comprehensive empirical evaluation of novel clustering methods, however, plays an important role in both scientific and practical applications, as it reveals their individual strengths and weaknesses. Therefore, we introduce ClustPy, a unified framework for benchmarking deep clustering algorithms, and perform a comparison of several fundamental deep clustering methods and some recently introduced ones. We compare these methods on multiple well known image data sets using different evaluation metrics, perform a sensitivity analysis w.r.t. important hyperparameters and perform ablation studies, e.g., for different autoencoder architectures and image augmentation. To our knowledge this is the first in depth benchmarking of deep clustering algorithms in a unified setting.

• Dokumententyp: Konferenzbeitrag (Paper)
• Fakultät: Mathematik, Informatik und Statistik > Informatik
• Themengebiete: 000 Informatik, Informationswissenschaft, allgemeine Werke > 004 Informatik
• ISBN: 979-8-3503-8164-1 ; 979-8-3503-8165-8
• Ort: Piscataway, NJ
• Sprache: Englisch