4 个月前

Wasserstein Weisfeiler-Lehman 图核函数

Wasserstein Weisfeiler-Lehman 图核函数

摘要

大多数图核方法都是$\mathcal{R}$-卷积核类的实例,通过比较对象的子结构来衡量其相似性。尽管这些图核在实际应用中取得了成功,但它们通常采用一种简单的最终子结构集合聚合方法(通常是求和或平均),这可能导致丢弃关于各个组件分布的宝贵信息。此外,只有少数这类方法可以扩展到连续属性图上。我们提出了一种新颖的方法,该方法基于两个图的节点特征向量分布之间的Wasserstein距离,通过将图视为高维对象而非简单均值,能够发现数据集中的细微差异。我们进一步提出了一种受Weisfeiler-Lehman启发的嵌入方案,适用于具有连续节点属性和加权边的图,并通过计算得到的Wasserstein距离对其进行增强,从而在多个图分类任务中提升了最先进的预测性能。

代码仓库

wenbintum/wwl-gpr
GitHub 中提及
BorgwardtLab/WWL
官方
GitHub 中提及

基准测试

基准方法指标
graph-classification-on-ddWWL
Accuracy: 79.69%
graph-classification-on-enzymesWWL
Accuracy: 59.13%
graph-classification-on-mutagWWL
Accuracy: 87.27%
graph-classification-on-nci1WWL
Accuracy: 85.75%
graph-classification-on-proteinsWWL
Accuracy: 74.28%
graph-classification-on-ptcWWL
Accuracy: 66.31%

用 AI 构建 AI

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

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

Hyper Newsletters

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