
CapsGNN: PyTorch中的“胶囊图神经网络”实现(ICLR 2019)
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
CapsGNN是基于PyTorch框架实现的一种新型模型,结合了胶囊网络和图神经网络的优势,适用于处理图结构数据,已在ICLR 2019会议上展示。
CapsGNN 是一种基于胶囊图神经网络的PyTorch实现(ICLR 2019)。从图神经网络(GNN)学习到的高质量节点嵌入技术已被广泛应用在基于节点的应用程序中,并且一些应用已经达到了最先进的性能水平。然而,当使用从GNN学习得到的节点嵌入来生成图形嵌入时,简单的标量表示可能不足以有效保留节点或整个图的重要属性,导致次优的结果。
受到胶囊神经网络(CapsNet)的启发,我们提出了胶囊图神经网络(CapsGNN),利用了胶囊的概念以解决现有基于GNN的方法中的不足。通过采用胶囊形式来提取节点特征,并使用路由机制在图形级别上捕获重要信息,我们的模型能够为每个图生成多个嵌入。
全部评论 (0)
还没有任何评论哟~


