HyperAIHyperAI

Command Palette

Search for a command to run...

Réseau neuronal graphique à étiquetage des arêtes pour l'apprentissage à quelques exemples

Jongmin Kim*1,3, Taesup Kim2,3, Sungwoong Kim3, and Chang D.Yoo1

Résumé

Dans cet article, nous proposons un nouveau réseau neuronal graphique à étiquetage des arêtes (EGNN), qui adapte un réseau neuronal profond sur le graphe d'étiquetage des arêtes, pour l'apprentissage par tirs limités. Les approches précédentes de réseaux neuronaux graphiques (GNN) dans l'apprentissage par tirs limités étaient basées sur le cadre d'étiquetage des nœuds, qui modélise implicitement la similarité intra-cluster et la dissimilarité inter-cluster. En revanche, l'EGNN proposé apprend à prédire les étiquettes des arêtes plutôt que celles des nœuds sur le graphe, ce qui permet une évolution explicite du clustering en mettant à jour itérativement les étiquettes des arêtes en exploitant directement à la fois la similarité intra-cluster et la dissimilarité inter-cluster. Il est également bien adapté pour fonctionner avec divers nombres de classes sans réentraînement et peut être facilement étendu pour effectuer une inférence transductive. Les paramètres de l'EGNN sont appris par entraînement épisodique avec une perte d'étiquetage des arêtes afin d'obtenir un modèle bien généralisable pour les problèmes à faibles données non vus. Sur les tâches de classification d'images supervisée et semi-supervisée par tirs limités utilisant deux jeux de données de référence, l'EGNN proposé améliore significativement les performances par rapport aux GNN existants.


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