gtrick: 与其绞尽脑汁炼丹,不如给你的GNN撒点trick吧!

95690ad578f04f6048627ac8ff5a66ea.png

2nd OGB-LSC已经开启,你还不知道这款GNN上分利器吗?

gtrick (https://github.com/sangyx/gtrick) 是一个简单易用,面向GNN的trick工具箱。

idea

这一项目的idea来自于对OGB榜单的观察,你会发现虽然模型变来变去,但大家上分的trick却是有限的。

既然如此,为何不直接打包一下这些trick做一个方便大家使用的库呢?

此外,GNN的任务场景比较集中,输入也比较固定,十分适合实现一些外挂式的trick,不用涉及对模型的修改。

基于这样一个想法,gtrick应运而生。

优点

gtrick有三个值得称道的优点:

  • 简单易用。如下图所示,只需要添加简单的几行代码即可将一个trick引入您的GNN模型中,对已有的代码做尽可能少的变动。

dcbce957c241211e6e4a104729d8058f.png
原始模型
b433986ee60778ffbdede4d9ab6febca.png
引入trick: Random Feature
  • 每个Trick都会被验证。我们会选择数据集对trick进行验证,只有能够确实提高模型在测试数据集上表现的trick会被收录到gtrick中。

1fe7758abc113cf58792902eb116284a.png
一些收录trick在ogbn-arxiv上的测试结果
  • 同时支持DGL与PyG。我们对每个trick同时提供DGL与PyG两种实现以及相应的示例。无论您喜欢哪款图深度学习库,都可以尝试使用它。

4059b7e9acf96617c569176d0e2c348d.png
gtrick已实现10个trick并提供相应的示例

此外,值得注意的是,虽然一些trick已在PyG或DGL中实现,这种trick我们仍然会纳入gtrick中并提供另一个图深度学习库的实现,因为gtrick的第一目标是让大家在各种鱼目混珠的模型中了解到有这么一些可以真实有效地提高模型表现的方法存在。

安装

目前gtrick已经收录了10个针对不同任务的trick,并且可以通过pip直接安装:

pip install gtrick

目前gtrick还在开发中,并且各种实现也比较粗糙。您也可以直接将相应trick的代码加入到您的项目中。

最后,欢迎使用,欢迎提建议,热烈欢迎点个star: https://github.com/sangyx/gtrick

  • 2
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值