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](https://img-blog.csdnimg.cn/img_convert/dcbce957c241211e6e4a104729d8058f.png)
![b433986ee60778ffbdede4d9ab6febca.png](https://img-blog.csdnimg.cn/img_convert/b433986ee60778ffbdede4d9ab6febca.png)
每个Trick都会被验证。我们会选择数据集对trick进行验证,只有能够确实提高模型在测试数据集上表现的trick会被收录到gtrick中。
![1fe7758abc113cf58792902eb116284a.png](https://img-blog.csdnimg.cn/img_convert/1fe7758abc113cf58792902eb116284a.png)
同时支持DGL与PyG。我们对每个trick同时提供DGL与PyG两种实现以及相应的示例。无论您喜欢哪款图深度学习库,都可以尝试使用它。
![4059b7e9acf96617c569176d0e2c348d.png](https://img-blog.csdnimg.cn/img_convert/4059b7e9acf96617c569176d0e2c348d.png)
此外,值得注意的是,虽然一些trick已在PyG或DGL中实现,这种trick我们仍然会纳入gtrick中并提供另一个图深度学习库的实现,因为gtrick的第一目标是让大家在各种鱼目混珠的模型中了解到有这么一些可以真实有效地提高模型表现的方法存在。
安装
目前gtrick已经收录了10个针对不同任务的trick,并且可以通过pip直接安装:
pip install gtrick
目前gtrick还在开发中,并且各种实现也比较粗糙。您也可以直接将相应trick的代码加入到您的项目中。
最后,欢迎使用,欢迎提建议,热烈欢迎点个star: https://github.com/sangyx/gtrick