
摘要
字典学习是表示学习中的关键工具,能够将数据解释为少量基本元素的线性组合。然而,在图学习的背景下,这种分析方法难以适用,因为图通常属于不同的度量空间。为此,本文提出了一种新型的在线图字典学习方法,采用Gromov-Wasserstein散度作为数据拟合项。在本研究中,图通过其节点间的成对关系进行编码,并被建模为图原子(即字典元素)的凸组合,这些图原子通过一种在线随机算法进行估计,该算法作用于一组未配准的图数据集,且这些图可能具有不同数量的节点。所提出的方法自然可扩展至有标签图,并进一步引入了一种新颖的上界估计,可在嵌入空间中作为Gromov-Wasserstein距离的快速近似。数值实验结果表明,该方法在无监督图数据嵌入以及在线图子空间估计与追踪任务中均展现出显著优势。
代码仓库
cedricvincentcuaz/GDL
官方
pytorch
GitHub 中提及
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| graph-classification-on-bzr | GDL-g (ADJ) | Accuracy: 87.81 |
| graph-classification-on-cox2 | GDL-g (ADJ) | Accuracy(10-fold): 78.11 |
| graph-classification-on-enzymes | GDL-g (SP) | Accuracy: 71.47 |
| graph-classification-on-imdb-b | GDL | Accuracy: 72.06% Rand index: 51.64 |
| graph-classification-on-imdb-m | GDL | Accuracy: 50.64% |
| graph-classification-on-mutag | GDL-g (SP) | Accuracy: 87.09% |
| graph-classification-on-mutag | GDL-g (ADJ) | Accuracy: 58.45% |
| graph-classification-on-proteins | GDL-g (SP) | Accuracy: 74.86 |