arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2002.00727stat.MLcs.LG
收藏

图结构数据的距离度量学习

Distance Metric Learning for Graph Structured Data

Nagoya Institute of Technology(名古屋工业大学);National Institute for Material Science(国立材料科学研究所);RIKEN Center for Advanced Intelligence Project(理化学研究所先进智能项目中心)

Tomoki Yoshida, Ichiro Takeuchi, Masayuki Karasuyama

摘要:

图是表示结构化数据的通用工具。因此,已经研究了用于图形数据分析的各种机器学习方法。虽然许多这样的学习方法依赖于输入图之间的差异的测量,定义一个适当的距离度量图仍然是一个有争议的问题。因此,我们提出了一个监督的距离度量学习方法的图分类问题。我们的方法,命名为可解释的图度量学习(IGML),学习判别度量的子图为基础的特征空间,它具有很强的图形表示能力。通过对每个子图的权重引入稀疏诱导惩罚,IGML可以识别少量重要的子图,这些子图可以提供对给定分类任务的洞察。由于我们的配方有大量的优化变量,一个有效的算法,使用修剪技术的基础上安全筛选和工作集选择方法也被提出。IGML的一个重要性质是,解决方案的最优性得到保证,因为问题是制定为凸问题,我们的剪枝策略只删除不必要的子图。此外,我们表明,IGML也适用于其他结构化数据,如项目集和序列数据,它可以通过使用基于运输的子图功能,将顶点标签相似性。我们经验性地评估了IGML在几个基准数据集上的计算效率和分类性能,并提供了一些说明性的例子,说明IGML如何从给定的图数据集中识别重要的子图。

英文摘要:

Graphs are versatile tools for representing structured data. As a result, a variety of machine learning methods have been studied for graph data analysis. Although many such learning methods depend on the measurement of differences between input graphs, defining an appropriate distance metric for graphs remains a controversial issue. Hence, we propose a supervised distance metric learning method for the graph classification problem. Our method, named interpretable graph metric learning (IGML), learns discriminative metrics in a subgraph-based feature space, which has a strong graph representation capability. By introducing a sparsity-inducing penalty on the weight of each subgraph, IGML can identify a small number of important subgraphs that can provide insight into the given classification task. Because our formulation has a large number of optimization variables, an efficient algorithm that uses pruning techniques based on safe screening and working set selection methods is also proposed. An important property of IGML is that solution optimality is guaranteed because the problem is formulated as a convex problem and our pruning strategies only discard unnecessary subgraphs. Furthermore, we show that IGML is also applicable to other structured data such as itemset and sequence data, and that it can incorporate vertex-label similarity by using a transportation-based subgraph feature. We empirically evaluate the computational efficiency and classification performance of IGML on several benchmark datasets and provide some illustrative examples of how IGML identifies important subgraphs from a given graph dataset.

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

↑