基本信息:
- 专利标题: METHOD AND APPARATUS FOR CLASSIFYING NODES OF A GRAPH
- 申请号:PCT/CN2021/132082 申请日:2021-11-22
- 公开(公告)号:WO2023087303A1 公开(公告)日:2023-05-25
- 发明人: KHARLAMOV, Evgeny , TANG, Jie , FENG, Wenzheng
- 申请人: ROBERT BOSCH GMBH , TSINGHUA UNIVERSITY
- 申请人地址: Postfach 30 02 20; Qinghua Yuan 1, Haidian District
- 专利权人: ROBERT BOSCH GMBH,TSINGHUA UNIVERSITY
- 当前专利权人: ROBERT BOSCH GMBH,TSINGHUA UNIVERSITY
- 当前专利权人地址: Postfach 30 02 20; Qinghua Yuan 1, Haidian District
- 代理机构: NTD PATENT & TRADEMARK AGENCY LTD.
- 主分类号: G06N3/04
- IPC分类号: G06N3/04 ; G06N3/08 ; G06Q50/00
摘要:
The present disclosure provides a method for training a Graph Neural Network (GNN) model to perform a task of classifying nodes of a graph based on semi-supervised learning. The method comprises: sampling a batch of labeled nodes and a batch of unlabeled nodes from the nodes of the graph, wherein the graph comprising nodes represented by a feature matrix and edges represented by an adjacency matrix, each of the nodes of graph being represented by a corresponding feature vector of the feature matrix; obtaining a plurality of augmented feature vectors for each node in the batch of labeled nodes and the batch of unlabeled nodes by randomly propagating feature vectors of neighboring nodes of the node based on an adjacency vector of the node; obtaining a plurality of classification predictions for each node in the batch of labeled nodes and the batch of unlabeled nodes by respectively applying the plurality of augmented feature vectors of the node to the GNN model; obtaining a loss based on the classification predictions of the nodes in the batch of labeled nodes and the batch of unlabeled nodes; and updating parameters of the GNN model based on the loss.
IPC结构图谱:
G | 物理 |
--G06 | 计算;推算;计数 |
----G06N | 基于特定计算模型的计算机系统 |
------G06N3/00 | 基于生物学模型的计算机系统 |
--------G06N3/02 | .采用神经网络模型 |
----------G06N3/04 | ..体系结构,例如,互连拓扑 |