HyperAIHyperAI

Command Palette

Search for a command to run...

il y a 4 mois

Apprentissage du correspondance de graphes profonds avec une embedding indépendante des canaux et une attention de Hungarian

{Runzhong Wang Junchi Yan Baoxin Li Tianshu Yu}

Apprentissage du correspondance de graphes profonds avec une embedding indépendante des canaux et une attention de Hungarian

Résumé

L’appariement de graphes vise à établir une correspondance entre les nœuds de deux graphes, un problème classique de combinatoire qui est en général NP-complet. Jusqu’à récemment, les méthodes d’appariement de graphes basées sur les réseaux profonds ont commencé à recourir à des réseaux neuronaux profonds afin d’atteindre une précision d’appariement sans précédent. Dans cette direction, ce papier apporte deux contributions complémentaires, pouvant également être réutilisées comme modules dans des travaux existants : i) une nouvelle stratégie d’embedding pour les nœuds et les arêtes, inspirée de la stratégie à multi-têtes des modèles d’attention, permettant la fusion indépendante des informations dans chaque canal ; en comparaison, seuls les embeddings de nœuds étaient pris en compte dans les travaux antérieurs ; ii) un mécanisme général de masquage appliqué à la fonction de perte, conçu pour améliorer la régularité de l’apprentissage de l’objectif dans l’appariement de graphes. En utilisant l’algorithme hongrois, notre méthode construit dynamiquement une couche structurée et fortement creuse, en tenant compte des paires d’appariement les plus pertinentes comme une attention rigide. Notre approche obtient des performances compétitives et peut également améliorer les méthodes de pointe actuelles lorsqu’elle est intégrée comme module, en termes de précision d’appariement sur trois benchmarks publics.

Benchmarks

BenchmarkMéthodologieMétriques
graph-matching-on-pascal-vocCIE-H
matching accuracy: 0.6756

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
Apprentissage du correspondance de graphes profonds avec une embedding indépendante des canaux et une attention de Hungarian | Articles de recherche | HyperAI