Knowledge graph reasoning method and apparatus, model training method and apparatus, and computer device

Patent metadata

Field	Value
Publication number	US-2023351221-A1
Application number	US-202318305279-A
Country	US
Kind code	A1
Filing date	Apr 21, 2023
Priority date	Apr 28, 2022
Publication date	Nov 2, 2023
Grant date	—

Abstract

Implementations of the present specification disclose a knowledge graph reasoning method and apparatus, a model training method and apparatus, and a computer device. The method includes: obtaining a query entity and a query relationship; selecting one or more nearest neighbor entities of the query entity from a knowledge graph; determining a first probability of a nearest neighbor entity of the one or more nearest neighbor entities, where the first probability is used to indicate a possibility that the nearest neighbor entity is in communication with the query relationship; selecting a nearest neighbor entity of the one or more nearest neighbor entities as a candidate entity based on the first probability; and selecting a candidate entity matching the query entity and the query relationship as a result entity. In the implementations of the present specification, the efficiency of knowledge graph reasoning can be improved.

First claim

What is claimed is:

1 . A method, comprising:
obtaining a query entity and a query relationship; selecting one or more nearest neighbor entities of the query entity from a knowledge graph; determining a first probability of a nearest neighbor entity of the one or more nearest neighbor entities, the first probability being used to indicate a possibility that the nearest neighbor entity is in communication with the query relationship; selecting a nearest neighbor entity of the one or more nearest neighbor entities as a candidate entity based on the first probability; and selecting a candidate entity matching the query entity and the query relationship as a result entity.

2 . The method according to claim 1 , wherein the selecting the one or more nearest neighbor entities of the query entity from the knowledge graph includes:
selecting an entity whose proximity to the query entity is less than or equal to a K1 order from the knowledge graph as a nearest neighbor entity.

3 . The method according to claim 1 , wherein the determining the first probability of the nearest neighbor entity includes:
determining type distribution information of the nearest neighbor entity, wherein the type distribution information indicates possibilities that the nearest neighbor entity is in communication with a plurality of known entity relationships; and determining the first probability of the nearest neighbor entity based on the type distribution information and the query relationship.

4 . The method according to claim 3 , wherein the determining the type distribution information of the nearest neighbor entity includes:
extracting a sub-knowledge graph of the nearest neighbor entity from the knowledge graph; and determining the type distribution information of the nearest neighbor entity based on the sub-knowledge graph and a type distribution prediction model.

5 . The method according to claim 4 , wherein the extracting the sub-knowledge graph of the nearest neighbor entity from the knowledge graph includes:
selecting one or more entities each having proximity to the nearest neighbor entity less than or equal to a K2 order from the knowledge graph; and extracting a sub-knowledge graph that includes the one or more entities each having proximity to the nearest neighbor entity less than or equal to the K2 order.

6 . The method according to claim 4 , wherein the type distribution prediction model includes a graph neural network model; and
the determining the type distribution information of the nearest neighbor entity includes:
inputting graph structure data of the sub-knowledge graph to the graph neural network model to obtain the type distribution information of the nearest neighbor entity, wherein the graph structure data includes an embedding representation of an entity and an embedding representation of an entity relationship.

7 . The method according to claim 4 , wherein the type distribution prediction model is obtained by training including:
masking one or more entity relationships of a target entity in a knowledge graph sample to obtain a masked knowledge graph sample; determining type distribution information of the target entity based on the masked knowledge graph sample and the type distribution prediction model, wherein the type distribution information indicates possibilities that the target entity is in communication with a plurality of known entity relationships; determining a third probability of the target entity based on the type distribution information and the masked entity relationship, wherein the third probability indicates a possibility that the target entity is in communication with the masked entity relationship; and optimizing a model parameter of the type distribution prediction model based on the third probability.

8 . The method according to claim 7 , wherein the training further comprises:
determining a fourth probability of the target entity based on the type distribution information and an identified entity relationship, wherein the fourth probability indicates a possibility that the target entity is in communication with the identified entity relationship, and the identified entity relationship includes an entity relationship not in communication with the target entity; and wherein the optimizing the model parameter of the type distribution prediction model includes: optimizing the model parameter of the type distribution prediction model based on the third probability and the fourth probability.

9 . The method according to claim 1 , wherein the selecting a nearest neighbor entity of the one or more nearest neighbor entities as the candidate entity includes:
calculating a second probability of a nearest neighbor entity of the one or more nearest neighbor entities based on the first probability and a degree of the nearest neighbor entity; and selecting a nearest neighbor entity of the one or more nearest neighbor entities as the candidate entity based on the second probability.

10 . The method according to claim 1 , wherein the selecting a candidate entity matching the query entity and the query relationship as the result entity includes:
constructing a candidate triplet based on a candidate entity, the query entity, and the query relationship; selecting a candidate triplet as a target triplet based on a confidence of the candidate triplet; and determining a candidate entity in the target triplet as the result entity.

11 . The method according to claim 1 , wherein the query entity is a head entity, and the result entity is a tail entity; or the query entity is the tail entity and the result entity is the head entity.

12 . A method, comprising:
masking one or more entity relationships of a target entity in a knowledge graph sample; determining type distribution information of the target entity based on a masked knowledge graph sample and a type distribution prediction model, wherein the type distribution information indicates possibilities that the target entity is in communication with a plurality of known entity relationships; determining a first probability of the target entity based on the type distribution information and the masked entity relationship, wherein the first probability indicates a possibility that the target entity is in communication with the masked entity relationship; and optimizing a model parameter of the type distribution prediction model based on the first probability.

13 . The method according to claim 12 , wherein further comprising:
determining a second probability of the target entity based on the type distribution information and an identified entity relationship, wherein the second probability indicates a possibility that the target entity is in communication with the identified entity relationship, and the identified entity relationship includes an entity relationship not in communication with the target entity; and the optimizing the model parameter of the type distribution prediction model includes: optimizing the model parameter of the type distribution prediction model based on the first probability and the second probability.

14 . A computer system, comprising:
at least one processor; and at least one memory device having executable instructions stored thereon, the executable instructions when executed by the at least one processor enabling the at least one processor to implement acts including: obtaining a query entity and a query relationship; selecting one or more nearest neighbor entities of the query entity from a knowledge graph; determining a first probability of a nearest neighbor entity of the one or

Assignees

• Alipay Hangzhou Inf Tech Co Ltd

Inventors

• Chao Linlin
• Wang Taifeng
• Chu Wei

Classifications

• G06N5/022 Primary

  Knowledge engineering; Knowledge acquisition · CPC title

• G06N3/047

  Probabilistic or stochastic networks · CPC title

• G06F16/367 Primary

  Ontology · CPC title

• G06F16/3346

  using probabilistic model · CPC title

• G06N5/02

  Knowledge representation; Symbolic representation · CPC title