4 个月前

IsoNN:同构神经网络用于图表示学习和分类

IsoNN:同构神经网络用于图表示学习和分类

摘要

深度学习模型在计算机视觉和自然语言处理等多个领域取得了巨大成功。然而,由于图数据具有“节点无序”(node-orderless)特性,传统深度学习模型难以直接应用于图数据。通常情况下,邻接矩阵会为图施加一种人为且随机的节点顺序,这使得深度模型在图分类任务中的性能极为不稳定,且这些模型所学习到的表示缺乏明确的可解释性。为了消除不必要的节点顺序约束,我们提出了一种名为同构神经网络(Isomorphic Neural Network, IsoNN)的新模型,该模型通过输入图与模板之间的图匹配来提取其同构特征,从而学习图表示。IsoNN 主要包含两个组成部分:图同构特征提取组件和分类组件。图同构特征提取组件利用一组子图模板作为核变量,学习输入图中可能存在的子图模式,并计算同构特征。该组件中使用了一组置换矩阵来打破由矩阵表示带来的节点顺序。在 IsoNN 中,分类组件由三个全连接层组成。我们在基准数据集上进行了广泛的实验,实验结果证明了 IsoNN 的有效性,尤其是在与经典和最先进的图分类方法相比时更为显著。

代码仓库

jwzhanggy/isocapsnet
pytorch
GitHub 中提及

基准测试

基准方法指标
graph-classification-on-bp-fmri-97IsoNN
Accuracy: 64.9%
F1: 69.7%
graph-classification-on-bp-fmri-97IsoNN-fast
Accuracy: 62.3%
F1: 63.2%
graph-classification-on-hiv-dti-77IsoNN
Accuracy: 67.5%
F1: 68.3%
graph-classification-on-hiv-dti-77IsoNN-fast
Accuracy: 60.1%
F1: 61.9%
graph-classification-on-hiv-fmri-77IsoNN-Fast
Accuracy: 70.5%
F1: 69.9%
graph-classification-on-hiv-fmri-77IsoNN
Accuracy: 73.4%
F1: 72.2%
graph-classification-on-hiv-fmri-77-1IsoNN
Accuracy: 73.4
F1: 72.2
graph-classification-on-mutagFunction Space Pooling
Accuracy: 83.3%
graph-classification-on-ptcIsoNN
Accuracy: 59.9%

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
IsoNN:同构神经网络用于图表示学习和分类 | 论文 | HyperAI超神经