Transformation of Node to Knowledge Graph Embeddings for Faster Link Prediction in Social Networks
This work addresses the computational bottleneck in link prediction for social networks, offering a practical solution for real-time applications, though it is incremental as it builds on existing embedding methods.
The paper tackles the problem of computationally expensive knowledge graph embeddings for link prediction by proposing a transformation model that converts inexpensive random walk-based node embeddings into knowledge graph embeddings without increasing computational cost, enabling real-time link prediction as demonstrated through extensive experimentation.
Recent advances in neural networks have solved common graph problems such as link prediction, node classification, node clustering, node recommendation by developing embeddings of entities and relations into vector spaces. Graph embeddings encode the structural information present in a graph. The encoded embeddings then can be used to predict the missing links in a graph. However, obtaining the optimal embeddings for a graph can be a computationally challenging task specially in an embedded system. Two techniques which we focus on in this work are 1) node embeddings from random walk based methods and 2) knowledge graph embeddings. Random walk based embeddings are computationally inexpensive to obtain but are sub-optimal whereas knowledge graph embeddings perform better but are computationally expensive. In this work, we investigate a transformation model which converts node embeddings obtained from random walk based methods to embeddings obtained from knowledge graph methods directly without an increase in the computational cost. Extensive experimentation shows that the proposed transformation model can be used for solving link prediction in real-time.