Determining missing relationship information and augmenting a knowledge graph

A computer system comprising:

• a memory;
• and
• a processor in communication with the memory, wherein the computer system is configured to perform a method comprising:
• obtaining an initial knowledge graph comprising nodes representing a plurality of entities and edges representing relationships between related entities of the plurality of entities;
• building a dataset based on the initial knowledge graph, the dataset indicating, for each related entity pair, of a collection of related entity pairs, between which a relationship exists as represented in the initial knowledge graph:
• at least one attribute and at least one attribute value of a first entity of the related entity pair;
• at least one attribute and at least one attribute value of a second entity of the related entity pair;
• a relationship type of the relationship between the related entity pair;
• and
• a weight of the relationship type;
• using the dataset to build a collection of machine learning models for identifying missing relationship information, the collection of machine learning models comprising at least one clustering model and at least one classification model;
• receiving a partial specification of a new relationship triple to include in the initial knowledge graph, the new relationship triple comprising a new source entity, a target entity, and a relationship type of a relationship between the new source entity and the target entity, wherein the partial specification of the new relationship triple comprises a specification of the new source entity and is missing at least some relationship information of the new relationship triple;
• applying one or more machine learning models of the collection of machine learning models and identifying the missing at least some relationship information of the new relationship triple to provide a complete specification of the new relationship triple;
• and
• augmenting the initial knowledge graph using the complete specification of the new relationship triple, the augmenting providing an augmented knowledge graph that includes the new relationship triple, including the new source entity, the target entity, and the relationship type of the relationship between the new source entity and the target entity.

A computer program product comprising:

• a computer readable storage medium readable by a processing circuit and storing instructions for execution by the processing circuit to:
• obtaining an initial knowledge graph comprising nodes representing a plurality of entities and edges representing relationships between related entities of the plurality of entities;
• building a dataset based on the initial knowledge graph, the dataset indicating, for each related entity pair, of a collection of related entity pairs, between which a relationship exists as represented in the initial knowledge graph:
• at least one attribute and at least one attribute value of a first entity of the related entity pair;
• at least one attribute and at least one attribute value of a second entity of the related entity pair;
• a relationship type of the relationship between the related entity pair;
• and
• a weight of the relationship type;
• using the dataset to build a collection of machine learning models for identifying missing relationship information, the collection of machine learning models comprising at least one clustering model and at least one classification model;
• receiving a partial specification of a new relationship triple to include in the initial knowledge graph, the new relationship triple comprising a new source entity, a target entity, and a relationship type of a relationship between the new source entity and the target entity, wherein the partial specification of the new relationship triple comprises a specification of the new source entity and is missing at least some relationship information of the new relationship triple;
• applying one or more machine learning models of the collection of machine learning models and identifying the missing at least some relationship information of the new relationship triple to provide a complete specification of the new relationship triple;
• and
• augmenting the initial knowledge graph using the complete specification of the new relationship triple, the augmenting providing an augmented knowledge graph that includes the new relationship triple, including the new source entity, the target entity, and the relationship type of the relationship between the new source entity and the target entity.