图神经网络及其变体
图卷积网络 (GCN) [研究论文] [教程] [PyTorch 代码] [MXNet 代码]
图注意力网络 (GAT) [研究论文] [教程] [PyTorch 代码] [MXNet 代码]:GAT 通过在节点的邻域部署多头注意力来扩展 GCN 的功能。这大大增强了模型的容量和表达能力。
关系图卷积网络 (Relational-GCN) [研究论文] [教程] [PyTorch 代码] [MXNet 代码]:关系图卷积网络允许图中两个实体之间存在多条边。具有不同关系的边被区别编码。
线图神经网络 (LGNN) [研究论文] [教程] [PyTorch 代码]:该网络通过检查图结构来关注社区检测。它使用原始图及其线图伴随的表示。除了展示算法如何利用多个图之外,此实现还展示了如何明智地混合简单的张量操作和稀疏矩阵张量操作,以及使用 DGL 进行消息传递。