Link Prediction Using Hebbian Graph Embeddings
share
Methods and systems for generating link predictions are provided. In one aspect, a method includes initializing a graph including a plurality of nodes representing selections of items in a training dataset to a multivariate normal distribution having a predetermined mean and a predetermined initial variance, the items in the training dataset comprising items in an item collection. The method includes, for each node in the graph, modeling embeddings for the node as embeddings of each neighboring node having a shared edge, with each being updated based at least in part on a transition probability and a variance. A predetermined number of iterations of updating are executed each iteration including an updated variance based on a learning rate. Based on receipt of an identification of an item from among the item collection, a plurality of predicted selections of items are identified using the embeddings for a node corresponding to the item.
READ FULL TEXT