arXiv reaDer
ニューラルグラフマッチングネットワーク:ハイパーグラフおよびマルチグラフマッチングへの拡張によるローラーの二次割り当て問題の学習
Neural Graph Matching Network: Learning Lawler's Quadratic Assignment Problem with Extension to Hypergraph and Multiple-graph Matching
グラフマッチングには、エッジツーエッジアフィニティマトリックスに基づく組み合わせ最適化が含まれます。これは、一般に、ローラーの2次割り当て問題(QAP)として定式化できます。この論文では、親和性行列(同等に関連グラフ)を使用して直接学習するQAPネットワークを紹介します。これにより、マッチング問題が制約付き頂点分類タスクに変換されます。アソシエーショングラフは、頂点分類のための埋め込みネットワークによって学習され、続いてシンクホーン正規化とエンドツーエンド学習のためのクロスエントロピー損失が続きます。 Sinkhornベースのマッチング対応制約と、不均等なサイズのグラフを処理するためのダミーノードを導入することにより、関連グラフへの埋め込みモデルをさらに改善します。私たちの知る限り、これは一般的なローラーのQAPで直接学習した最初のネットワークの1つです。対照的に、最近のディープマッチング手法は、それぞれ2つのグラフのノード/エッジ特徴の学習に焦点を合わせています。また、ネットワークをハイパーグラフマッチング、および複数のグラフのマッチングに拡張する方法も示します。合成グラフと実世界の画像の両方での実験結果は、その有効性を示しています。合成データとQAPLIBベンチマークでの純粋なQAPタスクの場合、私たちの方法は競争力を発揮し、最先端のグラフマッチングとQAPソルバーを大幅に少ない時間コストで超えることができます。プロジェクトのホームページはhttp://thinklab.sjtu.edu.cn/project/NGM/index.htmlで提供されています。
Graph matching involves combinatorial optimization based on edge-to-edge affinity matrix, which can be generally formulated as Lawler's Quadratic Assignment Problem (QAP). This paper presents a QAP network directly learning with the affinity matrix (equivalently the association graph) whereby the matching problem is translated into a constrained vertex classification task. The association graph is learned by an embedding network for vertex classification, followed by Sinkhorn normalization and a cross-entropy loss for end-to-end learning. We further improve the embedding model on association graph by introducing Sinkhorn based matching-aware constraint, as well as dummy nodes to deal with unequal sizes of graphs. To our best knowledge, this is one of the first network to directly learn with the general Lawler's QAP. In contrast, recent deep matching methods focus on the learning of node/edge features in two graphs respectively. We also show how to extend our network to hypergraph matching, and matching of multiple graphs. Experimental results on both synthetic graphs and real-world images show its effectiveness. For pure QAP tasks on synthetic data and QAPLIB benchmark, our method can perform competitively and even surpass state-of-the-art graph matching and QAP solvers with notable less time cost. We provide a project homepage at http://thinklab.sjtu.edu.cn/project/NGM/index.html.
updated: Thu May 06 2021 17:18:34 GMT+0000 (UTC)
published: Tue Nov 26 2019 02:06:57 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト