4 个月前

关系池化用于图表示

关系池化用于图表示

摘要

这项研究将图神经网络(GNNs)的应用范围扩展到了超越基于Weisfeiler-Lehman(WL)算法、图拉普拉斯矩阵和扩散模型的范畴。我们提出的方法称为关系池化(Relational Pooling,RP),借鉴了有限部分可交换性的理论,为图提供了一个具有最大表示能力的框架。RP可以与现有的图表示模型结合使用,并且有些反直觉的是,它可以使得这些模型比原始的WL同构测试更加强大。此外,RP允许递归神经网络(Recurrent Neural Networks,RNNs)和卷积神经网络(Convolutional Neural Networks,CNNs)以一种理论上严谨的方式应用于图分类任务。我们在多个任务上展示了基于RP的图表示方法相较于现有最先进方法的性能提升。

代码仓库

PurdueMINDS/RelationalPooling
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
drug-discovery-on-hiv-datasetRNN-DFS
AUC: 0.627
drug-discovery-on-muvRNN-DFS
AUC: 0.648
drug-discovery-on-tox21RNN-DFS
AUC: 0.748

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
关系池化用于图表示 | 论文 | HyperAI超神经