链接:https://www.aminer.cn/pub/5f03f3b611dc830562231fe7?conf=kdd2020 图表示学习目前受到了广泛关注,但目前绝大多数的图表示学习方法都是针对特定领域的图进行学习和建模,所产出的图神经网络难以迁移。近期,预训练在多个领域都取得了巨大的成功,显著地提升了模型在各大下游任务的表现。受到 BERT (Devlin et al., 2018),MoCo (He et al., 2020) 等工作的启发,我们开始研究图神经网络的预训练,希望能够从中学习到通用的图拓扑结构特征。我们提出了 Graph Contrastive Coding的图神经网络预训练框架,利用对比学习(Contrastive Learning)的方法学习到内在的可迁移的图结构信息。本工作 GCC: Graph Contrastive Coding for Graph Neural Network Pre-Training 已被 KDD 2020 research track 录用。
GCC 可以应用于许多下游的图学习任务。我们探索了其在节点分类、图分类和相似搜索中的应用。具体来说,在节点分类任务中,我们可以使用 GCC 的预训练好的编码器(encoder)对节点的 r-ego network 或者一个采样得到的子图进行编码;在图分类任务中,我们可以直接使用 GCC 的编码器对输入的图直接进行编码;而在相似搜索中,我们想要衡量两个节点的结构相似度,我们可以使用 GCC 的编码器分别对两个节点的 r-ego network(或者采样得到的子图)进行编码,然后计算编码后的距离作为相似度。
实验结果分别如下:
相似搜索
从实验结果不难看出,GCC 在多个任务多个数据集上都取得了比较突出的表现,和现有最优模型相比能够取得更优或相近的表现。这也表现了 GCC 的有效性。
综上所述,本文提出的 GCC 的图神经网络预训练框架利用对比学习的方法,有效地学习了图结构的通用表征征,并且学习到图的结构化信息,可以迁移到各类下游任务和各类图中。实验表明了该方法的有效性。未来我们会在更多的任务和实验上进行实验,并探索 GCC 在其他领域的应用。
参考文献:
Devlin, J., Chang, M. W., Lee, K., & Toutanova, K. (2018). Bert: Pre-training of deep bidirectional transformers for language understanding. arXiv preprint arXiv:1810.04805.
He, K., Fan, H., Wu, Y., Xie, S., & Girshick, R. (2020). Momentum contrast for unsupervised visual representation learning. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (pp. 9729-9738).
联系客服