HyperAIHyperAI

Command Palette

Search for a command to run...

il y a 4 mois

Agrégateurs à Masques Multiples pour les Réseaux de Neurones Graphes

{Ahmet Sureyya Rifaioglu Ahmet Sarıgün}

Agrégateurs à Masques Multiples pour les Réseaux de Neurones Graphes

Résumé

L’une des opérations les plus critiques dans les réseaux de neurones sur graphes (GNN) est l’opération d’agrégation, dont l’objectif est d’extraiter des informations provenant des voisins d’un nœud cible. Plusieurs méthodes de convolution ont été proposées, telles que la convolution graphique standard (GCN), l’attention graphique (GAT) et le passage de messages (MPNN). Dans cette étude, nous proposons une méthode d’agrégation appelée Multi-Mask Aggregators (MMA), dans laquelle le modèle apprend un masque pondéré pour chaque opérateur d’agrégation avant de collecter les messages des voisins. MMA présente des similarités avec GAT et MPNN, tout en offrant des avantages théoriques et pratiques. Intuitivement, notre cadre n’est pas limité par le nombre de têtes comme dans GAT, et il offre une capacité discriminative supérieure à celle de MPNN. Les performances de MMA ont été comparées aux méthodes de référence bien établies sur des tâches de classification de nœuds et de régression sur graphes, sur des jeux de données standards largement utilisés, et les résultats montrent une amélioration significative. Les données et le code sont disponibles à l’adresse suivante : https://github.com/asarigun/mma.

Benchmarks

BenchmarkMéthodologieMétriques
graph-regression-on-zincMMA
MAE: 0.156
node-classification-on-citeseerMMA
Accuracy: 76.30%
node-classification-on-coraMMA
Accuracy: 85.80%
node-classification-on-pubmedMMA
Accuracy: 86.00%

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
Agrégateurs à Masques Multiples pour les Réseaux de Neurones Graphes | Articles de recherche | HyperAI