基于边对注意力的跨图嵌入深度图匹配
Cross-Graph Embedding with Edge-Wise Attention for Deep Graph Matching
摘要: 图匹配问题(GM)旨在建立两个图的点集之间的一致性,同时保持它们的边集之间的一致性,在数学上可以被刻画为二次组合优化问题。随着图神经网络的兴起和发展,基于深度学习的图匹配方法应用图神经网络技术将每个图的几何结构信息嵌入到结点特征,从而将二阶组合优化的图匹配问题松弛为线性分配问题。然而,这样的图匹配模型在本质上忽略了二阶的边对相似度信息,即没有考虑边集之间的一致性,从而导致这类模型在精度上有所损失。针对上述问题,本文提出了一种新型基于嵌入技术的深度图匹配网络,该网络充分考虑了边对相似性信息:基于二图匹配生成的关联图,将边对相似性信息(等价于关联图上的边)作为跨图嵌入的注意力信息,构造一个跨图注意力模块。本文基于公开数据集进行评估实验,结论表明本文的方法在精度上超越了已有的模型。除此之外,本文的方法相比于目前精度最高的模型,在内存消耗上具有优势,体现了该模型具有处理大规模图匹配任务的潜力。
Abstract: Graph matching problem (GM) refers to establishing correspondences between two set of points while keeping the consistency between their edge sets and can be mathematically formulated as a quadratic combinatorial optimization problem. With the success of graph neural network (GNN), recent learning-based graph matching research attempts to solve the problem by way of linear assignment via transferring local structure information into node embedding at each graph. However, such a technique omits second-order pairwise edge similarity in essence causing the decrease of accuracy. Based on this issue, this paper proposes a novel embedding-based graph matching network fully considering the pairwise edge similarity. Edge-to-edge similarity, equivalent to associated edges, is aggregated into nodes in the association graph derived from two matching graphs, to amplify the discriminativeness of crossgraph node embedding. Experimental evaluations are conducted on public datasets, and results show that our method outperforms the state-of-the-art in accuracy. In addition, compared with the state-of-the-art, the proposed method has superiority in space consumption, indicating the potential to solve large-scale graph matching problems.
文章引用:杨益枘, 石林. 基于边对注意力的跨图嵌入深度图匹配[J]. 计算机科学与应用, 2022, 12(4): 1089-1098. https://doi.org/10.12677/CSA.2022.124112

参考文献

