HyperAIHyperAI

Command Palette

Search for a command to run...

Apprentissage Contrastif Jumeau pour le Clustering en Ligne

Yunfan Li Mouxing Yang Dezhong Peng Taihao Li Jiantao Huang Xi Peng

Résumé

Ce papier propose de réaliser un clustering en ligne en effectuant un apprentissage contrastif jumeau (TCL) au niveau des instances et des clusters. Plus précisément, nous avons constaté que lorsque les données sont projetées dans un espace de caractéristiques dont la dimensionnalité correspond au nombre cible de clusters, les lignes et les colonnes de sa matrice de caractéristiques correspondent respectivement à la représentation des instances et des clusters. Sur la base de cette observation, pour un ensemble de données donné, le TCL proposé construit d'abord des paires positives et négatives grâce à des augmentations de données. Ensuite, dans l'espace des lignes et des colonnes de la matrice de caractéristiques, l'apprentissage contrastif est respectivement effectué au niveau des instances et des clusters en rapprochant les paires positives tout en éloignant les négatives. Pour atténuer l'influence des paires négatives intrinsèques et corriger les affectations de clusters, nous adoptons un critère basé sur la confiance pour sélectionner des pseudo-étiquettes afin d'améliorer à la fois l'apprentissage contrastif au niveau des instances et celui au niveau des clusters. Par conséquent, les performances du clustering sont encore améliorées. Outre l'idée élégante de l'apprentissage contrastif jumeau, un autre avantage du TCL est qu'il peut prédire indépendamment l'affectation de cluster pour chaque instance, ce qui permet une adaptation facile aux scénarios en ligne. Des expériences approfondies sur six jeux de données largement utilisés pour les images et le texte démontrent l'efficacité du TCL. Le code sera publié sur GitHub.


Créer de l'IA avec l'IA

De l'idée au lancement — accélérez votre développement IA avec le co-codage IA gratuit, un environnement prêt à l'emploi et le meilleur prix pour les GPU.

Codage assisté par IA
GPU prêts à l’emploi
Tarifs les plus avantageux

HyperAI Newsletters

Abonnez-vous à nos dernières mises à jour
Nous vous enverrons les dernières mises à jour de la semaine dans votre boîte de réception à neuf heures chaque lundi matin
Propulsé par MailChimp
Apprentissage Contrastif Jumeau pour le Clustering en Ligne | Articles | HyperAI