公开(公告)号：CN117391142A

公开(公告)日：2024-01-12

申请号：CN202210774931.1

申请日：2022-07-01

Applicant: 第四范式(北京)技术有限公司

Inventor： 赵欢 ,  王旭

IPC: G06N3/0464 ,  G06N3/08

Abstract: 公开了一种图神经网络模型设计方法和系统。所述方法包括：构建与图分类任务对应的图神经网络结构搜索框架，并基于所述图神经网络结构搜索框架构建结构搜索空间；在所述结构搜索空间中搜索针对所述图分类任务的图神经网络结构；基于搜索出的图神经网络结构构建图神经网络模型，得到具有搜索出的图神经网络结构的图神经网络模型。

公开(公告)号：CN118114711A

公开(公告)日：2024-05-31

申请号：CN202211528400.0

申请日：2022-11-30

Applicant: 第四范式(北京)技术有限公司

Inventor： 王旭 ,  赵欢 ,  涂威威

IPC: G06N3/042 ,  G06N3/045 ,  G06N3/0499 ,  G06N3/09

Abstract: 提供了一种图神经网络模型生成方法、使用该模型的预测方法及装置。上述图神经网络模型生成方法包括：获取第一训练样本和第二训练样本，其中，所述第一训练样本包括第一图样本以及第一图样本的真实三维结构标签，所述第二训练样本包括第二图样本以及第二图样本的真实样本性质标签；基于第一训练样本对预训练模型进行预训练，得到训练好的预训练模型，所述预训练模型的输出被用于预测图样本的三维样本结构；基于训练好的预训练模型构建图神经网络模型；以及基于第二训练样本对所述图神经网络模型进行训练，得到训练好的图神经网络模型，其中，所述图神经网络模型的输出被用于预测图样本的样本性质。