HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Learning deep graph matching with channel-independent embedding and Hungarian attention

{Runzhong Wang Junchi Yan Baoxin Li Tianshu Yu}

Learning deep graph matching with channel-independent embedding and Hungarian attention

Abstract

Graph matching aims to establishing node-wise correspondence between two graphs, which is a classic combinatorial problem and in general NP-complete. Until very recently, deep graph matching methods start to resort to deep networks to achieve unprecedented matching accuracy. Along this direction, this paper makes two complementary contributions which can also be reused as plugin in existing works: i) a novel node and edge embedding strategy which stimulates the multi-head strategy in attention models and allows the information in each channel to be merged independently. In contrast, only node embedding is accounted in previous works; ii) a general masking mechanism over the loss function is devised to improve the smoothness of objective learning for graph matching. Using Hungarian algorithm, it dynamically constructs a structured and sparsely connected layer, taking into account the most contributing matching pairs as hard attention. Our approach performs competitively, and can also improve state-of-the-art methods as plugin, regarding with matching accuracy on three public benchmarks.

Benchmarks

BenchmarkMethodologyMetrics
graph-matching-on-pascal-vocCIE-H
matching accuracy: 0.6756

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Learning deep graph matching with channel-independent embedding and Hungarian attention | Papers | HyperAI