HyperAIHyperAI

Command Palette

Search for a command to run...

il y a 4 mois

Réseau de neurones graphiques à capsules

{Zhang Xinyi Lihui Chen}

Réseau de neurones graphiques à capsules

Résumé

Les embeddings de nœuds de haute qualité appris à partir des réseaux de neurones graphiques (GNN) ont été largement appliqués à diverses tâches centrées sur les nœuds, certaines atteignant même des performances de pointe (SOTA). Toutefois, lorsqu’on utilise ces embeddings de nœuds issus des GNN pour générer des embeddings de graphe, la représentation scalaire des nœuds s’avère souvent insuffisante pour préserver efficacement les propriétés des nœuds et des graphes, conduisant à des embeddings de graphe sous-optimisés.Inspirés par le réseau de neurones à capsules (CapsNet), nous proposons le Capsule Graph Neural Network (CapsGNN), qui intègre le concept de capsules afin de surmonter les limites des méthodes existantes d’embeddings de graphe basées sur les GNN. En extrayant les caractéristiques des nœuds sous forme de capsules, le mécanisme de routage permet de capturer efficacement des informations importantes au niveau du graphe. En conséquence, notre modèle génère plusieurs embeddings par graphe, permettant ainsi de saisir différentes facettes des propriétés du graphe. Le module d’attention intégré dans CapsGNN permet également de traiter des graphes de tailles variées, tout en orientant l’attention du modèle vers les parties critiques du graphe.Nos évaluations étendues sur 10 jeux de données structurées en graphe démontrent que CapsGNN dispose d’un mécanisme puissant, fondé sur une approche data-driven, pour capturer les propriétés macroscopiques globales du graphe. Il surpasser d’autres techniques de pointe sur plusieurs tâches de classification de graphes, grâce à cet outil innovant.

Benchmarks

BenchmarkMéthodologieMétriques
graph-classification-on-collabCapsGNN
Accuracy: 79.62%
graph-classification-on-ddCapsGNN
Accuracy: 75.38%
graph-classification-on-enzymesCapsGNN
Accuracy: 54.67%
graph-classification-on-imdb-bCapsGNN
Accuracy: 73.10%
graph-classification-on-imdb-mCapsGNN
Accuracy: 50.27%
graph-classification-on-mutagCapsGNN
Accuracy: 86.67%
graph-classification-on-nci1CapsGNN
Accuracy: 78.35%
graph-classification-on-proteinsCapsGNN
Accuracy: 76.28%
graph-classification-on-re-m12kCapsGNN
Accuracy: 46.62%
graph-classification-on-re-m5kCapsGNN
Accuracy: 52.88%

Construire l'IA avec l'IA

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

Co-codage IA
GPU prêts à utiliser
Meilleurs prix
Commencer

Hyper 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
Réseau de neurones graphiques à capsules | Articles de recherche | HyperAI