.png){kind=icon}
返回
Edge contrastive learning for link prediction
.png){kind=icon}
PRE
AI
DOI：10.1016/j.ipm.2024.103847.png)
.png){kind=icon}
原文求助.png){kind=icon}
分享.png){kind=icon}
收藏
摘要
En 中文
Link prediction is a critical task within the realm of graph machine learning. While recent advancements mainly emphasize node representation learning, the rich information encapsulated within edges, proven advantageous in various graph-related tasks, has been somewhat overlooked. To bridge the gap, this paper explores the potential of incorporating edge representation learning for link prediction and identifies three inherent challenges associated with this approach. We introduce the Edge Contrastive Learning for Link Prediction (ECLiP) framework to tackle these challenges. ECLiP integrates edge information into node representations through edge-level contrastive learning, with a distinctive perspective on treating edges, rather than nodes, as the units of instance discrimination. We first illustrate the implementation of this framework using an established edge representation learning method. However, it incurs significant additional training overhead when the number of edges is huge. To mitigate this issue, we present a computationally efficient variant employing a multi-layer perceptron (MLP) for direct edge representation learning. Conducting rigorous experiments across eight distinct datasets with node counts spanning from 2k to 235k, we demonstrate a noteworthy improvement of over 10% on certain datasets, validating the efficacy of our proposed methodology.
Keyword:
Link prediction
Graph neural networks
Contrastive learning
Edge representation learning
期刊
I
IF:
6.9
论文数:
5.2K
被引数:
1.4W
机构
引用论文
分享收藏分享收藏Learning dual disentangled representation with self-supervision for temporal knowledge graph reasoning学习具有自监督的双重解纠缠表示的时间知识图推理
PREAI
分享收藏分享收藏Graph-based multi-label disease prediction model learning from medical data and domain knowledge
PREAI
分享收藏Light dual hypergraph convolution for collaborative filtering用于协同过滤的轻型对偶超图卷积.png)
PATTERN RECOGNITION
PATTERN RECOGNITIONIF7.6
PREAI
分享收藏
分享收藏
查看更多内容