作者:Rex Ying, Joe Lou, Jiaxuan You, Chengtao Wen (Arquimedes Canedo from Siemens, Jure Leskovec
{width=40%}
用于判断图是否同构(graph isomorphism)。
target graph: 大图
query graph: 小图
给定一个query,NeuroMatch的目标是在target graph中找到一些节点,这些节点的K-hop邻居包含了query。
将包含这个query的大图中所有节点的N-hop邻域识别为子图(subgraph)【可以理解为一个节点识别问题】。利用得到的子图,NeuroMatch使用一个GNN来学习每个节点的embedding,进而根据query和子图的embedding相似与否来刻画子图的结构与性质。
子图(subgraph)的结构同质性匹配是一个NP-complete问题,但是又有非常广泛的应用:
对target graph中的节点u,使用GNN得到$z_u$。这里作者使用GIN2模型的变体,GIN的效果和Weisfeiler-Lehman (WL) test一样有效,但是变体GIN比WL test更有效。
目标:确定$G_Q$是不是$G_T$的子图
问题转化:确定$G_q$是不是$G_u$的子图
方法:构造一个“子图预测函数”$f\left(z_{q}, z_{u}\right)$,进行二分类预测
k-hop邻居选取时,k的设定至少等于query graph的直径。
使用order embedding的方式,保证在通过子图结构构造embedding之后,得到的向量可以体现子图关系性质。具体地,考虑以下4个性质:
训练embedding的目标函数:最大边际损失 \(\begin{array}{c} \mathcal{L}\left(z_{q}, z_{u}\right)=\sum_{\left(z_{q}, z_{u}\right) \in P} E\left(z_{q}, z_{u}\right)+\sum_{\left(z_{q}, z_{u}\right) \in N} \max \left\{0, \alpha-E\left(z_{q}, z_{u}\right)\right\}, \text { where } \\ E\left(z_{q}, z_{u}\right)=\left\|\max \left\{0, z_{q}-z_{u}\right\}\right\|_{2}^{2} \end{array}\) 进而构造子图预测函数 \(f\left(z_{q}, z_{u}\right)=\left\{\begin{array}{ll} 1 & \text { iff } E\left(z_{q}, z_{u}\right)<t \\ 0 & \text { otherwise } \end{array}\right.\)
$\mathcal{N}^{(k)}(u)$表示节点u的kth-hop邻居。如果$q \in G_q$和节点$u \in G_u$ 是匹配的,那么所有的节点$i \in N^{(k)}(q)$,存在节点$j \in \mathcal{N}^{(l)}(u), l \leq k$,使得节点i和节点j是匹配的。
为了使得整个模型inductive,即可以在query graph未参与训练的情况下仍然有较高的预测精度,在训练时采用随机生成的query graph。
NeuroMatch是有监督的训练过程:构造positive pair和negative pair
合成数据集:Renyi random graphs, extended Barabasi graphs
真实数据集:(化学)COX2,(生物)Enzymes, DD, PPI networks,(图像处理)MSRC_21,(点云?)FIRSTMMDB,(知识图谱)WORDNET18
Graph Matching Neural Networks
Kun Xu, Liwei Wang, Mo Yu, Yansong Feng, Yan Song, Zhiguo Wang, and Dong Yu. Cross-lingual knowledge graph alignment via graph matching neural network. 2019.
RDGCN
Yuting Wu, Xiao Liu, Yansong Feng, Zheng Wang, Rui Yan, and Dongyan Zhao. Relation-aware entity alignment for heterogeneous knowledge graphs. 2019.
Shervashidze, N., Schweitzer, P., Van Leeuwen, E. J., Mehlhorn, K., & Borgwardt, K. M. (2011). Weisfeiler-lehman graph kernels. Journal of Machine Learning Research, 12(9). ↩
Xu, K., Hu, W., Leskovec, J., & Jegelka, S. (2018). How powerful are graph neural networks?. arXiv preprint arXiv:1810.00826. ↩