支撑异构图、集成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.0PyG 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/

原创文章,作者:机器之心,如若转载,请注明出处:https://www.iaiol.com/news/zhi-cheng-yi-gou-tu-ji-cheng-graphgym-chao-hao-yong-de-tu/

(0)
上一篇 2021年 9月 12日 下午12:43
下一篇 2021年 9月 16日 下午2:09

相关推荐

  • 新石器无人车获得海内首批无人配送车上路天分

    2021年5月25日,北京高级别主动驾驭示范区发布无人配送车管理政策,新石器无人车作为首批无人配送车头部企业获得海内首批无人配送车上路天分。预计到今年6月底,超过150辆新石器无人车作为离用户最近的无人驾驭便利店,将在海内首个智能网联汽车政策先行区内构建无人车服务网络,常态化供应便民服务。新石器无人车作为新形态道路交通参与者,在零卖、安防和金融等众多领域有着广泛的应用场景。在“上路”(公开道路行驶)前,新石器无人车早已在亦城国际、BDA企业大道、北京亦庄生物医药园等多个园区,面向&ldquo

    2021年 5月 26日
  • 如何通过呆板进修算法,将EV电池运用到极致?

    编译 / 刘梦婷近日,剑桥大学的研讨职员开发了一种呆板进修算法,可以通过展望分别的驾驭形式对电池性能的影响,帮助电动车削减充电时光,延长电池寿命,提高安全性和可靠性。研讨结果发表在《自然通讯》(Nature Communications)杂志上。该团队开发了一种非侵入式要领来检测电池,并获得电池整体健全状况。然后,将这些结果输入呆板进修算法,该算法可以展望分别驾驭形式将如何影响电池未来的健全状况。研讨职员表示,该算法可以通过建议路线和驾驭形式,最大限度地削减电池退步和充电时光,来充分利用电动汽车的电池。如果将其用于

    2022年 8月 25日
  • 欧洲投资银行供应1.5亿欧元反对欧洲人工智能企业

    欧洲投资银行团体(EIB Group)启动了一项高达1.5亿欧元的新的融资东西,反对人工智能及与人工智能直接相干/互补的区块链、物联网和机器人技能。与新的融资东西相干的资金反对将在将来四年内布置,供应给欧盟和Horizon 2020相干国家与地区,重点投资开发突破性人工智能应用的晚期和成长阶段的公司。这项东西是EIB团体和欧盟委员会更大计划的一部分,旨在反对高性能计算、量子技能和网络安全等规模的欧洲数字将来倒退。12月3日,在2020年Web峰会上,欧洲投资银行团体(EIB Group)启动了一项新的融资东西,以支

    2020年 12月 9日
  • 小鹏P5上海车展正式开启预订,载人飞行器同台亮相

    2021年4月19日,小鹏汽车携旗下第三款产品,全球首款搭载激光雷达的量产智能汽车——小鹏P5亮相2021上海车展,并举办以“天生智能 探究不止”为主题的新闻发布会。小鹏汽车董事长 CEO何小鹏与所有到场嘉宾一起分享如何让智能汽车更好地改变我们的出行生活,并宣布小鹏P5正式开启预订。2021上海车展小鹏汽车展台汽车董事长 CEO何小鹏在发布会现场,何小鹏表示:“我们始终在探究未来智能出行的可能性,无论是小鹏P5,还是旅航者X1,都是我们阶段性思考出来的产品

    2021年 4月 22日
  • 浙大校友李旻辰获SIGGRAPH 2021最佳博士论文奖,连续四年华人学者包揽此奖项

    由于疫情的影响,盘算机图形顶级会议ACM SIGGRAPH 2021于8月9日至15日线上举行。该大会颁布了最佳博士论文奖以及盘算机图形学成就奖等奖项,其中最佳博士论文奖由UCLA博士后李旻辰摘得,这也是华人学者连续四年获得该殊荣。

    2021年 8月 10日
  • CVPR 2022 | 联邦学习审计隐衷新手段,田纳西大学等提出天生式梯度流露格式GGL

    本文提出一种利用天生模型作为图片先验的梯度袭击格式GGL,由来自美国田纳西大学,美国橡树岭国家试验室,和谷歌共同完成,论文已被 CVPR 2022 接收。

    2022年 4月 10日
  • 27亿参数的「野生版」GPT-3开源,GitHub项目2.9K Star量

    OpenAI 著名的说话模型 GPT-3 可以懂数学、翻译文字,还能写论文拿到及格成绩,这样的 AI 什么时候才能开源呢?现在已有了一个差不多的项目。

    2021年 3月 26日
  • 人工智能帮助创造系外行星上的份子,甚至有一天大概会创造新的物理定律

    编辑 | 萝卜皮你知道地球的大气是由什么组成的吗?你大概记得它是氧气,也大概是氮气。借助 Google 的一点帮助,您可以轻松得出更精确的谜底:78% 的氮气、21% 的氧气和 1% 的氩气。大气可以表明行星的性质,以及它们是不是可以承载生命。然而,当谈到外大气层——太阳系外行星的大气层——的组成时,谜底是未知的。由于系外行星如此遥远,已证明探测它们的大气层极其困难。研讨表明,人工智能 (AI) 大概是探索它们的最佳选择——但前提是能够证明这些算

    2021年 12月 10日
  • 神经搜集debug太难了,这里有六个实用技巧

    神经搜集的 debug 过程着实不容易,这里是一些有所帮助的 tips。

    2021年 6月 11日
  • 可信AI的驱动力——隐衷估计

    「机器之心2021-2022年度AI趋势大咖说」聚焦「驱动未来的AI手艺」与「重塑产业的AI科技」,推出线上分享,共邀请近40位AI领域知名学者、产业专家及企业高管通过主题分享及多人圆桌等形式,与行业精英、读者、观众共同回顾 2021年中的重要手艺和学术热点,盘点AI产业的年度研究方向以及重大科技突破,展望2022年度AI手艺发展方向、AI手艺与产业科技融合趋势。

    2022年 7月 21日

发表回复

您的电子邮箱地址不会被公开。 必填项已用 * 标注