HyperAIHyperAI

Command Palette

Search for a command to run...

Pondération par ensemble indépendant à entropie maximale pour le pooling dans les réseaux de neurones sur graphes

Amirhossein Nouranizadeh Mohammadjavad Matinkia Mohammad Rahmati Reza Safabakhsh

Résumé

Dans cet article, nous proposons une nouvelle couche de pooling pour les réseaux de neurones sur graphes fondée sur la maximisation de l’information mutuelle entre le graphe agrégé et le graphe d’entrée. Étant donné que l’information mutuelle maximale est difficile à calculer, nous utilisons la capacité de Shannon d’un graphe comme biais inductif dans notre méthode de pooling. Plus précisément, nous montrons que le graphe d’entrée de la couche de pooling peut être interprété comme une représentation d’un canal de communication bruité. Pour un tel canal, l’envoi des symboles appartenant à un ensemble indépendant du graphe permet une transmission fiable et sans erreur d’information. Nous démontrons que l’atteinte de l’information mutuelle maximale équivaut à la recherche d’un ensemble indépendant de poids maximal, où les poids représentent les contenus d’entropie. À partir de cette perspective théorique de la communication, nous offrons une vision distincte du problème de pooling de graphe, en le reformulant comme la maximisation du débit de transmission d’information à travers un canal de communication bruité, mise en œuvre par un réseau de neurones sur graphe. Nous évaluons notre méthode, désignée sous le nom de Maximum Entropy Weighted Independent Set Pooling (MEWISPool), sur des tâches de classification de graphes ainsi que sur le problème d’optimisation combinatoire de l’ensemble indépendant maximal. Les résultats expérimentaux montrent que notre approche atteint des performances de pointe et compétitives sur les tâches de classification de graphes ainsi que sur le problème de l’ensemble indépendant maximal, sur plusieurs jeux de données standard.


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