CogDL:基于图的深度学习的研究工具
CogDL是由清华大学计算机系知识工程实验室(KEG)开发的基于图的深度学习的研究工具,基于Python语言和PyTorch库。CogDL允许研究人员和开发人员可以轻松地训练和比较基线算法或自定义模型,以进行结点分类,链接预测,图分类,社区发现等基于图结构的任务。 它提供了许多流行模型的实现,包括:非图神经网络算法例如Deepwalk、LINE、Node2vec、NetMF、ProNE、methpath2vec、PTE、graph2vec、DGK等;图神经网络算法例如GCN、GAT、GraphSAGE、FastGCN、GTN、HAN、GIN、DiffPool等。它也提供了一些下游任务,包括结点分类(分为是否具有节点属性),链接预测(分为同构和异构),图分类(分有监督和?监督)以及为这些任务构建各种算法效果的排行榜。
CogDL的整体框架如上图所示,针对不同的任务,CogDL支持以下模型:
- 无监督结点分类: ProNE?(Zhang et al, IJCAI’19), NetMF?(Qiu et al, WSDM’18), Node2vec?(Grover et al, KDD’16), NetSMF?(Qiu et at, WWW’19), DeepWalk?(Perozzi et al, KDD’14), LINE?(Tang et al, WWW’15), Hope?(Ou et al, KDD’16), SDNE?(Wang et al, KDD’16), GraRep?(Cao et al, CIKM’15), DNGR?(Cao et al, AAAI’16).
- 半监督结点分类: SGC-PN?(Zhao & Akoglu, 2019), Graph U-Net?(Gao et al., 2019), MixHop?(Abu-El-Haija et al., ICML’19), DR-GAT?(Zou et al., 2019), GAT?(Veli?kovi? et al., ICLR’18), DGI?(Veli?kovi? et al., ICLR’19), GCN?(Kipf et al., ICLR’17), GraphSAGE?(Hamilton et al., NeurIPS’17), Chebyshev?(Defferrard et al., NeurIPS’16).
- 异构结点分类: GTN?(Yun et al, NeurIPS’19), HAN?(Xiao et al, WWW’19), PTE?(Tang et al, KDD’15), Metapath2vec?(Dong et al, KDD’17), Hin2vec?(Fu et al, CIKM’17).
- 链接预测: ProNE?(Zhang et al, IJCAI’19), NetMF?(Qiu et al, WSDM’18), Node2vec?(Grover et al, KDD’16), DeepWalk?(Perozzi et al, KDD’14), LINE?(Tang et al, WWW’15), Hope?(Ou et al, KDD’16), NetSMF?(Qiu et at, WWW’19), SDNE?(Wang et al, KDD’16).
- 多重边链接预测: GATNE?(Cen et al, KDD’19), NetMF?(Qiu et al, WSDM’18), ProNE?(Zhang et al, IJCAI’19), Node2vec?(Grover et al, KDD’16), DeepWalk?(Perozzi et al, KDD’14), LINE?(Tang et al, WWW’15), Hope?(Ou et al, KDD’16), GraRep?(Cao et al, CIKM’15).
- 无监督图分类: Infograph?(Sun et al, ICLR’20), Graph2Vec?(Narayanan et al, CoRR’17), DGK?(Yanardag et al, KDD’15).
- 有监督图分类: GIN?(Xu et al, ICLR’19), DiffPool?(Ying et al, NeuIPS’18), SortPool?(Zhang et al, AAAI’18), PATCH_SAN?(Niepert et al, ICML’16), DGCNN?(Wang et al, ACM Transactions on Graphics’17).
1.安装方法
PyTorch安装好之后,cogdl能够直接通过pip来安装:
pip install cogdl
2.使用说明
https://github.com/THUDM/cogdl/blob/master/README_CN.md
项目页面:http://keg.cs.tsinghua.edu.cn/cogdl
GitHub 链接:https://github.com/THUDM/cogdl
中文介绍:https://github.com/THUDM/cogdl/blob/master/README_CN.md
智源链接:http://open.baai.ac.cn/cogdl-toolkit
网站(中文):http://keg.cs.tsinghua.edu.cn/cogdl/cn/
文档:https://cogdl.readthedocs.io/