Clustering (k-means ou autre) avec une contrainte de taille de cluster minimum


14

J'ai besoin de regrouper les unités en grappes pour minimiser la somme des carrés au sein du groupe (WSS), mais je dois m'assurer que les grappes contiennent chacune au moins unités. Une idée si l'une des fonctions de clustering de R permet le clustering en clusters soumis à une contrainte de taille de cluster minimum? kmeans () ne semble pas offrir une option de contrainte de taille.m kkmk

Réponses:


5

Utiliser le clustering EM

Dans le clustering EM, l'algorithme affine de manière itérative un modèle de cluster initial pour ajuster les données et détermine la probabilité qu'un point de données existe dans un cluster. L'algorithme met fin au processus lorsque le modèle probabiliste correspond aux données. La fonction utilisée pour déterminer l'ajustement est la log-vraisemblance des données compte tenu du modèle.

Si des clusters vides sont générés au cours du processus, ou si l'appartenance à un ou plusieurs des clusters tombe en dessous d'un seuil donné, les clusters avec de faibles populations sont réensemencés à de nouveaux points et l'algorithme EM est réexécuté.


Merci, Marianna. Je préférerais une solution qui s'appuie moins sur des modèles paramétriques (généralement injustifiables), mais qui va certainement y réfléchir.
Cyrus S

4

Ce problème est traité dans cet article:

Bradley, PS, KP Bennett et Ayhan Demiriz. "Cluster k-means contraint." Microsoft Research, Redmond (2000) : 1-8.

J'ai une implémentation de l'algorithme en python.


C'est parfait, merci! J'ai utilisé le rPythonpackage dans R pour créer une interface vers cette implémentation à laquelle j'ai accédé depuis mon script R.
Michael Ohlrogge

@MichaelOhlrogge avez-vous un exemple quelque part (github?) Sur l'interface que vous avez écrite pour appeler ce package python sous la forme R? Merci!
Matifou

Désolé, j'ai regardé mon ancien code mais je ne l'ai plus trouvé.
Michael Ohlrogge

3

Je pense qu'il s'agirait simplement d'exécuter les k moyennes dans le cadre d'une boucle if avec un test pour les tailles de cluster, c'est-à-dire le nombre n dans le cluster k - rappelez-vous également que k moyennes donnera des résultats différents pour chaque exécution sur les mêmes données afin vous devriez probablement l'exécuter dans le cadre d'une boucle pour extraire le "meilleur" résultat


1
Merci, Alex. Je vois cependant un problème avec cela: que se passe-t-il si sur les boucles les solutions générées ne satisfont jamais la contrainte? Cela pourrait se produire si k moyennes étaient définies pour s'exécuter sans contrainte de taille de cluster. J'aimerais une solution qui évite cela. (La nature de l'application est telle que je dois vraiment m'assurer que les clusters sont d'une taille minimale.)
Cyrus S

1

Quelle est la taille de votre ensemble de données? Vous pouvez peut-être essayer d'exécuter un clustering hiérarchique, puis décider quels clusters conserver en fonction de votre dendrogramme.

Si votre ensemble de données est énorme, vous pouvez également combiner les deux méthodes de clustering: un clustering non hiérarchique initial puis un clustering hiérarchique utilisant les groupes de l'analyse non hiérarchique. Vous pouvez trouver un exemple de cette approche dans Martínez-Pastor et al (2005)


Merci, Manuel. Cela ressemble en fait à une possibilité très intrigante. Je dois me demander si le partitionnement hiérarchique imposerait certaines contraintes qui empêcheraient l'algorithme d'atteindre le partitionnement optimal de cluster directement sous la contrainte de taille. Mais intuitivement, je peux voir que cela pourrait fonctionner.
Cyrus S

0

Cela peut être réalisé en modifiant l'étape d'affectation de cluster (E dans EM) en la formulant comme un problème d'optimisation de réseau linéaire à coût minimal (MCF).

J'ai écrit un package python qui utilise SimpleMinCostFlow des outils de recherche opérationnelle de Google, qui est une implémentation C ++ rapide. Il possède une API standard adaptée à scikit.

En utilisant notre site, vous reconnaissez avoir lu et compris notre politique liée aux cookies et notre politique de confidentialité.
Licensed under cc by-sa 3.0 with attribution required.