Abstract:Link prediction is a key technique for connecting entities and relationships in a graph reasoning field. It leverages known information about the graph structure data to predict missing factual information. Previous studies have either focused on the semantic representation of a single triplet or on the graph structure data built on triples. The former ignores the association between different triples, and the latter ignores the true meaning of the node itself. Furthermore, common graph-structured datasets inherently face challenges, such as missing information and incompleteness. In light of this challenge, we present a novel model called Multi-source Information Graph Embedding with Ensemble Learning for Link Prediction (EMGE), which can effectively improve the reasoning of link prediction. Ensemble learning is systematically applied throughout the model training process. At the data level, this approach enhances entity embeddings by integrating structured graph information and unstructured textual data as multi-source information inputs. The fusion of these inputs is effectively addressed by introducing an attention mechanism. During the training phase, the principle of ensemble learning is employed to extract semantic features from multiple neural network models, facilitating the interaction of enriched information. To ensure effective model learning, a novel loss function based on contrastive learning is devised, effectively minimizing the discrepancy between predicted values and the ground truth. Moreover, to enhance the semantic representation of graph nodes in link prediction, two rules are introduced during the aggregation of graph structure information. These rules incorporate the concept of spreading activation, enabling a more comprehensive understanding of the relationships between nodes and edges in the graph. During the testing phase, the EMGE model is validated on three datasets, including WN18RR, FB15k-237, and a private Chinese financial dataset. The experimental results demonstrate a reduction in the mean rank (MR) by 0.2 times, an improvement in the mean reciprocal rank (MRR) by 5.9%, and an increase in the Hit@1 by 12.9% compared to the baseline model.

Heterogeneous Link Prediction Via Mutual Information Maximization Between Node Pairs.

Link Prediction Based on Contrastive Multiple Heterogeneous Graph Convolutional Networks

Link Prediction Based on Deep Global Information in Heterogeneous Graph

Heterogeneous hypergraph representation learning for link prediction

Mixture of Link Predictors on Graphs

Heterogeneous Deep Graph Infomax

Pay Attention to Weak Ties: A Heterogeneous Multiplex Representation Learning Framework for Link Prediction

Link prediction in heterogeneous information networks: An improved deep graph convolution approach

Multi-Tasking Link Prediction On Coupled Networks Via The Factor Graph Model

Topic-aware Heterogeneous Graph Neural Network for Link Prediction

Link Prediction Based on Heterogeneous Degree Penalization with Extending Neighbors and Clustering Coefficient

Learning Heterogeneous Graph Embedding with Metapath-Based Aggregation for Link Prediction

Motifs-based link prediction for heterogeneous multilayer networks

Research on a link-prediction method based on a hierarchical hybrid-feature graph

Research on improved algorithm for collaborative prediction of het- erogeneous links

Link prediction for heterogeneous information networks based on enhanced meta-path aggregation and attention mechanism

How to Bridge Spatial and Temporal Heterogeneity in Link Prediction? A Contrastive Method

Multi-Source Information Graph Embedding with Ensemble Learning for Link Prediction

Neighborhood Overlap-aware Heterogeneous Hypergraph Neural Network for Link Prediction

Link Prediction in Complex Networks: A Mutual Information Perspective

Exploiting Node Similarity Based on Graphical Markov Models for Link Prediction