Saved in:
Bibliographic Details
Main Authors: Song, Xianfeng, Zou, Yi, Shi, Zheng, Liu, Zheng
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2412.18221
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915467068702720
author Song, Xianfeng
Zou, Yi
Shi, Zheng
Liu, Zheng
author_facet Song, Xianfeng
Zou, Yi
Shi, Zheng
Liu, Zheng
contents Feature-based image matching has extensive applications in computer vision. Keypoints detected in images can be naturally represented as graph structures, and Graph Neural Networks (GNNs) have been shown to outperform traditional deep learning techniques. Consequently, the paradigm of image matching via GNNs has gained significant prominence in recent academic research. In this paper, we first introduce an innovative adaptive graph construction method that utilizes a filtering mechanism based on distance and dynamic threshold similarity. This method dynamically adjusts the criteria for incorporating new vertices based on the characteristics of existing vertices, allowing for the construction of more precise and robust graph structures while avoiding redundancy. We further combine the vertex processing capabilities of GNNs with the global awareness capabilities of Transformers to enhance the model's representation of spatial and feature information within graph structures. This hybrid model provides a deeper understanding of the interrelationships between vertices and their contributions to the matching process. Additionally, we employ the Sinkhorn algorithm to iteratively solve for optimal matching results. Finally, we validate our system using extensive image datasets and conduct comprehensive comparative experiments. Experimental results demonstrate that our system achieves an average improvement of 3.8x-40.3x in overall matching performance. Additionally, the number of vertices and edges significantly impacts training efficiency and memory usage; therefore, we employ multi-GPU technology to accelerate the training process. Our code is available at https://github.com/songxf1024/GIMS.
format Preprint
id arxiv_https___arxiv_org_abs_2412_18221
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle GIMS: Image Matching System Based on Adaptive Graph Construction and Graph Neural Network
Song, Xianfeng
Zou, Yi
Shi, Zheng
Liu, Zheng
Computer Vision and Pattern Recognition
Machine Learning
Feature-based image matching has extensive applications in computer vision. Keypoints detected in images can be naturally represented as graph structures, and Graph Neural Networks (GNNs) have been shown to outperform traditional deep learning techniques. Consequently, the paradigm of image matching via GNNs has gained significant prominence in recent academic research. In this paper, we first introduce an innovative adaptive graph construction method that utilizes a filtering mechanism based on distance and dynamic threshold similarity. This method dynamically adjusts the criteria for incorporating new vertices based on the characteristics of existing vertices, allowing for the construction of more precise and robust graph structures while avoiding redundancy. We further combine the vertex processing capabilities of GNNs with the global awareness capabilities of Transformers to enhance the model's representation of spatial and feature information within graph structures. This hybrid model provides a deeper understanding of the interrelationships between vertices and their contributions to the matching process. Additionally, we employ the Sinkhorn algorithm to iteratively solve for optimal matching results. Finally, we validate our system using extensive image datasets and conduct comprehensive comparative experiments. Experimental results demonstrate that our system achieves an average improvement of 3.8x-40.3x in overall matching performance. Additionally, the number of vertices and edges significantly impacts training efficiency and memory usage; therefore, we employ multi-GPU technology to accelerate the training process. Our code is available at https://github.com/songxf1024/GIMS.
title GIMS: Image Matching System Based on Adaptive Graph Construction and Graph Neural Network
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2412.18221