[1] Fischler, M.A. and Bolles, R.C. (1981) Random Sample Consensus: A Paradigm for Model Fitting with Applications to Image Analysis and Automated Cartography. Communications of the ACM, 24, 381-395. [Google Scholar] [CrossRef
[2] Zhang, Z. (1994) Iterative Point Matching for Registration of Free-Form Curves and Surfaces. International Journal of Computer Vision, 13, 119-152. [Google Scholar] [CrossRef
[3] 张飞, 黄国兴. 基于ASIFT与图匹配的物体识别方法[J]. 计算机工程, 2016, 42(7): 153-158.
[4] Zhou, F. and De la Torre, F. (2016) Spatio-Temporal Matching for Human Pose Estima-tion in Video. IEEE Transactions on Pattern Analysis and Machine Intelligence, 38, 1492-1504. [Google Scholar] [CrossRef
[5] He, J., Huang, Z., Wang, N. and Zhang, Z. (2021) Learnable Graph Matching: Incorporating Graph Partitioning with Deep Feature Learning for Multiple Object Tracking. Proceed-ings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Nashville, 20-25 June 2021, 5299-5309. [Google Scholar] [CrossRef
[6] Chang, J.R. and Chen, Y.S. (2018) Pyramid Stereo Match-ing Network. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Salt Lake City, 18-23 June 2018, 5410-5418. [Google Scholar] [CrossRef
[7] Bai, X., Cheng, J. and Hancock, E. (Eds.) (2012) Graph-Based Methods in Computer Vision: Developments and Applications: Developments and Applications. IGI Global, Her-shey.
[8] Leordeanu, M. and Hebert, M. (2005) A Spectral Technique for Correspondence Problems Using Pairwise Constraints. International Conference of Computer Vision (ICCV), Beijing, 17-21 October 2005, 1482-1489.
[9] Schellewald, C. and Schnörr, C. (2005) Probabilistic Subgraph Matching Based on Convex Relaxation. International Workshop on Energy Minimization Methods in Computer Vision and Pattern Recognition, St. Augustine, 9-11 November 2005, 171-186. [Google Scholar] [CrossRef
[10] Cho, M., Lee, J. and Lee, K.M. (2010) Reweighted Random Walks for Graph Matching. European Conference on Computer Vision, Heraklion, 5-11 September 2010, 492-505. [Google Scholar] [CrossRef
[11] Zaslavskiy, M., Bach, F. and Vert, J.P. (2008) A Path Following Algorithm for the Graph Matching Problem. IEEE Transactions on Pattern Analysis and Machine In-telligence, 31, 2227-2242. [Google Scholar] [CrossRef
[12] Gold, S. and Rangarajan, A. (1996) A Graduated Assignment Algorithm for Graph Matching. IEEE Transactions on Pattern Analysis and Machine Intelligence, 18, 377-388. [Google Scholar] [CrossRef
[13] Zanfir, A. and Sminchisescu, C. (2018) Deep Learning of Graph Matching. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Salt Lake City, 18-23 June 2018, 2684-2693. [Google Scholar] [CrossRef
[14] Wang, R., Yan, J. and Yang, X. (2020) Combinatorial Learning of Robust Deep Graph Matching: An Embedding Based Approach. IEEE Transactions on Pattern Analysis and Machine Intelligence. [Google Scholar] [CrossRef
[15] Kipf, T.N. and Welling, M. (2016) Semi-Supervised Classifi-cation with Graph Convolutional Networks. arXiv preprint arXiv:1609.02907.
[16] Cuturi, M. (2013) Sinkhorn Dis-tances: Lightspeed Computation of Optimal Transport. Proceedings of the 26th International Conference on Neural In-formation Processing Systems, Vol. 2, Lake Tahoe, 5-10 December 2013, 2292-2300.
[17] Wang, T., Liu, H., Li, Y., Jin, Y., Hou, X. and Ling, H. (2020) Learning Combinatorial Solver for Graph Matching. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Seattle, 13-19 June 2020, 7565-7574. [Google Scholar] [CrossRef
[18] Qu, J., Ling, H., Zhang, C., Lyu, X. and Tang, Z. (2021) Adaptive Edge Attention for Graph Matching with Outliers. Proceedings of the 20th International Joint Conference on Artificial Intelligence, Montreal, 19-27 August 2021, 966-972. [Google Scholar] [CrossRef
[19] Wang, R., Yan, J. and Yang, X. (2021) Neural Graph Matching Net-work: Learning Lawler’s Quadratic Assignment Problem with Extension to Hypergraph and Multiple-Graph Matching. IEEE Transactions on Pattern Analysis and Machine Intelligence. [Google Scholar] [CrossRef
[20] Zhou, F. and De la Torre, F. (2015) Factorized Graph Match-ing. IEEE Transactions on Pattern Analysis and Machine Intelligence, 38, 1774-1789. [Google Scholar] [CrossRef
[21] Fey, M., Lenssen, J.E., Morris, C., Masci, J. and Kriege, N.M. (2020) Deep Graph Matching Consensus. International Conference on Learning Representations, Addis Ababa, 26-30 April 2020.
[22] Lee, D.T. and Schachter, B.J. (1980) Two Algorithms for Constructing a Delaunay Triangulation. In-ternational Journal of Computer & Information Sciences, 9, 219-242. [Google Scholar] [CrossRef
[23] Everingham, M., Van Gool, L., Williams, C.K.I., Winn, J. and Zisser-man, A. (2010) The Pascal Visual Object Classes (VOC) Challenge. International Journal of Computer Vision, 88, 303-338. [Google Scholar] [CrossRef
[24] Cho, M., Alahari, K. and Ponce, J. (2013) Learning Graphs to Match. Proceedings of the IEEE International Conference on Computer Vision, Sydney, 1-8 December 2013, 25-32. [Google Scholar] [CrossRef
[25] Kuhn, H.W. (1955) The Hungarian Method for the Assignment Problem. Naval Research Logistics Quarterly, 2, 83-97. [Google Scholar] [CrossRef
[26] Yu, T., Wang, R., Yan, J. and Li, B. (2019) Learning Deep Graph Matching with Channel-Independent Embedding and Hungarian Attention. International Conference on Learning Rep-resentations, New Orleans, 6-9 May 2019.
[27] Bourdev, L. and Malik, J. (2009) Poselets: Body Part Detectors Trained Using 3D Human Pose Annotations. 2009 IEEE 12th International Conference on Computer Vision, Kyoto, 29 Septem-ber-2 October 2009, 1365-1372. [Google Scholar] [CrossRef