Graph-MLP

本文提出了一种名为Graph-MLP的新型模型,首次在节点分类任务中摒弃了传统GNN的信息传递机制。Graph-MLP仅包含多层感知机、激活函数和正则化层,通过NeighboringContrastiveLoss来利用邻接矩阵,实现在提高计算效率的同时,达到与GNN相当甚至超越的效果。模型在不损失性能的情况下简化了结构,提高了模型的稳定性和计算速度。

Info

最近很热门的一篇文章

代码已经开源
https://github.com/yanghu819/Graph-MLP

1 Motivation

基于谱和基于空间的GNN的特征聚合方法都依赖邻接矩阵去完成信息传递
文章的motivation认为GNN里面的信息传递模块是不必要的

2 Contribution

提出了不使用信息传递机制的Graph-MLP,这是第一次应用在节点分类任务上

提出了Graph-MLP,去掉信息传递模块

model level仅包含MLP,激活函数,正则化层

loss level设计了neighboring contrastive loss,使用邻接矩阵去缩小MLP和GNN的差距

提出了Neighboring Contrastive loss

实验显示Graph-MLP的效果很理想,在提高计算效率的情况下能达到和现有GNN的同样效果,某些情况下甚至能超过现有的baseline

3 Solution

传统GNN计算方法

使用正则化后的邻接矩阵更新节点特征X
一般经过L层得到结点特征的最终表示

Graph-MLP

结构

结构
input:随机采样的B个结点,以及它们的特征

然后传入多层MLP,得到节点特征以及两个输出
公式化

其中X是结点特征矩阵,LN是图中的Layer Norm
Z 用作计算NContrast loss
Y 用作计算classification loss

Loss

在这里使用邻接矩阵的信息

  • NContrast loss

    计算公式


sim衡量两个节点的相似度,

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值