Self-Supervised Bidirectional Learning for Graph Matching

Authors

  • Wenqi Guo Shanghai Jiao Tong University
  • Lin Zhang Shanghai Jiao Tong University
  • Shikui Tu Shanghai Jiao Tong University
  • Lei Xu Shanghai Jiao Tong University

DOI:

https://doi.org/10.1609/aaai.v37i6.25943

Keywords:

ML: Graph-based Machine Learning, ML: Optimization

Abstract

Deep learning methods have demonstrated promising performance on the NP-hard Graph Matching (GM) problems. However, the state-of-the-art methods usually require the ground-truth labels, which may take extensive human efforts or be impractical to collect. In this paper, we present a robust self-supervised bidirectional learning method (IA-SSGM) to tackle GM in an unsupervised manner. It involves an affinity learning component and a classic GM solver. Specifically, we adopt the Hungarian solver to generate pseudo correspondence labels for the simple probabilistic relaxation of the affinity matrix. In addition, a bidirectional recycling consistency module is proposed to generate pseudo samples by recycling the pseudo correspondence back to permute the input. It imposes a consistency constraint between the pseudo affinity and the original one, which is theoretically supported to help reduce the matching error. Our method further develops a graph contrastive learning jointly with the affinity learning to enhance its robustness against the noise and outliers in real applications. Experiments deliver superior performance over the previous state-of-the-arts on five real-world benchmarks, especially under the more difficult outlier scenarios, demon- strating the effectiveness of our method.

Downloads

Published

2023-06-26

How to Cite

Guo, W., Zhang, L., Tu, S., & Xu, L. (2023). Self-Supervised Bidirectional Learning for Graph Matching. Proceedings of the AAAI Conference on Artificial Intelligence, 37(6), 7784-7792. https://doi.org/10.1609/aaai.v37i6.25943

Issue

Section

AAAI Technical Track on Machine Learning I