System, Method, and Computer Program Product for Predictive Modeling Using Hyperbolic Knowledge Graph Embeddings

A computer-implemented method comprising:

• receiving, with at least one processor, graph data associated with a knowledge graph comprising a plurality of nodes and a plurality of edges, each node of the plurality of nodes associated with an entity of a plurality of entities, each edge of the plurality of edges associated with a relationship between at least two of the plurality of entities, the knowledge graph comprising at least one triple, each respective triple comprising a respective head vector associated with a first respective entity of the plurality of entities, a respective tail vector associated with a second entity of the plurality of entities, and a respective relation vector associated with a respective relationship between the first respective entity and the second respective entity;
• generating, with at least one processor, a respective head embedding in a hyperbolic space for each respective head vector of at least a subset of the at least one triple, a respective tail embedding in the hyperbolic space for each respective tail vector of the at least the subset of the at least one triple, and a respective relation embedding in the hyperbolic space for each respective relation vector of the at least the subset of the at least one triple;
• determining, with at least one processor, a respective score for each respective triple of the at least the subset of the at least one triple based on the respective head embedding, the respective tail embedding, and the respective relation embedding;
• determining, with at least one processor, a loss based on the respective score for each respective triple of the at least the subset of the at least one triple;
• updating, with at least one processor, the respective head embedding, the respective tail embedding, and the respective relation embedding for each respective triple of the at least the subset of the at least one triple based on the loss;
• and
• repeating, with at least one processor, determining the respective score, determining the loss, and updating until a termination condition is satisfied.

A computer program product comprising at least one non-transitory computer-readable medium comprising program instructions that, when executed by at least one processor, cause the at least one processor to:

• receive graph data associated with a knowledge graph comprising a plurality of nodes and a plurality of edges, each node of the plurality of nodes associated with an entity of a plurality of entities, each edge of the plurality of edges associated with a relationship between at least two of the plurality of entities, the knowledge graph comprising at least one triple, each respective triple comprising a respective head vector associated with a first respective entity of the plurality of entities, a respective tail vector associated with a second entity of the plurality of entities, and a respective relation vector associated with a respective relationship between the first respective entity and the second respective entity;
• generate a respective head embedding in a hyperbolic space for each respective head vector of at least a subset of the at least one triple, a respective tail embedding in the hyperbolic space for each respective tail vector of the at least the subset of the at least one triple, and a respective relation embedding in the hyperbolic space for each respective relation vector of the at least the subset of the at least one triple;
• determine a respective score for each respective triple of the at least the subset of the at least one triple based on the respective head embedding, the respective tail embedding, and the respective relation embedding;
• determine a loss based on the respective score for each respective triple of the at least the subset of the at least one triple;
• update the respective head embedding, the respective tail embedding, and the respective relation embedding for each respective triple of the at least the subset of the at least one triple based on the loss;
• and
• repeat determining the respective score, determining the loss, and updating until a termination condition is satisfied.