A penalized criterion for selecting the number of clusters for K-medians

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Godichon-Baggioni, Antoine, Surendran, Sobihan
Format: Preprint
Veröffentlicht: 2022
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866911784308310016
author Godichon-Baggioni, Antoine
Surendran, Sobihan
author_facet Godichon-Baggioni, Antoine
Surendran, Sobihan
contents Clustering is a usual unsupervised machine learning technique for grouping the data points into groups based upon similar features. We focus here on unsupervised clustering for contaminated data, i.e in the case where K-medians should be preferred to K-means because of its robustness. More precisely, we concentrate on a common question in clustering: how to chose the number of clusters? The answer proposed here is to consider the choice of the optimal number of clusters as the minimization of a risk function via penalization. In this paper, we obtain a suitable penalty shape for our criterion and derive an associated oracle-type inequality. Finally, the performance of this approach with different types of K-medians algorithms is compared on a simulation study with other popular techniques. All studied algorithms are available in the R package Kmedians on CRAN.
format Preprint
id arxiv_https___arxiv_org_abs_2209_03597
institution arXiv
publishDate 2022
record_format arxiv
spellingShingle A penalized criterion for selecting the number of clusters for K-medians
Godichon-Baggioni, Antoine
Surendran, Sobihan
Statistics Theory
Clustering is a usual unsupervised machine learning technique for grouping the data points into groups based upon similar features. We focus here on unsupervised clustering for contaminated data, i.e in the case where K-medians should be preferred to K-means because of its robustness. More precisely, we concentrate on a common question in clustering: how to chose the number of clusters? The answer proposed here is to consider the choice of the optimal number of clusters as the minimization of a risk function via penalization. In this paper, we obtain a suitable penalty shape for our criterion and derive an associated oracle-type inequality. Finally, the performance of this approach with different types of K-medians algorithms is compared on a simulation study with other popular techniques. All studied algorithms are available in the R package Kmedians on CRAN.
title A penalized criterion for selecting the number of clusters for K-medians
topic Statistics Theory
url https://arxiv.org/abs/2209.03597