Multi-feature Based Link Prediction Algorithm Fusing Graph Attention

ZHANG Yancao,ZHAO Yuhai,SHI Lan
DOI: https://doi.org/10.3778/j.issn.1673-9418.2012092
2022-01-01
Abstract:Link prediction is one of the important research directions in complex networks. Using neural networks to learn predefined heuristic features has attracted extensive attention in recent years. However, at present, this kind of method mainly uses the local subgraph of the target link to predict the link, which has strong locality. To solve this problem, based on SEAL (learning from subgraphs, embeddings and attributes for link prediction) algorithm, this paper proposes an algorithm ADNSL (learning from double-radius node labeling, node2vec and struc2vec for link prediction with attention mechanism) for link prediction using multi-feature fusing graph attention, which supports multi-type node embedding features as inputs, including local feature generation and global feature extraction. For the local feature generation module, the node features in the local subgraph are interactively fused by using the graph convolution layer. In order to make up for the feature ineffectiveness and node unbiasedness in SEAL, a bidirectional parameterless attention is proposed. The global feature extraction module uses iterative formula to generate agglomerative graph to reduce the complexity of struc2vec node embedding algorithm, and then mines interpretable structural features from a global perspective, which can effectively improve the performance of link prediction algorithm. Experiments show that ADNSL algorithm can reasonably utilize multi-type node embedding features, and its performance is obviously better than that of multiple benchmark algorithms across eight real datasets in different fields.
What problem does this paper attempt to address?