Saved in:
Bibliographic Details
Main Authors: Tan, Haoru, Wang, Chuang, Wu, Sitong, Zhang, Xu-Yao, Yin, Fei, Liu, Cheng-Lin
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2403.06457
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914709351956480
author Tan, Haoru
Wang, Chuang
Wu, Sitong
Zhang, Xu-Yao
Yin, Fei
Liu, Cheng-Lin
author_facet Tan, Haoru
Wang, Chuang
Wu, Sitong
Zhang, Xu-Yao
Yin, Fei
Liu, Cheng-Lin
contents Graph matching is a commonly used technique in computer vision and pattern recognition. Recent data-driven approaches have improved the graph matching accuracy remarkably, whereas some traditional algorithm-based methods are more robust to feature noises, outlier nodes, and global transformation (e.g.~rotation). In this paper, we propose a graph neural network (GNN) based approach to combine the advantages of data-driven and traditional methods. In the GNN framework, we transform traditional graph-matching solvers as single-channel GNNs on the association graph and extend the single-channel architecture to the multi-channel network. The proposed model can be seen as an ensemble method that fuses multiple algorithms at every iteration. Instead of averaging the estimates at the end of the ensemble, in our approach, the independent iterations of the ensembled algorithms exchange their information after each iteration via a 1x1 channel-wise convolution layer. Experiments show that our model improves the performance of traditional algorithms significantly. In addition, we propose a random sampling strategy to reduce the computational complexity and GPU memory usage, so the model applies to matching graphs with thousands of nodes. We evaluate the performance of our method on three tasks: geometric graph matching, semantic feature matching, and few-shot 3D shape classification. The proposed model performs comparably or outperforms the best existing GNN-based methods.
format Preprint
id arxiv_https___arxiv_org_abs_2403_06457
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Ensemble Quadratic Assignment Network for Graph Matching
Tan, Haoru
Wang, Chuang
Wu, Sitong
Zhang, Xu-Yao
Yin, Fei
Liu, Cheng-Lin
Computer Vision and Pattern Recognition
Graph matching is a commonly used technique in computer vision and pattern recognition. Recent data-driven approaches have improved the graph matching accuracy remarkably, whereas some traditional algorithm-based methods are more robust to feature noises, outlier nodes, and global transformation (e.g.~rotation). In this paper, we propose a graph neural network (GNN) based approach to combine the advantages of data-driven and traditional methods. In the GNN framework, we transform traditional graph-matching solvers as single-channel GNNs on the association graph and extend the single-channel architecture to the multi-channel network. The proposed model can be seen as an ensemble method that fuses multiple algorithms at every iteration. Instead of averaging the estimates at the end of the ensemble, in our approach, the independent iterations of the ensembled algorithms exchange their information after each iteration via a 1x1 channel-wise convolution layer. Experiments show that our model improves the performance of traditional algorithms significantly. In addition, we propose a random sampling strategy to reduce the computational complexity and GPU memory usage, so the model applies to matching graphs with thousands of nodes. We evaluate the performance of our method on three tasks: geometric graph matching, semantic feature matching, and few-shot 3D shape classification. The proposed model performs comparably or outperforms the best existing GNN-based methods.
title Ensemble Quadratic Assignment Network for Graph Matching
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2403.06457