HyperAIHyperAI

Command Palette

Search for a command to run...

il y a 4 mois

Apprentissage profond du correspondance partielle de graphes par Top-K différentiable

{Junchi Yan Xiaokang Yang Shaofei Jiang Ziao Guo Runzhong Wang}

Apprentissage profond du correspondance partielle de graphes par Top-K différentiable

Résumé

L'appariement de graphes (Graph Matching, GM) vise à découvrir des correspondances entre les nœuds de graphes en maximisant les affinités aux niveaux des nœuds et des arêtes entre les éléments appariés. Étant un problème NP-dur, sa difficulté est encore accrue par la présence de nœuds aberrants (outliers) présents dans les deux graphes, phénomène fréquent dans les applications pratiques, notamment dans les problèmes de vision par ordinateur. Toutefois, les paradigmes courants fondés sur la maximisation des affinités manquent souvent d’un cadre théoriquement fondé pour supprimer les appariements erronés et recourent à des seuils artificiels pour éliminer les outliers. Cette limitation est également héritée par les solveurs neuronaux de GM, malgré leurs performances supérieures dans le cadre idéal sans outliers. Dans cet article, nous proposons de formuler le problème de GM partiel comme une tâche de sélection des meilleurs k éléments, avec un nombre k donné ou estimé de points d’alignement valides (inliers). Plus précisément, nous introduisons un module différentiable de sélection des k meilleurs éléments, permettant une descente de gradient efficace au sein d'une couche de transport optimal, et pouvant être facilement intégré aux pipelines de GM profondes de pointe, tels que le réseau de correspondance quadratique NGMv2 ou le réseau linéaire GCAN. Par ailleurs, nous avons conçu des couches d’agrégation fondées sur l’attention afin d’estimer automatiquement k, permettant ainsi un appariement robuste aux outliers dans des conditions réelles. Enfin, nous avons révisé et publié une nouvelle base de données appelée IMC-PT-SparseGM, issue de l’ensemble de données IMC-PT pour la correspondance stéréo. Cette nouvelle base inclut un plus grand nombre de graphes à échelle variable ainsi que des instances d’appariement partiel issues du monde réel. Les expériences montrent que notre méthode surpasser les autres approches d’appariement partiel sur des benchmarks courants.

Benchmarks

BenchmarkMéthodologieMétriques
graph-matching-on-imcpt-sparsegm-100NGMv2
F1 score: 0.676
graph-matching-on-imcpt-sparsegm-100PCA-GM
F1 score: 0.575
graph-matching-on-imcpt-sparsegm-100NGMv2-AFAT-I
F1 score: 0.701
graph-matching-on-imcpt-sparsegm-100GCAN-AFAT-I
F1 score: 0.709
graph-matching-on-imcpt-sparsegm-100NGMv2-AFAT-U
F1 score: 0.703
graph-matching-on-imcpt-sparsegm-100GCAN-AFAT-U
F1 score: 0.715
graph-matching-on-imcpt-sparsegm-50NGMv2-AFAT-U
F1 score: 0.720
graph-matching-on-imcpt-sparsegm-50PCA-GM
F1 score: 0.631
graph-matching-on-imcpt-sparsegm-50GCAN-AFAT-U
F1 score: 0.711
graph-matching-on-imcpt-sparsegm-50NGMv2
F1 score: 0.703
graph-matching-on-imcpt-sparsegm-50GCAN-AFAT-I
F1 score: 0.729
graph-matching-on-imcpt-sparsegm-50NGMv2-AFAT-I
F1 score: 0.728
graph-matching-on-pascal-vocNGMv2-AFAT-U
F1 score: 0.602
graph-matching-on-pascal-vocGCAN-AFAT-I
F1 score: 0.616
graph-matching-on-pascal-vocGCAN-AFAT-U
F1 score: 0.620
graph-matching-on-pascal-vocNGMv2-AFAT-I
F1 score: 0.599
graph-matching-on-willow-object-classNGMv2-AFAT-I
F1 score: 0.831
graph-matching-on-willow-object-classNGMv2-AFAT-U
F1 score: 0.817
graph-matching-on-willow-object-classGCAN-AFAT-I
F1 score: 0.837
graph-matching-on-willow-object-classGCAN-AFAT-U
F1 score: 0.823

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 profond du correspondance partielle de graphes par Top-K différentiable | Articles de recherche | HyperAI