提出新图增强方法,提升链接预测性能
Enhancing Contrastive Link Prediction With Edge Balancing Augmentation
- 基于节点度平衡设计新型图增强策略
- 在8个数据集上显著超越现有最优模型
- 首次为对比学习链接预测提供理论分析
链接预测是图挖掘中最基础的任务之一,近期研究尝试利用对比学习提升性能。然而我们发现现有研究存在两大缺陷:一是在链接预测中缺乏对比学习的理论分析;二是未充分考虑节点度的影响。为此,我们首次为链接预测中的对比学习提供了正式的理论分析,其结果可推广至基于自编码器的对比学习模型。基于分析结果,我们提出一种新图增强方法——边平衡增强(EBA),通过调整图中节点度实现增强。进一步提出对比链接预测与边平衡增强(CoEBA)方法,结合EBA与新型对比损失以提升模型性能。在8个基准数据集上进行实验,结果表明所提CoEBA显著优于其他最先进的链接预测模型。
原文摘要 · Abstract (English)
Link prediction is one of the most fundamental tasks in graph mining, which motivates the recent studies of leveraging contrastive learning to enhance the performance. However, we observe two major weaknesses of these studies: i) the lack of theoretical analysis for contrastive learning on link prediction, and ii) inadequate consideration of node degrees in contrastive learning. To address the above weaknesses, we provide the first formal theoretical analysis for contrastive learning on link prediction, where our analysis results can generalize to the autoencoder-based link prediction models with contrastive learning. Motivated by our analysis results, we propose a new graph augmentation approach, Edge Balancing Augmentation (EBA), which adjusts the node degrees in the graph as the augmentation. We then propose a new approach, named Contrastive Link Prediction with Edge Balancing Augmentation (CoEBA), that integrates the proposed EBA and the proposed new contrastive losses to improve the model performance. We conduct experiments on 8 benchmark datasets. The results demonstrate that our proposed CoEBA significantly outperforms the other state-of-the-art link prediction models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。