Revocable Deep Reinforcement Learning with Affinity Regularization for Outlier-Robust Graph Matching
Chang Liu, Zetian Jiang, Runzhong Wang, Lingxiao Huang, Pinyan Lu, Junchi Yan
Abstract
Graph matching (GM) has been a building block in various areas including computer vision and pattern recognition. Despite recent impressive progress, existing deep GM methods often have obvious difficulty in handling outliers, which are ubiquitous in practice. We propose a deep reinforcement learning based approach RGM, whose sequential node matching scheme naturally fits the strategy for selective inlier matching against outliers. A revocable action framework is devised to improve the agent's flexibility against the complex constrained GM. Moreover, we propose a quadratic approximation technique to regularize the affinity score, in the presence of outliers. As such, the agent can finish inlier matching timely when the affinity score stops growing, for which otherwise an additional parameter i.e. the number of inliers is needed to avoid matching outliers. In this paper, we focus on learning the back-end solver under the most general form of GM: the Lawler's QAP, whose input is the affinity matrix. Especially, our approach can also boost existing GM methods that use such input. Experiments on multiple real-world datasets demonstrate its performance regarding both accuracy and robustness.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 12c94fb5-9761-4bb4-9ac7-39f0f08bcbacCited by top-tier papers8
- Deep Neural Network Fusion via Graph Matching with Applications to Model Ensemble and Federated LearningChang Liu, Chenfei Lou, Runzhong Wang, Alan Yuhan Xi et al.ICML 2022 · 72 citations
- LinSATNet: The Positive Linear Satisfiability Neural NetworksRunzhong Wang, Yunhao Zhang, Ziao Guo, Tianyi Chen et al.ICML 2023 · 27 citations
- L2P-MIP: Learning to Presolve for Mixed Integer ProgrammingChang Liu, Zhichen Dong, Haobo Ma, Weilin Luo et al.ICLR 2024 · 10 citations
- Learning Solution-Aware Transformers for Efficiently Solving Quadratic Assignment ProblemZhentao Tan, Yadong MuICML 2024 · 5 citations
- MixSATGEN: Learning Graph Mixing for SAT Instance GenerationXinyan Chen, Yang Li, Runzhong Wang, Junchi YanICLR 2024 · 3 citations
Builds on13
- Learning Two-View Correspondences and Geometry Using Order-Aware NetworkJiahui Zhang, Dawei Sun, Zixin Luo, Anbang Yao et al.ICCV 2019 · 362 citations
- Learning Combinatorial Embedding Networks for Deep Graph MatchingRunzhong Wang, Junchi Yan, Xiaokang YangICCV 2019 · 268 citations
- Deep Graph Matching ConsensusMatthias Fey, Jan Eric Lenssen, Christopher Morris, Jonathan Masci et al.ICLR 2020 · 227 citations
- Exploratory Combinatorial Optimization with Reinforcement LearningThomas D. Barrett, William R. Clements, Jakob N. Foerster, A. I. LvovskyAAAI 2020 · 218 citations
- Learning deep graph matching with channel-independent embedding and Hungarian attentionTianshu Yu, Runzhong Wang, Junchi Yan, Baoxin LiICLR 2020 · 113 citations
Related papers
- Learning Combinatorial Solver for Graph MatchingTao Wang, He Liu, Yidong Li, Yi Jin et al.CVPR 2020
- GAMnet: Robust Feature Matching via Graph Adversarial-Matching NetworkBo Jiang, Pengfei Sun, Ziyan Zhang, Jin Tang et al.ACM MM 2021 · 8 citations
- Self-Supervised Bidirectional Learning for Graph MatchingWenqi Guo, Lin Zhang, Shikui Tu, Lei XuAAAI 2023 · 5 citations
- Deep Learning of Partial Graph Matching via Differentiable Top-KRunzhong Wang, Ziao Guo, Shaofei Jiang, Xiaokang Yang et al.CVPR 2023
- IA-GM: A Deep Bidirectional Learning Method for Graph MatchingKaixuan Zhao, Shikui Tu, Lei XuAAAI 2021 · 13 citations
