Learnable Graph Matching: A Practical Paradigm for Data Association
- URL: http://arxiv.org/abs/2303.15414v2
- Date: Tue, 6 Feb 2024 10:24:49 GMT
- Title: Learnable Graph Matching: A Practical Paradigm for Data Association
- Authors: Jiawei He, Zehao Huang, Naiyan Wang, Zhaoxiang Zhang
- Abstract summary: We propose a general learnable graph matching method to address these issues.
Our method achieves state-of-the-art performance on several MOT datasets.
For image matching, our method outperforms state-of-the-art methods on a popular indoor dataset, ScanNet.
- Score: 74.28753343714858
- License: http://creativecommons.org/licenses/by/4.0/
- Abstract: Data association is at the core of many computer vision tasks, e.g., multiple
object tracking, image matching, and point cloud registration. however, current
data association solutions have some defects: they mostly ignore the intra-view
context information; besides, they either train deep association models in an
end-to-end way and hardly utilize the advantage of optimization-based
assignment methods, or only use an off-the-shelf neural network to extract
features. In this paper, we propose a general learnable graph matching method
to address these issues. Especially, we model the intra-view relationships as
an undirected graph. Then data association turns into a general graph matching
problem between graphs. Furthermore, to make optimization end-to-end
differentiable, we relax the original graph matching problem into continuous
quadratic programming and then incorporate training into a deep graph neural
network with KKT conditions and implicit function theorem. In MOT task, our
method achieves state-of-the-art performance on several MOT datasets. For image
matching, our method outperforms state-of-the-art methods on a popular indoor
dataset, ScanNet. For point cloud registration, we also achieve competitive
results. Code will be available at https://github.com/jiaweihe1996/GMTracker.
Related papers
- MGNet: Learning Correspondences via Multiple Graphs [78.0117352211091]
Learning correspondences aims to find correct correspondences from the initial correspondence set with an uneven correspondence distribution and a low inlier rate.
Recent advances usually use graph neural networks (GNNs) to build a single type of graph or stack local graphs into the global one to complete the task.
We propose MGNet to effectively combine multiple complementary graphs.
arXiv Detail & Related papers (2024-01-10T07:58:44Z) - Co-attention Graph Pooling for Efficient Pairwise Graph Interaction
Learning [19.58671020943416]
Graph Neural Networks (GNNs) have proven to be effective in processing and learning from graph-structured data.
We propose a novel and efficient graph-level approach for extracting interaction representations using co-attention in graph pooling.
Our method, Co-Attention Graph Pooling (CAGPool), exhibits competitive performance relative to existing methods in both classification and regression tasks.
arXiv Detail & Related papers (2023-07-28T07:53:34Z) - A Robust Stacking Framework for Training Deep Graph Models with
Multifaceted Node Features [61.92791503017341]
Graph Neural Networks (GNNs) with numerical node features and graph structure as inputs have demonstrated superior performance on various supervised learning tasks with graph data.
The best models for such data types in most standard supervised learning settings with IID (non-graph) data are not easily incorporated into a GNN.
Here we propose a robust stacking framework that fuses graph-aware propagation with arbitrary models intended for IID data.
arXiv Detail & Related papers (2022-06-16T22:46:33Z) - Neural Graph Matching for Pre-training Graph Neural Networks [72.32801428070749]
Graph neural networks (GNNs) have been shown powerful capacity at modeling structural data.
We present a novel Graph Matching based GNN Pre-Training framework, called GMPT.
The proposed method can be applied to fully self-supervised pre-training and coarse-grained supervised pre-training.
arXiv Detail & Related papers (2022-03-03T09:53:53Z) - Learnable Graph Matching: Incorporating Graph Partitioning with Deep
Feature Learning for Multiple Object Tracking [58.30147362745852]
Data association across frames is at the core of Multiple Object Tracking (MOT) task.
Existing methods mostly ignore the context information among tracklets and intra-frame detections.
We propose a novel learnable graph matching method to address these issues.
arXiv Detail & Related papers (2021-03-30T08:58:45Z) - Scalable Graph Neural Networks for Heterogeneous Graphs [12.44278942365518]
Graph neural networks (GNNs) are a popular class of parametric model for learning over graph-structured data.
Recent work has argued that GNNs primarily use the graph for feature smoothing, and have shown competitive results on benchmark tasks.
In this work, we ask whether these results can be extended to heterogeneous graphs, which encode multiple types of relationship between different entities.
arXiv Detail & Related papers (2020-11-19T06:03:35Z) - NodeNet: A Graph Regularised Neural Network for Node Classification [0.0]
Most AI/ML techniques leave out the linkages among data points.
Recent surge of interest in graph-based AI/ML techniques is aimed to leverage the linkages.
We propose a model using NGL - NodeNet, to solve node classification task for citation graphs.
arXiv Detail & Related papers (2020-06-16T09:41:58Z) - Graph Ordering: Towards the Optimal by Learning [69.72656588714155]
Graph representation learning has achieved a remarkable success in many graph-based applications, such as node classification, prediction, and community detection.
However, for some kind of graph applications, such as graph compression and edge partition, it is very hard to reduce them to some graph representation learning tasks.
In this paper, we propose to attack the graph ordering problem behind such applications by a novel learning approach.
arXiv Detail & Related papers (2020-01-18T09:14:16Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.