撑持异构图、集成GraphGym,超好用的图神经网络库PyG更新2.0版本

当前最流行和广泛使用的 GNN 库 PyG(PyTorch Geometric)现在出 2.0 版本了,新版本提供了全面的异构图撑持、GraphGam 以及很多其他特性,这一系列改进,为使用者带来了更好的用户体验。

PyTorch Geometric(PyG)是一个构建于 PyTorch 之上的库,用来为一系列与结构化数据相关的应用编写和训练图神经网络(GNN)。PyG 对机器进修研究者以及机器进修工具包的首次使用者都很友好。

撑持异构图、集成GraphGym,超好用的图神经网络库PyG更新2.0版本

就其特征和功能而言,PyG 具有易用和统一的 API,用户花费更少的时间在实现和运行图神经网络的底层机制上,只需要 10 至 20 行代码就可以训练自定义 GNN 模型。PyG 涵盖了大量的 SOTA GNN 架构以及训练和可扩展流程,并且易于扩展以匹配用户自己的特定用例或进行自己的 GNN 研究。

此外,PyG 中的全新 GraphGym 使得用户可以轻松地复现 GNN 实验,执行和分析数千个 GNN 实验,并将自定义模块嵌入到 GNN 进修 pipeline 中。

近日,PyG 焦点领导者(Core lead)Matthias Fey 宣布 PyG 2.0 版本来了!新版本提供了全面的异构图撑持、GraphGam 以及很多其他特性。

撑持异构图、集成GraphGym,超好用的图神经网络库PyG更新2.0版本

PyG 2.0 详情地址:https://github.com/pyg-team/pytorch_geometric/releases/tag/2.0.0

PyG 2.0 新特性

PyG 2.0 是一个新的版本,提供了复杂的异构图撑持、GraphGym 集成以及其他新特性。

PyG 2.0 中提供了完全的异构图撑持。异构图撑持包括数据保存层的完整重写(同时保持向后兼容性)、异构图转换、通过邻采样的关系型数据加载例程,以及一整套异构 GNN 模型 / 示例。

突出亮点

异构图保存:异构图现在可以保存在它们自己的专用 data.HeteroData 类中。

异构 Mini-Batch 加载:异构图可以分别通过 loader.DataLoader 和 loader.NextorLoader 对许多小的、单个的巨大图进行转化,转换成 mini-batches。这些 loaders 现在可以处理同构图和异构图。

异构图神经网络:异构 GNN 现在可以通过 nn.to_hetero、nn.to_hetero_with_bases 从同构 GNN 轻松创建。这些进程采用现有的 GNN 模型并复制其消息功能,以考虑不同的节点和边缘类型。

撑持异构图、集成GraphGym,超好用的图神经网络库PyG更新2.0版本

使用 GraphGym 管理实验

PyG 2.0 现在通过 torch_geometric.graphgym 正式撑持 GraphGym。总的来说,GraphGym 是一个平台,用于通过高度模块化的 pipeline 从配置文件中设计和评估图神经网络:

GraphGym 是开始进修标准化 GNN 实现和评估的最佳平台;

GraphGym 提供了一个简单的接口来并行尝试数千个 GNN 架构,以找到适合特定任务的最佳设计;

GraphGym 可轻松进行超参数搜索并可视化哪些设计选择更好。

撑持异构图、集成GraphGym,超好用的图神经网络库PyG更新2.0版本

重大改变

datasets.AMiner 数据集现在返回 data.HeteroData 对象;transforms.AddTrainValTestMask 已被替换为 transforms.RandomNodeSplit;由于 data.Data 的保存布局发生了显着变化以撑持异构图,因此需要通过删除 root/processed 文件夹来重新处理已处理的数据集。如果你修改了自定义 data.Data 对象中的__cat_dim__或__inc__ ,请确保应用上述更改。

了解更多变化,请参考原项目。

焦点团队成员

PyG 焦点团队共有 6 位成员,包括多特蒙德工业大学四年级图机器进修博士生 Matthias Fey、多特蒙德工业大学博士生 Jan Eric Lenssen(焦点贡献者)和图网络领域大牛 & 斯坦福计算机科学副教授 Jure Leskovec(担任顾问)。

撑持异构图、集成GraphGym,超好用的图神经网络库PyG更新2.0版本

从左至右依次为 Matthias Fey、Jan Eric Lenssen 和 Jure Leskovec。

此外还有三位华人成员,分别是斯坦福大学计算机科学四年级博士生尤佳轩(Jiaxuan You,焦点领导者)、斯坦福大学计算机科学博士生 Rex Ying(焦点贡献者,他将于 2022 年加入耶鲁大学担任助理教授)和卡内基梅隆大学 – 汉斯学院三年级博士生赵越(Yue Zhao,焦点贡献者)。

撑持异构图、集成GraphGym,超好用的图神经网络库PyG更新2.0版本

从左至右依次为尤佳轩、Rex Ying 和赵越。

日前,赵越在知乎分享了自己参与 PyG 2.0 版本设计与升级过程中的心得体会,感兴趣的读者可以参考阅读。

知乎链接:https://www.zhihu.com/pin/1420674585365196800

参考链接:

http://www.pyg.org/

给TA打赏
共{{data.count}}人
人已打赏
AI

一块V100运转上千个智能体、数千个情况,这个「曲率引擎」框架实行RL百倍提速

2021-9-12 12:43:00

AI

DeepMind联合UCL,推出2021加强进修最新课程

2021-9-16 14:09:00

0 条回复 A文章作者 M管理员
    暂无讨论,说说你的看法吧
个人中心
今日签到
搜索