可能是最好用的图神经网络框架, PyG 2.0发布

2021 年 9 月 13 日 图与推荐
PyG (PyTorch Geometric) 可能是最好用的图神经网络框架,今天刚刚发布其2.0版本。

PyG 2.0 有如下的核心变化:

  • 异质图的支持。多种类型和节点的图如何简单高效的载入和实现GNN模型?
  • GraphGym的整合。考虑不同的层数,聚合函数,架构,任务,GNN可以有多种组合形式(笛卡尔积)的设计。20NIPS GraphGym Design Space for Graph Neural Networks 针对上述问题进行了系统性的研究,分析了 315,000 种设计在 32种任务上设计空间。
  • GNN选取的指导意见(小抄)。如何针对不同图数据选取合适的GNN架构?
  • 效率提升和Bug修复。

异质图神经网络

很多GNN都在研究只有一种类型节点和边的简单图。然而,在现实生活中,图往往包含多种类型节点和关系,称作异质图。例如,推荐场景下,只要也有建模用户和商品之间的交互关系。

异质图有很高的实用价值,但是其设计到多种类型节点及其属性,不同类型的交互,非常复杂,实现起来也很麻烦。亚马逊DGL已经支持异质图了,PyG终于赶上了。

GraphGym

考虑不同的层数,聚合函数,架构,任务,GNN可以有多种组合形式(笛卡尔积)的设计。20NIPS GraphGym Design Space for Graph Neural Networks 针对上述问题进行了系统性的研究,分析了315,000 种设计在 32种任务上设计空间。感兴趣的可以去围观一波原文(https://arxiv.org/abs/2011.08843)。

GNN小抄

如何根据图的特性来选取合适的GNN模型?

这里给出了稀疏图,带权图,二部图,静态图等不同类型图如何选取相应的GNN模型的指导意见。

https://github.com/pyg-team/pytorch_geometric/releases/tag/2.0.0

登录查看更多
0

相关内容

图神经网络 (GNN) 是一种连接模型,它通过图的节点之间的消息传递来捕捉图的依赖关系。与标准神经网络不同的是,图神经网络保留了一种状态,可以表示来自其邻域的具有任意深度的信息。近年来,图神经网络(GNN)在社交网络、知识图、推荐系统、问答系统甚至生命科学等各个领域得到了越来越广泛的应用。

知识荟萃

精品入门和进阶教程、论文和代码整理等

更多

查看相关VIP内容、论文、资讯等
专知会员服务
45+阅读 · 2021年6月20日
专知会员服务
95+阅读 · 2021年5月25日
【WWW2021】用优化框架解释和统一图神经网络
专知会员服务
44+阅读 · 2021年2月1日
【NeurIPS 2020】图神经网络GNN架构设计
专知会员服务
83+阅读 · 2020年11月19日
一份简单《图神经网络》教程,28页ppt
专知会员服务
123+阅读 · 2020年8月2日
专知会员服务
118+阅读 · 2020年7月22日
一网打尽!100+深度学习模型TensorFlow与Pytorch代码实现集合
TensorFlow 2.0 学习资源汇总
专知会员服务
66+阅读 · 2019年10月9日
推出 TensorFlow 图神经网络 (GNNs)
谷歌开发者
0+阅读 · 2021年12月30日
推出 TensorFlow 图神经网络(GNNs)
TensorFlow
0+阅读 · 2021年12月29日
TensorFlow官方出了个GNN框架,YYDS!
图与推荐
0+阅读 · 2021年11月20日
入群交流一波图神经网络框架 PyG 2.0 吧~
图与推荐
3+阅读 · 2021年11月17日
ICLR2021 | 初探GNN的表示能力
专知
1+阅读 · 2021年5月2日
图神经网络框架 DGL V0.6发布,更快更好用
图与推荐
1+阅读 · 2021年3月1日
PyTorch & PyTorch Geometric图神经网络(GNN)实战
专知
81+阅读 · 2019年6月1日
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2013年12月31日
国家自然科学基金
2+阅读 · 2013年12月31日
国家自然科学基金
0+阅读 · 2012年12月31日
国家自然科学基金
0+阅读 · 2012年12月31日
国家自然科学基金
0+阅读 · 2012年12月31日
国家自然科学基金
0+阅读 · 2012年12月31日
国家自然科学基金
0+阅读 · 2012年12月31日
国家自然科学基金
1+阅读 · 2012年12月31日
Arxiv
0+阅读 · 2022年4月20日
Arxiv
0+阅读 · 2022年4月19日
Arxiv
13+阅读 · 2021年6月14日
Arxiv
19+阅读 · 2021年2月4日
Heterogeneous Graph Transformer
Arxiv
27+阅读 · 2020年3月3日
Arxiv
15+阅读 · 2020年2月5日
Arxiv
35+阅读 · 2020年1月2日
VIP会员
相关VIP内容
专知会员服务
45+阅读 · 2021年6月20日
专知会员服务
95+阅读 · 2021年5月25日
【WWW2021】用优化框架解释和统一图神经网络
专知会员服务
44+阅读 · 2021年2月1日
【NeurIPS 2020】图神经网络GNN架构设计
专知会员服务
83+阅读 · 2020年11月19日
一份简单《图神经网络》教程,28页ppt
专知会员服务
123+阅读 · 2020年8月2日
专知会员服务
118+阅读 · 2020年7月22日
一网打尽!100+深度学习模型TensorFlow与Pytorch代码实现集合
TensorFlow 2.0 学习资源汇总
专知会员服务
66+阅读 · 2019年10月9日
相关资讯
推出 TensorFlow 图神经网络 (GNNs)
谷歌开发者
0+阅读 · 2021年12月30日
推出 TensorFlow 图神经网络(GNNs)
TensorFlow
0+阅读 · 2021年12月29日
TensorFlow官方出了个GNN框架,YYDS!
图与推荐
0+阅读 · 2021年11月20日
入群交流一波图神经网络框架 PyG 2.0 吧~
图与推荐
3+阅读 · 2021年11月17日
ICLR2021 | 初探GNN的表示能力
专知
1+阅读 · 2021年5月2日
图神经网络框架 DGL V0.6发布,更快更好用
图与推荐
1+阅读 · 2021年3月1日
PyTorch & PyTorch Geometric图神经网络(GNN)实战
专知
81+阅读 · 2019年6月1日
相关基金
国家自然科学基金
0+阅读 · 2015年12月31日
国家自然科学基金
0+阅读 · 2013年12月31日
国家自然科学基金
2+阅读 · 2013年12月31日
国家自然科学基金
0+阅读 · 2012年12月31日
国家自然科学基金
0+阅读 · 2012年12月31日
国家自然科学基金
0+阅读 · 2012年12月31日
国家自然科学基金
0+阅读 · 2012年12月31日
国家自然科学基金
0+阅读 · 2012年12月31日
国家自然科学基金
1+阅读 · 2012年12月31日
相关论文
Arxiv
0+阅读 · 2022年4月20日
Arxiv
0+阅读 · 2022年4月19日
Arxiv
13+阅读 · 2021年6月14日
Arxiv
19+阅读 · 2021年2月4日
Heterogeneous Graph Transformer
Arxiv
27+阅读 · 2020年3月3日
Arxiv
15+阅读 · 2020年2月5日
Arxiv
35+阅读 · 2020年1月2日
Top
微信扫码咨询专知VIP会员