arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2106.04739cs.LG
收藏

基于Weisfeiler-Lehman图核的子树模式重要性学习

Learning subtree pattern importance for Weisfeiler-Lehmanbased graph kernels

The University of Tokyo(东京大学);Kyoto University(京都大学);Alato University(阿尔托大学)

Dai Hai Nguyen, Canh Hao Nguyen, Hiroshi Mamitsuka

摘要:

图是关系数据的一种常用表示方法,在分子、生物、社交网络等领域中普遍存在。一种流行的方法来学习与图结构的数据是利用graph kernels,它衡量之间的相似性,并被插入到一个核心机器,如支持向量机。Weisfeiler-Lehman(WL)的基于图的核,采用WL标记方案来提取子树模式和执行节点嵌入,被证明可以实现很好的性能,同时是有效的可计算性。然而,通用核的一个主要缺点是核构造和学习过程的解耦。对于分子图,通常的核函数如WL子树,基于分子的子结构,认为所有可用的子结构具有相同的重要性,这可能不适合于实际应用。在本文中,我们提出了一种在WWL内核框架中学习子树模式权重的方法,这是图分类任务的最新方法[14]。为了克服大规模数据集上的计算问题,我们提出了一种有效的学习算法,并推导了一个泛化间隙界来证明其收敛性。最后,通过对合成数据集和真实数据集的实验,我们证明了我们提出的方法学习子树模式的权重的有效性。

英文摘要:

Graph is an usual representation of relational data, which are ubiquitous in manydomains such as molecules, biological and social networks. A popular approach to learningwith graph structured data is to make use of graph kernels, which measure the similaritybetween graphs and are plugged into a kernel machine such as a support vector machine.Weisfeiler-Lehman (WL) based graph kernels, which employ WL labeling scheme to extract subtree patterns and perform node embedding, are demonstrated to achieve great performance while being efficiently computable. However, one of the main drawbacks of ageneral kernel is the decoupling of kernel construction and learning process. For moleculargraphs, usual kernels such as WL subtree, based on substructures of the molecules, consider all available substructures having the same importance, which might not be suitable inpractice. In this paper, we propose a method to learn the weights of subtree patterns in the framework of WWL kernels, the state of the art method for graph classification task [14]. To overcome the computational issue on large scale data sets, we present an efficient learning algorithm and also derive a generalization gap bound to show its convergence. Finally, through experiments on synthetic and real-world data sets, we demonstrate the effectiveness of our proposed method for learning the weights of subtree patterns.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