Ling Xiang, Wu Lingfei, Wang Saizhuo, Ma Tengfei, Xu Fangli, Liu Alex X, Wu Chunming, Ji Shouling
IEEE Trans Neural Netw Learn Syst. 2023 Feb;34(2):799-813. doi: 10.1109/TNNLS.2021.3102234. Epub 2023 Feb 3.
While the celebrated graph neural networks (GNNs) yield effective representations for individual nodes of a graph, there has been relatively less success in extending to the task of graph similarity learning. Recent work on graph similarity learning has considered either global-level graph-graph interactions or low-level node-node interactions, however, ignoring the rich cross-level interactions (e.g., between each node of one graph and the other whole graph). In this article, we propose a multilevel graph matching network (MGMN) framework for computing the graph similarity between any pair of graph-structured objects in an end-to-end fashion. In particular, the proposed MGMN consists of a node-graph matching network (NGMN) for effectively learning cross-level interactions between each node of one graph and the other whole graph, and a siamese GNN to learn global-level interactions between two input graphs. Furthermore, to compensate for the lack of standard benchmark datasets, we have created and collected a set of datasets for both the graph-graph classification and graph-graph regression tasks with different sizes in order to evaluate the effectiveness and robustness of our models. Comprehensive experiments demonstrate that MGMN consistently outperforms state-of-the-art baseline models on both the graph-graph classification and graph-graph regression tasks. Compared with previous work, multilevel graph matching network (MGMN) also exhibits stronger robustness as the sizes of the two input graphs increase.
尽管著名的图神经网络(GNN)能为图的各个节点生成有效的表示,但在扩展到图相似性学习任务方面取得的成功相对较少。最近关于图相似性学习的工作要么考虑全局级别的图-图交互,要么考虑低级别的节点-节点交互,然而,却忽略了丰富的跨级别交互(例如,一个图的每个节点与另一个完整图之间的交互)。在本文中,我们提出了一种多级图匹配网络(MGMN)框架,用于以端到端的方式计算任意一对图结构对象之间的图相似性。具体而言,所提出的MGMN由一个节点-图匹配网络(NGMN)和一个孪生GNN组成,前者用于有效学习一个图的每个节点与另一个完整图之间的跨级别交互,后者用于学习两个输入图之间的全局级交互。此外,为了弥补标准基准数据集的不足,我们创建并收集了一组用于图-图分类和图-图回归任务的不同规模的数据集,以评估我们模型的有效性和鲁棒性。综合实验表明,MGMN在图-图分类和图-图回归任务上始终优于当前的基线模型。与先前的工作相比,随着两个输入图规模的增加,多级图匹配网络(MGMN)也表现出更强的鲁棒性。