论文标题
线图神经网络用于链接预测
Line Graph Neural Networks for Link Prediction
论文作者
论文摘要
我们考虑图形链接预测任务,这是许多现实世界应用程序的经典图分析问题。随着深度学习的进步,当前的链接预测方法通常从以两个相邻节点为中心的子图中计算特征,并使用这些功能来预测这两个节点之间的链接标签。在这种形式主义中,链接预测问题将转换为图形分类任务。为了提取用于分类的固定尺寸功能,在深度学习模型中需要进行图形池层,从而导致信息丢失。为了克服这一关键限制,我们建议通过使用图理论中的线图寻求一条根本不同和新颖的路径。特别是,线图中的每个节点都对应于原始图中的唯一边缘。因此,原始图中的链接预测问题可以等效地作为其相应的线图中的节点分类问题,而不是图形分类任务。来自不同应用程序的14个数据集的实验结果表明,我们所提出的方法始终优于最先进的方法,而其参数较少和高训练效率。
We consider the graph link prediction task, which is a classic graph analytical problem with many real-world applications. With the advances of deep learning, current link prediction methods commonly compute features from subgraphs centered at two neighboring nodes and use the features to predict the label of the link between these two nodes. In this formalism, a link prediction problem is converted to a graph classification task. In order to extract fixed-size features for classification, graph pooling layers are necessary in the deep learning model, thereby incurring information loss. To overcome this key limitation, we propose to seek a radically different and novel path by making use of the line graphs in graph theory. In particular, each node in a line graph corresponds to a unique edge in the original graph. Therefore, link prediction problems in the original graph can be equivalently solved as a node classification problem in its corresponding line graph, instead of a graph classification task. Experimental results on fourteen datasets from different applications demonstrate that our proposed method consistently outperforms the state-of-the-art methods, while it has fewer parameters and high training efficiency.