HyperAIHyperAI

Command Palette

Search for a command to run...

GMTR : Transformers pour le matching de graphes

Jinpei Guo Shaofeng Zhang Runzhong Wang Chang Liu Junchi Yan

Résumé

Les transformateurs d’images (Vision Transformers, ViTs) ont récemment été employés pour la correspondance visuelle au-delà de la détection d’objets et de la segmentation. Toutefois, la stratégie initiale de découpage en grille utilisée par les ViTs ignore l’information spatiale des points clés, ce qui limite leur sensibilité aux informations locales. Ainsi, nous proposons QueryTrans (Query Transformer), une architecture basée sur un module d’attention croisée et une stratégie de recadrage centrée sur les points clés, afin d’améliorer l’extraction d’informations spatiales. Nous intégrons par ailleurs un module d’attention sur graphe et proposons une méthode de correspondance de graphes fondée sur les transformateurs, nommée GMTR (Graph Matching TRansformers), dans laquelle la nature combinatoire du problème de correspondance de graphes (GM) est traitée par un solveur neuronal basé sur les transformateurs de graphe. Sur des benchmarks standards de correspondance de graphes, GMTR atteint des performances compétitives par rapport aux architectures de pointe (SOTA). Plus précisément, sur Pascal VOC, GMTR obtient une précision de 83,6%\mathbf{83,6\%}83,6%, soit 0,9%\mathbf{0,9\%}0,9% de plus que l’état de l’art. Sur Spair-71k, GMTR démontre un grand potentiel et surpasser la majorité des méthodes antérieures. Par ailleurs, sur Pascal VOC, QueryTrans améliore la précision de NGMv2 de 80,1%80,1\%80,1% à 83,3%\mathbf{83,3\%}83,3%, et celle de BBGM de 79,0%79,0\%79,0% à 84,5%\mathbf{84,5\%}84,5%. Sur Spair-71k, QueryTrans améliore NGMv2 de 80,6%80,6\%80,6% à 82,5%\mathbf{82,5\%}82,5%, et BBGM de 82,1%82,1\%82,1% à 83,9%\mathbf{83,9\%}83,9%. Le code source sera rendu publiquement disponible.


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