提出SNGCL方法,提升图数据节点分类的对比学习效果
Simple Network Graph Comparative Learning
- 用叠加拉普拉斯平滑获取全局与局部特征矩阵
- 在多个数据集上优于主流模型,最高提升3.2个百分点
- 无需大量负样本,适合标签稀疏场景
对比学习在图学习领域已获广泛认可,尤其适用于标签稀缺或难以标注的图数据。然而,现有方法在节点分类任务中仍面临挑战:一是数据增强易导致新视图与原图差异过大,降低视图相关性;二是多数算法依赖大量负样本。为此,本文提出一种新型节点分类对比学习方法——简单网络图对比学习(SNGCL)。SNGCL采用叠加多层拉普拉斯平滑滤波器分别生成全局与局部特征平滑矩阵,输入孪生网络的目标与在线网络,并引入改进的三重重组损失函数,使类内距离更近、类间距离更远。在多个节点分类任务中与当前最优模型对比,实验结果表明SNGCL在大多数任务中表现优异,具有较强竞争力。
原文摘要 · Abstract (English)
The effectiveness of contrastive learning methods has been widely recognized in the field of graph learning, especially in contexts where graph data often lack labels or are difficult to label. However, the application of these methods to node classification tasks still faces a number of challenges. First, existing data enhancement techniques may lead to significant differences from the original view when generating new views, which may weaken the relevance of the view and affect the efficiency of model training. Second, the vast majority of existing graph comparison learning algorithms rely on the use of a large number of negative samples. To address the above challenges, this study proposes a novel node classification contrast learning method called Simple Network Graph Comparative Learning (SNGCL). Specifically, SNGCL employs a superimposed multilayer Laplace smoothing filter as a step in processing the data to obtain global and local feature smoothing matrices, respectively, which are thus passed into the target and online networks of the siamese network, and finally employs an improved triple recombination loss function to bring the intra-class distance closer and the inter-class distance farther. We have compared SNGCL with state-of-the-art models in node classification tasks, and the experimental results show that SNGCL is strongly competitive in most tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。