论文笔记:GGNN (门控图神经网络)

基本概念
GGNN是一种基于GRU的经典的空间域message passing的模型

问题描述
一个图 G = (V, E), 节点v ∈ V中存储D维向量,边e ∈ E中存储D × D维矩阵, 目的是构建网络GGNN。
实现每一次参数更新时,每个节点既接受相邻节点的信息,又向相邻节点发送信息

主要贡献

基于GRU提出了GGNN,利用RNN类似原理实现了信息在graph中的传递。
论文笔记:GGNN (门控图神经网络)_第1张图片
传播模型
论文笔记:GGNN (门控图神经网络)_第2张图片
论文笔记:GGNN (门控图神经网络)_第3张图片
输出模型
论文笔记:GGNN (门控图神经网络)_第4张图片
来源论文
Gated Graph Sequence Neural Networks,ICLR 2016
链接:https://arxiv.org/abs/1511.05493
官方实现(Lua):https://github.com/yujiali/ggnn
第三方实现(pytorch):https://github.com/calebmah/ggnn.pytorch
GRU概念详见:https://blog.csdn.net/lthirdonel/article/details/88945257

你可能感兴趣的:(机器学习)