4 个月前

BernNet:通过伯恩斯坦逼近学习任意图谱滤波器

BernNet:通过伯恩斯坦逼近学习任意图谱滤波器

摘要

许多具有代表性的图神经网络,例如GPR-GNN和ChebNet,通过图谱滤波器来近似图卷积。然而,现有的研究要么使用预定义的滤波器权重,要么在没有必要约束的情况下学习这些权重,这可能导致滤波器过于简化或不适定。为了解决这些问题,我们提出了BernNet,这是一种具有理论支持的新型图神经网络,提供了一种简单而有效的方案来设计和学习任意图谱滤波器。具体而言,对于任何基于图的归一化拉普拉斯谱的滤波器,我们的BernNet通过K阶伯恩斯坦多项式近似来估计它,并通过设置伯恩斯坦基底的系数来设计其谱特性。此外,我们可以根据观察到的图及其相关信号来学习这些系数(以及相应的滤波器权重),从而实现针对数据专门化的BernNet。实验结果表明,BernNet可以学习任意谱滤波器,包括复杂的带阻和梳状滤波器,并在实际的图建模任务中表现出优越的性能。代码可在https://github.com/ivam-he/BernNet 获取。

代码仓库

ivam-he/BernNet
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
node-classification-on-chameleon-60-20-20BernNet
1:1 Accuracy: 68.29 ± 1.58
node-classification-on-citeseer-60-20-20BernNet
1:1 Accuracy: 80.09 ± 0.79
node-classification-on-cora-60-20-20-randomBernNet
1:1 Accuracy: 88.52 ± 0.95
node-classification-on-cornell-60-20-20BernNet
1:1 Accuracy: 92.13 ± 1.64
node-classification-on-film-60-20-20-randomBernNet
1:1 Accuracy: 41.79 ± 1.01
node-classification-on-non-homophilicBernNet
1:1 Accuracy: 92.13 ± 1.64
node-classification-on-non-homophilic-2BernNet
1:1 Accuracy: 93.12 ± 0.65
node-classification-on-non-homophilic-4BernNet
1:1 Accuracy: 68.29 ± 1.58
node-classification-on-pubmed-60-20-20-randomBernNet
1:1 Accuracy: 88.48 ± 0.41
node-classification-on-squirrel-60-20-20BernNet
1:1 Accuracy: 51.35 ± 0.73
node-classification-on-texas-60-20-20-randomBernNet
1:1 Accuracy: 93.12 ± 0.65

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
BernNet:通过伯恩斯坦逼近学习任意图谱滤波器 | 论文 | HyperAI超神经