3 个月前

在线图字典学习

在线图字典学习

摘要

字典学习是表示学习中的关键工具,能够将数据解释为少量基本元素的线性组合。然而,在图学习的背景下,这种分析方法难以适用,因为图通常属于不同的度量空间。为此,本文提出了一种新型的在线图字典学习方法,采用Gromov-Wasserstein散度作为数据拟合项。在本研究中,图通过其节点间的成对关系进行编码,并被建模为图原子(即字典元素)的凸组合,这些图原子通过一种在线随机算法进行估计,该算法作用于一组未配准的图数据集,且这些图可能具有不同数量的节点。所提出的方法自然可扩展至有标签图,并进一步引入了一种新颖的上界估计,可在嵌入空间中作为Gromov-Wasserstein距离的快速近似。数值实验结果表明,该方法在无监督图数据嵌入以及在线图子空间估计与追踪任务中均展现出显著优势。

代码仓库

cedricvincentcuaz/GDL
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
graph-classification-on-bzrGDL-g (ADJ)
Accuracy: 87.81
graph-classification-on-cox2GDL-g (ADJ)
Accuracy(10-fold): 78.11
graph-classification-on-enzymesGDL-g (SP)
Accuracy: 71.47
graph-classification-on-imdb-bGDL
Accuracy: 72.06%
Rand index: 51.64
graph-classification-on-imdb-mGDL
Accuracy: 50.64%
graph-classification-on-mutagGDL-g (SP)
Accuracy: 87.09%
graph-classification-on-mutagGDL-g (ADJ)
Accuracy: 58.45%
graph-classification-on-proteinsGDL-g (SP)
Accuracy: 74.86

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
在线图字典学习 | 论文 | HyperAI超神经